完成重构数据库操作代码

server/api/ticker.go

@@ -0,0 +1,62 @@
+package api
+
+import (
+	"github.com/sirupsen/logrus"
+	"next-terminal/server/constant"
+	"strconv"
+	"time"
+)
+
+func SetupTicker() {
+
+	// 每隔一小时删除一次未使用的会话信息
+	unUsedSessionTicker := time.NewTicker(time.Minute * 60)
+	go func() {
+		for range unUsedSessionTicker.C {
+			sessions, _ := sessionRepository.FindByStatusIn([]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 {
+						_ = sessionRepository.DeleteById(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 := propertyRepository.FindByName("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 := sessionRepository.FindOutTimeSessions(limit)
+			if err != nil {
+				return
+			}
+
+			if len(sessions) > 0 {
+				var sessionIds []string
+				for i := range sessions {
+					sessionIds = append(sessionIds, sessions[i].ID)
+				}
+				err := sessionRepository.DeleteByIds(sessionIds)
+				if err != nil {
+					logrus.Errorf("删除离线会话失败 %v", err)
+				}
+			}
+		}
+	}()
+}