GithubBackup/BackupX
1
0
Fork 0
mirror of https://github.com/Awuqing/BackupX.git synced 2026-08-09 22:43:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

功能: NodeRepository.FindByToken 支持 prev_token 24h 过渡

Browse Source
This commit is contained in:
Awuqing
2026-04-19 16:15:03 +08:00
parent 65cbbea3b6
commit fb043627bb
2 changed files with 90 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
server/internal/repository/node_repository.go
View File

@@ -49,7 +49,20 @@ func (r *GormNodeRepository) FindByID(ctx context.Context, id uint) (*model.Node
func (r *GormNodeRepository) FindByToken(ctx context.Context, token string) (*model.Node, error) {
var item model.Node
if err := r.db.WithContext(ctx).Where("token = ?", token).First(&item).Error; err != nil {
// 主 token 查询
err := r.db.WithContext(ctx).Where("token = ?", token).First(&item).Error
if err == nil {
return &item, nil
}
if !errors.Is(err, gorm.ErrRecordNotFound) {
return nil, err
}
// 回退：prev_token 且未过期
now := time.Now().UTC()
err = r.db.WithContext(ctx).
Where("prev_token = ? AND prev_token_expires IS NOT NULL AND prev_token_expires > ?", token, now).
First(&item).Error
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return nil, nil
}