fix: dict as raw data instead of slice

apache · Feb 20, 2025 · ef0abe7 · ef0abe7
1 parent 620636f
commit ef0abe7
Showing 1 changed file with 22 additions and 27 deletions.
diff --git a/backend/plugins/zentao/tasks/task_worklog_extractor.go b/backend/plugins/zentao/tasks/task_worklog_extractor.go
@@ -44,7 +44,7 @@ func ExtractTaskWorklogs(taskCtx plugin.SubTaskContext) errors.Error {
 			Table:   RAW_TASK_WORKLOGS_TABLE,
 		},
 		Extract: func(row *api.RawData) ([]interface{}, errors.Error) {
-			var worklogs []struct {
+			var input struct {
 				Id         uint64  `json:"id"`
 				ObjectType string  `json:"objectType"`
 				ObjectId   uint64  `json:"objectID"`
@@ -64,36 +64,31 @@ func ExtractTaskWorklogs(taskCtx plugin.SubTaskContext) errors.Error {
 				Deleted    string  `json:"deleted"`
 			}
 
-			err := errors.Convert(json.Unmarshal(row.Data, &worklogs))
+			err := errors.Convert(json.Unmarshal(row.Data, &input))
 			if err != nil {
 				return nil, err
 			}
-			results := make([]interface{}, 0, len(worklogs))
-			for _, effort := range worklogs {
-				worklog := &models.ZentaoWorklog{
-					ConnectionId: data.Options.ConnectionId,
-					Id:           effort.Id,
-					ObjectId:     effort.ObjectId,
-					ObjectType:   effort.ObjectType,
-					Project:      effort.Project,
-					Execution:    effort.Execution,
-					Product:      effort.Product,
-					Account:      effort.Account,
-					Work:         effort.Work,
-					Vision:       effort.Vision,
-					Date:         effort.Date,
-					Left:         effort.Left,
-					Consumed:     effort.Consumed,
-					Begin:        effort.Begin,
-					End:          effort.End,
-					Extra:        effort.Extra,
-					Order:        effort.Order,
-					Deleted:      effort.Deleted,
-				}
-				results = append(results, worklog)
+			worklog := &models.ZentaoWorklog{
+				ConnectionId: data.Options.ConnectionId,
+				Id:           input.Id,
+				ObjectId:     input.ObjectId,
+				ObjectType:   input.ObjectType,
+				Project:      input.Project,
+				Execution:    input.Execution,
+				Product:      input.Product,
+				Account:      input.Account,
+				Work:         input.Work,
+				Vision:       input.Vision,
+				Date:         input.Date,
+				Left:         input.Left,
+				Consumed:     input.Consumed,
+				Begin:        input.Begin,
+				End:          input.End,
+				Extra:        input.Extra,
+				Order:        input.Order,
+				Deleted:      input.Deleted,
 			}
-
-			return results, nil
+			return []interface{}{worklog}, nil
 		},
 	})