smart_intersectionV2.0/server/initialize/timer.go

package initialize

import (
	"fmt"
	"github.com/robfig/cron/v3"
	"server/dao"
	"server/service/devices"
	"server/task"
	"server/utils/protocol"
	"time"

	"server/global"
)

func Timer() {
	go func() {
		var option []cron.Option
		option = append(option, cron.WithSeconds())
		// 清理DB定时任务
		_, err := global.GVA_Timer.AddTaskByFunc("ClearDB", "@daily", func() {
			err := task.ClearTable(global.GVA_DB) // 定时任务方法定在task文件包中
			if err != nil {
				fmt.Println("timer error:", err)
			}
		}, "定时清理数据库【日志，黑名单】内容", option...)
		if err != nil {
			fmt.Println("add timer error:", err)
		}

		// 其他定时任务定在这里 参考上方使用方法

		//_, err := global.GVA_Timer.AddTaskByFunc("定时任务标识", "corn表达式", func() {
		//	具体执行内容...
		//  ......
		//}, option...)
		//if err != nil {
		//	fmt.Println("add timer error:", err)
		//}
	}()

	go IsOnline() //监测在线状态

}

func IsOnline() {
	t := time.NewTicker(1 * time.Minute) //每分钟
	for {
		select {
		case <-t.C:
			for sn, device := range devices.DeviceOnlineTimes {
				//符合条件
				if time.Now().Add(-2 * time.Minute).After(*device) {
					err := dao.UpdateScreensStatusBySn(sn, 0)
					if err != nil {
						global.GVA_LOG.Error(fmt.Sprintf("修改屏幕状态失败:%s", err.Error()))
					}
					topic := devices.MqttService.GetTopic(sn, protocol.TopicOffline)
					_ = devices.MqttService.Publish(topic, sn)
					delete(devices.DeviceOnlineTimes, sn)
					global.GVA_LOG.Info(fmt.Sprintf("设备【sn】%v 离线了", sn))
				}
			}
		}
	}
}