zhangzengfei
2025-02-12 f99b8a6d2ec0fc2414b939c3fef4d9338d13fce4
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
}