From 3c033759200ad7c02dd59521b1aebbbdc35b98fa Mon Sep 17 00:00:00 2001
From: zhaoqingang <zhaoqg0118@163.com>
Date: 星期六, 08 二月 2025 16:18:50 +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