130 lines
3.2 KiB
Go
130 lines
3.2 KiB
Go
package service
|
||
|
||
import (
|
||
"strings"
|
||
|
||
"next-terminal/server/constant"
|
||
"next-terminal/server/log"
|
||
"next-terminal/server/model"
|
||
"next-terminal/server/repository"
|
||
"next-terminal/server/utils"
|
||
)
|
||
|
||
type UserService struct {
|
||
userRepository *repository.UserRepository
|
||
loginLogRepository *repository.LoginLogRepository
|
||
}
|
||
|
||
func NewUserService(userRepository *repository.UserRepository, loginLogRepository *repository.LoginLogRepository) *UserService {
|
||
return &UserService{userRepository: userRepository, loginLogRepository: loginLogRepository}
|
||
}
|
||
|
||
func (r UserService) InitUser() (err error) {
|
||
|
||
users := r.userRepository.FindAll()
|
||
|
||
if len(users) == 0 {
|
||
initPassword := "admin"
|
||
var pass []byte
|
||
if pass, err = utils.Encoder.Encode([]byte(initPassword)); err != nil {
|
||
return err
|
||
}
|
||
|
||
user := model.User{
|
||
ID: utils.UUID(),
|
||
Username: "admin",
|
||
Password: string(pass),
|
||
Nickname: "超级管理员",
|
||
Type: constant.TypeAdmin,
|
||
Created: utils.NowJsonTime(),
|
||
}
|
||
if err := r.userRepository.Create(&user); err != nil {
|
||
return err
|
||
}
|
||
|
||
log.Infof("初始用户创建成功，账号：「%v」密码：「%v」", user.Username, initPassword)
|
||
} else {
|
||
for i := range users {
|
||
// 修正默认用户类型为管理员
|
||
if users[i].Type == "" {
|
||
user := model.User{
|
||
Type: constant.TypeAdmin,
|
||
ID: users[i].ID,
|
||
}
|
||
if err := r.userRepository.Update(&user); err != nil {
|
||
return err
|
||
}
|
||
log.Infof("自动修正用户「%v」ID「%v」类型为管理员", users[i].Nickname, users[i].ID)
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r UserService) FixUserOnlineState() error {
|
||
// 修正用户登录状态
|
||
onlineUsers, err := r.userRepository.FindOnlineUsers()
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if len(onlineUsers) > 0 {
|
||
for i := range onlineUsers {
|
||
logs, err := r.loginLogRepository.FindAliveLoginLogsByUsername(onlineUsers[i].Username)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
if len(logs) == 0 {
|
||
if err := r.userRepository.UpdateOnlineByUsername(onlineUsers[i].Username, false); err != nil {
|
||
return err
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
func (r UserService) Logout(token string) (err error) {
|
||
loginLog, err := r.loginLogRepository.FindById(token)
|
||
if err != nil {
|
||
log.Warnf("登录日志「%v」获取失败", token)
|
||
return
|
||
}
|
||
|
||
loginLogForUpdate := &model.LoginLog{LogoutTime: utils.NowJsonTime(), ID: token}
|
||
err = r.loginLogRepository.Update(loginLogForUpdate)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
loginLogs, err := r.loginLogRepository.FindAliveLoginLogsByUsername(loginLog.Username)
|
||
if err != nil {
|
||
return
|
||
}
|
||
|
||
if len(loginLogs) == 0 {
|
||
err = r.userRepository.UpdateOnlineByUsername(loginLog.Username, false)
|
||
}
|
||
return
|
||
}
|
||
|
||
func (r UserService) BuildCacheKeyByToken(token string) string {
|
||
cacheKey := strings.Join([]string{constant.Token, token}, ":")
|
||
return cacheKey
|
||
}
|
||
|
||
func (r UserService) GetTokenFormCacheKey(cacheKey string) string {
|
||
token := strings.Split(cacheKey, ":")[1]
|
||
return token
|
||
}
|
||
|
||
func (r UserService) OnEvicted(key string, value interface{}) {
|
||
if strings.HasPrefix(key, constant.Token) {
|
||
token := r.GetTokenFormCacheKey(key)
|
||
log.Debugf("用户Token「%v」过期", token)
|
||
err := r.Logout(token)
|
||
if err != nil {
|
||
log.Errorf("退出登录失败 %v", err)
|
||
}
|
||
}
|
||
}
|