04f8d0079b
* 优化图标和LOGO * 修改登录页面动画的速度为3 * 增加对websocket的异常处理 * 修复了用户组和用户名唯一判断错误的问题 * 提示版本号 * 修复readme错别字 * 修复单词拼写错误的问题 * 修复代码格式 * 修改Windows资产属性名称 * Docker 打包流程增加 upx 压缩 * 升级依赖文件，修改sqlite驱动为 github.com/glebarez/sqlite * 修复第一次查询「授权令牌」的错误 * 移除无关代码 * 修改docker打包脚本 * 增加打包脚本 * 增加微信群 * 修复单词拼写错误的问题 * 修复代码格式 * 修改Windows资产属性名称 * Docker 打包流程增加 upx 压缩 * 修改docker打包脚本 * - 替换 sqlite 驱动为 github.com/glebarez/sqlite - 修复数据库锁定的问题 - 修复部分代码不完善的问题 - 修复策略显示不完整的问题 - 修复编辑文件换行符的问题 - 优化guacd连接
82 lines
1.9 KiB
Go
82 lines
1.9 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) 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
|
|
}
|