From 16b34c939d6790fe6ebe61f8f0e85fe80ee52224 Mon Sep 17 00:00:00 2001 From: sunty <1172534965@qq.com> Date: 星期三, 29 五月 2024 14:44:16 +0800 Subject: [PATCH] 修改准备数据流程,适应较大数据量的操作 --- task/engine.go | 35 +++++++++++++++++++++++++++++------ 1 files changed, 29 insertions(+), 6 deletions(-) diff --git a/task/engine.go b/task/engine.go index 9ec9330..53c59c2 100644 --- a/task/engine.go +++ b/task/engine.go @@ -4,9 +4,13 @@ "basic.com/valib/logger.git" "ruleModelEngine/cache" "ruleModelEngine/db" + "time" ) func UnderageClimbingToRooftop(taskPerson db.TaskPerson) bool { + if taskPerson.Floor == "" { + return false + } age, err := db.GetAgeById(taskPerson.DocumentNumber) if err != nil { logger.Error("GetAgeById: ", err) @@ -30,14 +34,33 @@ } return false } -func ClimbingFloorsIllegally(taskPerson db.TaskPerson) bool { - - return true +func ClimbingFloorsIllegally(taskPerson db.TaskPerson, hours int, times int) bool { + layout := "2006-01-02 15:04:05" + picDate, err := time.Parse(layout, taskPerson.PicDate) + startTime := picDate.Add(time.Duration(-hours) * time.Hour).Format(layout) + floors, err := db.QueryCheckDataByDocumentNumber(taskPerson.DocumentNumber, startTime, taskPerson.PicDate) + if err != nil { + logger.Error("QueryCheckDataByDocumentNumber: ", err) + return false + } + if floors >= times { + return true + } + return false } -func EnteringButNotLeaving(taskPerson db.TaskPerson) bool { - return true +func EnteringButNotLeaving(docNumber string, days int) bool { + total, err := db.QueryTimesByDocNumberDays(days-1, docNumber) + if err != nil { + logger.Error("QueryTimesByDocNumberDays err: ", err) + } + if total == 0 { + return true + } + + return false } + func registeredAddressNotMatchActualResidence(taskPerson db.TaskPerson) bool { - return true + return false } -- Gitblit v1.8.0