任务状态读取

routers/images.go

@@ -40,13 +40,28 @@ func ImagesGet(w http.ResponseWriter, r *http.Request) {
 		}
 		defer conn.Close()
 
+		// 读取任务信息
 		task := r.URL.Query().Get("task")
 		if task == "" {
 			w.WriteHeader(http.StatusBadRequest)
 			w.Write([]byte("task 参数不能为空"))
 			return
 		}
+		// 从数据库中读取任务信息
+		var image_list []models.Image
+		if err := configs.ORMDB().Where("task = ?", task).Find(&image_list).Error; err != nil {
+			log.Println(err)
+			return
+		}
+		if len(image_list) == 0 {
+			w.WriteHeader(http.StatusBadRequest)
+			w.Write([]byte("任务不存在或已结束"))
+			return
+		}
 
+		log.Println("任务编号:", task, "任务数量:", len(image_list))
+
+		// 加入连接池
 		wsid := images_websocket_manager.AddConnection(conn)
 		defer images_websocket_manager.RemoveConnection(wsid)