release v1.2.0

server/service/user.go

@@ -0,0 +1,129 @@
+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)
+		}
+	}
+}