From 39d53cf7a2080b49224dec6c395fc4ed2463a424 Mon Sep 17 00:00:00 2001 From: zhangzengfei <zhangzengfei@smartai.com> Date: 星期三, 18 十二月 2024 18:23:20 +0800 Subject: [PATCH] 修改参数的获取逻辑 --- cron/cron.go | 10 +++++++--- 1 files changed, 7 insertions(+), 3 deletions(-) diff --git a/cron/cron.go b/cron/cron.go index b77c86e..551c6f2 100644 --- a/cron/cron.go +++ b/cron/cron.go @@ -1,12 +1,14 @@ package cron import ( + "time" + "github.com/go-co-op/gocron" + "model-engine/models" "model-engine/pkg/logger" "model-engine/pkg/safe" "model-engine/service" - "time" ) var s *gocron.Scheduler @@ -20,15 +22,17 @@ if err != nil { panic(err) } + for _, task := range tasks { model, err := models.GetModel(task.ModelID) if err != nil { logger.Errorf("can not find model for id:%v", task.ModelID) continue } - task := task + + t := task safe.Go(func() { - if err := model.Init(task); err != nil { + if err := model.Init(t); err != nil { return } if err := model.Run(); err != nil { -- Gitblit v1.8.0