package tasks
|
|
import (
|
"context"
|
"encoding/json"
|
"time"
|
"vamicro/report-service/models"
|
|
"vamicro/report-service/util"
|
|
"basic.com/valib/logger.git"
|
)
|
|
var (
|
devCtx, statCtx context.Context
|
devCancel, statCancel context.CancelFunc
|
)
|
|
func Start(chMsg chan []byte) {
|
// 启动设备信息上报
|
go deviceInfoReport()
|
|
// 启动设备状态信息上报
|
go deviceStateReport()
|
|
// 实时报警数据上报
|
go aiResultReport(chMsg)
|
}
|
|
func ResetDeviceInfoReport() {
|
devCancel()
|
}
|
|
func ResetDeviceStateReport() {
|
statCancel()
|
}
|
|
func deviceInfoReport() {
|
logger.Info("start device info report task")
|
devCtx, devCancel = context.WithCancel(context.Background())
|
|
for {
|
select {
|
case <-devCtx.Done():
|
logger.Info("deviceInfoReport exit")
|
go deviceInfoReport()
|
|
return
|
default:
|
logger.Info("deviceInfoReport")
|
|
time.Sleep(time.Duration(models.IntervalConfig.DevInfo) * time.Second)
|
}
|
}
|
}
|
|
func deviceStateReport() {
|
logger.Info("start device state report task")
|
statCtx, statCancel = context.WithCancel(context.Background())
|
|
for {
|
select {
|
case <-statCtx.Done():
|
logger.Info("deviceStateReport exit")
|
go deviceStateReport()
|
|
return
|
default:
|
logger.Info("deviceStateReport")
|
time.Sleep(time.Duration(models.IntervalConfig.DevState) * time.Second)
|
}
|
}
|
}
|
|
func aiResultReport(chMsg chan []byte) {
|
for {
|
select {
|
case msg := <-chMsg:
|
go handleRuleMsg(msg)
|
default:
|
time.Sleep(time.Millisecond * 100)
|
}
|
}
|
}
|
|
func handleRuleMsg(msg []byte) {
|
var ruleMsg = util.RuleMessage{}
|
err := json.Unmarshal(msg, &ruleMsg)
|
if err != nil {
|
logger.Warn("Unmarshal rule msg err", err)
|
return
|
}
|
|
logger.Info("Recive rule message:", ruleMsg.BaseInfo.CamName, " ", ruleMsg.BaseInfo.TaskName)
|
}
|
