package models
|
|
import (
|
"fmt"
|
"time"
|
|
"model-engine/db"
|
"model-engine/pkg/logger"
|
"model-engine/service"
|
)
|
|
type LocationModel struct {
|
AreaIds map[string]struct{}
|
Building string // 楼栋
|
AlarmType db.AlarmType // 预警方式
|
KeyPersonType string // 人员类型
|
PersonLabel string
|
PersonCount int // 人数, 出现的同类型人员数量
|
Appearances int // 出现次数,
|
Duration int // 时间范围, 单位天
|
Task *db.ModelTask
|
}
|
|
func (m *LocationModel) Init(task *db.ModelTask) error {
|
m.AreaIds = make(map[string]struct{})
|
for _, a := range task.DomainUnitIds {
|
m.AreaIds[a] = struct{}{}
|
}
|
|
m.Task = task
|
m.Building = task.Building
|
m.AlarmType = task.AlarmType
|
m.KeyPersonType = task.PersonType
|
m.PersonLabel = task.PersonLabel
|
|
for _, v := range task.Rules {
|
if v.Alias == "personCount" {
|
if val, ok := v.Value.(float64); ok {
|
m.PersonCount = int(val)
|
}
|
}
|
|
if v.Alias == "appearances" {
|
if val, ok := v.Value.(float64); ok {
|
m.Appearances = int(val)
|
}
|
}
|
|
if v.Alias == "duration" {
|
if val, ok := v.Value.(float64); ok {
|
m.Duration = int(val)
|
}
|
}
|
}
|
|
// 默认计算30天的数据
|
if m.Duration == 0 {
|
m.Duration = 30
|
}
|
|
logger.Debugf("LocationModel init finish ...task id:%s, name:%s, rule:%+v", task.ID, task.Name, m)
|
|
return nil
|
}
|
|
func (m *LocationModel) Run() error {
|
// 根据配置的时间段天数, 过滤规则配置的相同重点人员类型和相同标签人员频繁出现的楼层地址
|
|
results := make([]*db.ModelTaskResults, 0)
|
|
logger.Debugf("task %s last filter result %d", m.Task.Name, len(results))
|
return service.SaveTaskResults(results)
|
}
|
|
func (m *LocationModel) KeepAlive() error {
|
db.GetDB().Model(m.Task).Where("id = ?", m.Task.ID).Update("last_run_time", time.Now())
|
return nil
|
}
|
|
func (m *LocationModel) Shutdown() error {
|
// 清理资源
|
fmt.Println("Shutting down LocationModel Model")
|
return nil
|
}
|
|
func (m *LocationModel) eventFormat() string {
|
return ""
|
}
|
