105 lines
2.6 KiB
Go
105 lines
2.6 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
|
|
"next-terminal/server/common"
|
|
"next-terminal/server/common/nt"
|
|
"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"
|
|
)
|
|
|
|
var AccessTokenService = new(accessTokenService)
|
|
|
|
type accessTokenService struct {
|
|
baseService
|
|
}
|
|
|
|
func (service accessTokenService) GenAccessToken(userId string, tokenType string) (*model.AccessToken, error) {
|
|
var accessToken *model.AccessToken
|
|
err := env.GetDB().Transaction(func(tx *gorm.DB) error {
|
|
ctx := service.Context(tx)
|
|
|
|
user, err := repository.UserRepository.FindById(ctx, userId)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
token := "forever-" + utils.UUID()
|
|
accessToken = &model.AccessToken{
|
|
ID: utils.UUID(),
|
|
UserId: userId,
|
|
Token: token,
|
|
Type: tokenType,
|
|
Created: common.NowJsonTime(),
|
|
}
|
|
|
|
authorization := dto.Authorization{
|
|
Token: token,
|
|
Remember: false,
|
|
Type: nt.AccessToken,
|
|
User: &user,
|
|
}
|
|
|
|
cache.TokenManager.Set(token, authorization, cache.NoExpiration)
|
|
|
|
return repository.AccessTokenRepository.Create(ctx, accessToken)
|
|
})
|
|
return accessToken, err
|
|
}
|
|
|
|
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: nt.AccessToken,
|
|
User: &user,
|
|
}
|
|
|
|
cache.TokenManager.Set(accessToken.Token, authorization, cache.NoExpiration)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (service accessTokenService) DelAccessToken(ctx context.Context, id string, userId string) error {
|
|
oldAccessTokens, err := repository.AccessTokenRepository.FindByUserId(ctx, userId)
|
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return err
|
|
}
|
|
for _, t := range oldAccessTokens {
|
|
if t.Token != "" {
|
|
cache.TokenManager.Delete(t.Token)
|
|
}
|
|
}
|
|
return repository.AccessTokenRepository.DeleteById(ctx, id)
|
|
}
|
|
|
|
func (service accessTokenService) DelAccessTokenByUserId(ctx context.Context, userId string) error {
|
|
oldAccessTokens, err := repository.AccessTokenRepository.FindByUserId(ctx, userId)
|
|
if err != nil && !errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return err
|
|
}
|
|
for _, t := range oldAccessTokens {
|
|
if t.Token != "" {
|
|
cache.TokenManager.Delete(t.Token)
|
|
}
|
|
}
|
|
return repository.AccessTokenRepository.DeleteByUserId(ctx, userId)
|
|
}
|