refactor: 将定时任务抽取到独立的 scheduler 包
This commit is contained in:
+5
-13
@@ -7,10 +7,9 @@ import (
|
||||
"kenaito-conf/server/internal/dao"
|
||||
"kenaito-conf/server/internal/handler"
|
||||
"kenaito-conf/server/internal/middleware"
|
||||
"kenaito-conf/server/internal/model"
|
||||
"kenaito-conf/server/internal/scheduler"
|
||||
"kenaito-conf/server/internal/service"
|
||||
"log"
|
||||
"time"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
)
|
||||
@@ -128,17 +127,10 @@ func main() {
|
||||
// start server
|
||||
addr := fmt.Sprintf(":%d", config.Cfg.Server.Port)
|
||||
log.Printf("Server starting on %s", addr)
|
||||
|
||||
// 启动定时任务:清理过期的客户端实例记录
|
||||
go func() {
|
||||
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{})
|
||||
}
|
||||
}()
|
||||
|
||||
|
||||
// 启动定时任务
|
||||
scheduler.Start()
|
||||
|
||||
if err := r.Run(addr); err != nil {
|
||||
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