87 lines
2.1 KiB
Go
87 lines
2.1 KiB
Go
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
|
|
}
|