🐶 重构部分用户数据库操作代码

server/handle/runner.go

@@ -0,0 +1,303 @@
+package handle
+
+import (
+	"os"
+	"strconv"
+	"time"
+
+	"next-terminal/server/constant"
+	"next-terminal/server/guacd"
+	"next-terminal/server/model"
+	"next-terminal/server/utils"
+
+	"github.com/sirupsen/logrus"
+)
+
+func RunTicker() {
+
+	// 每隔一小时删除一次未使用的会话信息
+	unUsedSessionTicker := time.NewTicker(time.Minute * 60)
+	go func() {
+		for range unUsedSessionTicker.C {
+			sessions, _ := model.FindSessionByStatusIn([]string{constant.NoConnect, constant.Connecting})
+			if len(sessions) > 0 {
+				now := time.Now()
+				for i := range sessions {
+					if now.Sub(sessions[i].ConnectedTime.Time) > time.Hour*1 {
+						_ = model.DeleteSessionById(sessions[i].ID)
+						s := sessions[i].Username + "@" + sessions[i].IP + ":" + strconv.Itoa(sessions[i].Port)
+						logrus.Infof("会话「%v」ID「%v」超过1小时未打开，已删除。", s, sessions[i].ID)
+					}
+				}
+			}
+		}
+	}()
+
+	// 每日凌晨删除超过时长限制的会话
+	timeoutSessionTicker := time.NewTicker(time.Hour * 24)
+	go func() {
+		for range timeoutSessionTicker.C {
+			property, err := model.FindPropertyByName("session-saved-limit")
+			if err != nil {
+				return
+			}
+			if property.Value == "" || property.Value == "-" {
+				return
+			}
+			limit, err := strconv.Atoi(property.Value)
+			if err != nil {
+				return
+			}
+			sessions, err := model.FindOutTimeSessions(limit)
+			if err != nil {
+				return
+			}
+
+			if len(sessions) > 0 {
+				var sessionIds []string
+				for i := range sessions {
+					sessionIds = append(sessionIds, sessions[i].ID)
+				}
+				err := model.DeleteSessionByIds(sessionIds)
+				if err != nil {
+					logrus.Errorf("删除离线会话失败 %v", err)
+				}
+			}
+		}
+	}()
+}
+
+func RunDataFix() {
+	sessions, _ := model.FindSessionByStatus(constant.Connected)
+	if sessions == nil {
+		return
+	}
+
+	for i := range sessions {
+		session := model.Session{
+			Status:           constant.Disconnected,
+			DisconnectedTime: utils.NowJsonTime(),
+		}
+
+		_ = model.UpdateSessionById(&session, sessions[i].ID)
+	}
+}
+
+func InitProperties() error {
+	propertyMap := model.FindAllPropertiesMap()
+
+	if len(propertyMap[guacd.Host]) == 0 {
+		property := model.Property{
+			Name:  guacd.Host,
+			Value: "127.0.0.1",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.Port]) == 0 {
+		property := model.Property{
+			Name:  guacd.Port,
+			Value: "4822",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableRecording]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableRecording,
+			Value: "true",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.RecordingPath]) == 0 {
+		path, _ := os.Getwd()
+		property := model.Property{
+			Name:  guacd.RecordingPath,
+			Value: path + "/recording/",
+		}
+		if !utils.FileExists(property.Value) {
+			if err := os.Mkdir(property.Value, os.ModePerm); err != nil {
+				return err
+			}
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.CreateRecordingPath]) == 0 {
+		property := model.Property{
+			Name:  guacd.CreateRecordingPath,
+			Value: "true",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.DriveName]) == 0 {
+		property := model.Property{
+			Name:  guacd.DriveName,
+			Value: "File-System",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.DrivePath]) == 0 {
+
+		path, _ := os.Getwd()
+
+		property := model.Property{
+			Name:  guacd.DrivePath,
+			Value: path + "/drive/",
+		}
+		if !utils.FileExists(property.Value) {
+			if err := os.Mkdir(property.Value, os.ModePerm); err != nil {
+				return err
+			}
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.FontName]) == 0 {
+		property := model.Property{
+			Name:  guacd.FontName,
+			Value: "menlo",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.FontSize]) == 0 {
+		property := model.Property{
+			Name:  guacd.FontSize,
+			Value: "12",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.ColorScheme]) == 0 {
+		property := model.Property{
+			Name:  guacd.ColorScheme,
+			Value: "gray-black",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableDrive]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableDrive,
+			Value: "true",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableWallpaper]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableWallpaper,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableTheming]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableTheming,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableFontSmoothing]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableFontSmoothing,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableFullWindowDrag]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableFullWindowDrag,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableDesktopComposition]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableDesktopComposition,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.EnableMenuAnimations]) == 0 {
+		property := model.Property{
+			Name:  guacd.EnableMenuAnimations,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.DisableBitmapCaching]) == 0 {
+		property := model.Property{
+			Name:  guacd.DisableBitmapCaching,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.DisableOffscreenCaching]) == 0 {
+		property := model.Property{
+			Name:  guacd.DisableOffscreenCaching,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+
+	if len(propertyMap[guacd.DisableGlyphCaching]) == 0 {
+		property := model.Property{
+			Name:  guacd.DisableGlyphCaching,
+			Value: "false",
+		}
+		if err := model.CreateNewProperty(&property); err != nil {
+			return err
+		}
+	}
+	return nil
+}