完成重构数据库操作代码

server/service/user.go

@@ -0,0 +1,104 @@
+package service
+
+import (
+	"github.com/sirupsen/logrus"
+	"next-terminal/server/constant"
+	"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) *UserService {
+	return &UserService{userRepository: userRepository}
+}
+
+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
+		}
+		logrus.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
+				}
+				logrus.Infof("自动修正用户「%v」ID「%v」类型为管理员", users[i].Nickname, users[i].ID)
+			}
+		}
+	}
+	return nil
+}
+
+func (r UserService) FixedOnlineState() error {
+	// 修正用户登录状态
+	onlineUsers, err := r.userRepository.FindOnlineUsers()
+	if err != nil {
+		return err
+	}
+	for i := range onlineUsers {
+		logs, err := r.loginLogRepository.FindAliveLoginLogsByUserId(onlineUsers[i].ID)
+		if err != nil {
+			return err
+		}
+		if len(logs) == 0 {
+			if err := r.userRepository.UpdateOnline(onlineUsers[i].ID, false); err != nil {
+				return err
+			}
+		}
+	}
+	return nil
+}
+
+func (r UserService) Logout(token string) (err error) {
+
+	loginLog, err := r.loginLogRepository.FindById(token)
+	if err != nil {
+		logrus.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.FindAliveLoginLogsByUserId(loginLog.UserId)
+	if err != nil {
+		return
+	}
+
+	if len(loginLogs) == 0 {
+		err = r.userRepository.UpdateOnline(loginLog.UserId, false)
+	}
+	return
+}