modify task type

controllers/task.go

@@ -2035,6 +2035,15 @@ func (server *TaskController) Enumeration() {
 	server.respond(http.StatusOK, "", responseData)
 }
 
+func taskTypeCount() int64 {
+	qs := mysql.GetMysqlInstace().Ormer.QueryTable("task_type")
+	count, err := qs.Count()
+	if err != nil {
+		return 0
+	}
+	return count
+}
+
 func initTypeInRedis() []models.TaskHeat {
 	qs := mysql.GetMysqlInstace().Ormer.QueryTable("task_type")
 	count, _ := qs.Count()
@@ -2074,19 +2083,22 @@ func initTypeInRedis() []models.TaskHeat {
 
 func (server *TaskController) TaskHeat() {
 	data, err := redis.GetDataToString(cronjob.HeatKey)
-	if data == "" {
+	if data == "" || err != nil {
 		response := initTypeInRedis()
 		server.respond(http.StatusOK, "", response)
 		return
 	}
 
 	var response []models.TaskHeat
+	err = json.Unmarshal([]byte(data), &response)
 	if err != nil {
+		response = initTypeInRedis()
 		server.respond(http.StatusOK, "", response)
 		return
 	}
-	err = json.Unmarshal([]byte(data), &response)
-	if err != nil {
+	count := taskTypeCount()
+	if len(response) < int(count) {
+		response = initTypeInRedis()
 		server.respond(http.StatusOK, "", response)
 		return
 	}

lastupdate.tmp

-{"/Users/brent/Documents/wubanWork/ai_developer_admin/controllers":1709549304433646107}
+{"/Users/brent/Documents/wubanWork/ai_developer_admin/controllers":1709610847935101949}
\ No newline at end of file