From d96b150b7da20f8ed727104d0393b0be6743781e Mon Sep 17 00:00:00 2001
From: tzwang <wtz1003@sina.com>
Date: Sat, 22 Jun 2024 22:32:38 +0800
Subject: [PATCH] fix aitask bugs

Former-commit-id: be342bbe8f3cd5160f76445727a66720c9ba6230
---
 api/internal/cron/aiCronTask.go | 24 ++++++++++++++++++++----
 1 file changed, 20 insertions(+), 4 deletions(-)

diff --git a/api/internal/cron/aiCronTask.go b/api/internal/cron/aiCronTask.go
index 619c51e7..eb295311 100644
--- a/api/internal/cron/aiCronTask.go
+++ b/api/internal/cron/aiCronTask.go
@@ -225,6 +225,7 @@ func UpdateTaskStatus(svc *svc.ServiceContext, tasklist []*types.TaskModel) {
 
 	var start time.Time
 	var end time.Time
+	// distinguish train or infer temporarily
 	if task.TaskTypeDict == 11 {
 		start, _ = time.ParseInLocation(time.RFC3339, aiTask[0].StartTime, time.Local)
 		end, _ = time.ParseInLocation(time.RFC3339, aiTask[0].EndTime, time.Local)
@@ -236,8 +237,16 @@ func UpdateTaskStatus(svc *svc.ServiceContext, tasklist []*types.TaskModel) {
 	var status string
 	var count int
 	for _, a := range aiTask {
-		s, _ := time.ParseInLocation(constants.Layout, a.StartTime, time.Local)
-		e, _ := time.ParseInLocation(constants.Layout, a.EndTime, time.Local)
+		var s time.Time
+		var e time.Time
+		// distinguish train or infer temporarily
+		if task.TaskTypeDict == 11 {
+			s, _ = time.ParseInLocation(time.RFC3339, a.StartTime, time.Local)
+			e, _ = time.ParseInLocation(time.RFC3339, a.EndTime, time.Local)
+		} else {
+			s, _ = time.ParseInLocation(constants.Layout, a.StartTime, time.Local)
+			e, _ = time.ParseInLocation(constants.Layout, a.EndTime, time.Local)
+		}
 
 		if s.Before(start) {
 			start = s
@@ -274,8 +283,15 @@ func UpdateTaskStatus(svc *svc.ServiceContext, tasklist []*types.TaskModel) {
 
 	if status != "" {
 		task.Status = status
-		task.StartTime = start.Format(constants.Layout)
-		task.EndTime = end.Format(constants.Layout)
+		// distinguish train or infer temporarily
+		if task.TaskTypeDict == 11 {
+			task.StartTime = start.Format(time.RFC3339)
+			task.EndTime = end.Format(time.RFC3339)
+		} else {
+			task.StartTime = start.Format(constants.Layout)
+			task.EndTime = end.Format(constants.Layout)
+		}
+
 	}
 
 	task.UpdatedTime = time.Now().Format(constants.Layout)