暂停聚集模型

zhaoqingang

2025-02-19 fca319958029fa924308e50cb61202d7d6ff5008

 service/task_results.go

@@ -1,7 +1,28 @@
package service

import "model-engine/db"
import (
   "model-engine/db"
   "model-engine/pkg/logger"
)

func SaveTaskResults(results []*db.ModelTaskResults) error {
   return db.NewModelTaskResultsSearch().BatchCreate(results)
   for _, v := range results {
      err := db.NewModelTaskResultsSearch().SetTaskId(v.ModelTaskID).SetPicDate(v.PicDate).SetFirstPersonId(v.FirstPersonID).FirstOrCreate(v)
      if err != nil {
         logger.Errorf("save task result err:%v", err)
         return err
      }
   }
   return nil
}

func SaveLocationTaskResults(results []*db.ModelTaskResults) error {
   for _, v := range results {
      err := db.NewModelTaskResultsSearch().SetTaskId(v.ModelTaskID).SetFirstLocation(v.Location).FirstOrCreate(v)
      if err != nil {
         logger.Errorf("save task result err:%v", err)
         return err
      }
   }
   return nil
}