From 6b59a711b9af0825858c408cdba95102b8b51cb3 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期五, 13 十二月 2024 15:46:44 +0800
Subject: [PATCH] 结果表增加联合唯一索引,防止重复报警

---
 service/task.go |   19 ++++++++++++++++++-
 1 files changed, 18 insertions(+), 1 deletions(-)

diff --git a/service/task.go b/service/task.go
index b0afcfb..449970a 100644
--- a/service/task.go
+++ b/service/task.go
@@ -3,5 +3,22 @@
 import "model-engine/db"
 
 func GetTasks() (tasks []*db.ModelTask, err error) {
-	return db.NewModelTaskSearch().FindAll()
+	models, err := db.NewModelSearch().SetEnabled(true).FindAll()
+	if err != nil {
+		return nil, err
+	}
+	if len(models) == 0 {
+		return nil, nil
+	}
+
+	modelIds := make([]string, 0, len(models))
+	for _, model := range models {
+		modelIds = append(modelIds, model.ID)
+	}
+
+	return db.NewModelTaskSearch().
+		SetModelIDs(modelIds).
+		SetEnabled(true).
+		SetUnexpired().
+		FindAll()
 }

--
Gitblit v1.8.0