修复 「1.2.2 用户管理-用户列表勾选单一用户会全选 」 close #216

server/service/access_token.go

@@ -0,0 +1,86 @@
+package service
+
+import (
+	"context"
+	"errors"
+
+	"next-terminal/server/constant"
+	"next-terminal/server/dto"
+	"next-terminal/server/env"
+	"next-terminal/server/global/cache"
+	"next-terminal/server/model"
+	"next-terminal/server/repository"
+	"next-terminal/server/utils"
+
+	"gorm.io/gorm"
+)
+
+type accessTokenService struct {
+	baseService
+}
+
+func (service accessTokenService) FindByUserId(userId string) (model.AccessToken, error) {
+	return repository.AccessTokenRepository.FindByUserId(context.TODO(), userId)
+}
+
+func (service accessTokenService) GenAccessToken(userId string) error {
+	return env.GetDB().Transaction(func(tx *gorm.DB) error {
+		ctx := service.Context(tx)
+
+		user, err := repository.UserRepository.FindById(ctx, userId)
+		if err != nil {
+			return err
+		}
+		oldAccessToken, err := repository.AccessTokenRepository.FindByUserId(ctx, userId)
+		if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
+			return err
+		}
+		if oldAccessToken.Token != "" {
+			cache.TokenManager.Delete(oldAccessToken.Token)
+		}
+		if err := repository.AccessTokenRepository.DeleteByUserId(ctx, userId); err != nil {
+			return err
+		}
+
+		token := "forever-" + utils.UUID()
+		accessToken := &model.AccessToken{
+			ID:      utils.UUID(),
+			UserId:  userId,
+			Token:   token,
+			Created: utils.NowJsonTime(),
+		}
+
+		authorization := dto.Authorization{
+			Token:    token,
+			Remember: false,
+			Type:     constant.AccessToken,
+			User:     &user,
+		}
+
+		cache.TokenManager.Set(token, authorization, cache.NoExpiration)
+
+		return repository.AccessTokenRepository.Create(ctx, accessToken)
+	})
+}
+
+func (service accessTokenService) Reload() error {
+	accessTokens, err := repository.AccessTokenRepository.FindAll(context.TODO())
+	if err != nil {
+		return err
+	}
+	for _, accessToken := range accessTokens {
+		user, err := repository.UserRepository.FindById(context.TODO(), accessToken.UserId)
+		if err != nil {
+			return err
+		}
+		authorization := dto.Authorization{
+			Token:    accessToken.Token,
+			Remember: false,
+			Type:     constant.AccessToken,
+			User:     &user,
+		}
+
+		cache.TokenManager.Set(accessToken.Token, authorization, cache.NoExpiration)
+	}
+	return nil
+}