refactor: 将定时任务抽取到独立的 scheduler 包
This commit is contained in:
+5
-13
@@ -7,10 +7,9 @@ import (
|
|||||||
"kenaito-conf/server/internal/dao"
|
"kenaito-conf/server/internal/dao"
|
||||||
"kenaito-conf/server/internal/handler"
|
"kenaito-conf/server/internal/handler"
|
||||||
"kenaito-conf/server/internal/middleware"
|
"kenaito-conf/server/internal/middleware"
|
||||||
"kenaito-conf/server/internal/model"
|
"kenaito-conf/server/internal/scheduler"
|
||||||
"kenaito-conf/server/internal/service"
|
"kenaito-conf/server/internal/service"
|
||||||
"log"
|
"log"
|
||||||
"time"
|
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
)
|
)
|
||||||
@@ -128,17 +127,10 @@ func main() {
|
|||||||
// start server
|
// start server
|
||||||
addr := fmt.Sprintf(":%d", config.Cfg.Server.Port)
|
addr := fmt.Sprintf(":%d", config.Cfg.Server.Port)
|
||||||
log.Printf("Server starting on %s", addr)
|
log.Printf("Server starting on %s", addr)
|
||||||
|
|
||||||
// 启动定时任务:清理过期的客户端实例记录
|
// 启动定时任务
|
||||||
go func() {
|
scheduler.Start()
|
||||||
ticker := time.NewTicker(1 * time.Minute)
|
|
||||||
defer ticker.Stop()
|
|
||||||
for range ticker.C {
|
|
||||||
fiveMinutesAgo := time.Now().Add(-5 * time.Minute)
|
|
||||||
dao.Engine.Where("last_ping_at < ?", fiveMinutesAgo).Delete(&model.ClientInstance{})
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
|
|
||||||
if err := r.Run(addr); err != nil {
|
if err := r.Run(addr); err != nil {
|
||||||
log.Fatalf("server error: %v", err)
|
log.Fatalf("server error: %v", err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,27 @@
|
|||||||
|
package scheduler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"kenaito-conf/server/internal/dao"
|
||||||
|
"kenaito-conf/server/internal/logger"
|
||||||
|
"kenaito-conf/server/internal/model"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
func Start() {
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(1 * time.Minute)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for range ticker.C {
|
||||||
|
fiveMinutesAgo := time.Now().Add(-5 * time.Minute)
|
||||||
|
affected, err := dao.Engine.Where("last_ping_at < ?", fiveMinutesAgo).Delete(&model.ClientInstance{})
|
||||||
|
if err != nil {
|
||||||
|
logger.SugaredLogger.Errorf("[scheduler] cleanup expired client instances error: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if affected > 0 {
|
||||||
|
logger.SugaredLogger.Infof("[scheduler] cleaned up %d expired client instance(s)", affected)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
logger.SugaredLogger.Info("[scheduler] client instance cleanup task started, interval: 1m, expire: 5m")
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user