背景

有一个需求是这样的，前端需要通过http请求的form-data上传图片文件，后端接收图片后调用AI接口执行命令，由于命令执行时间较长，需要持续返回当前任务在全局任务列表中的位置，以便前端即时更新排队信息。

思考

如果直接在gin的请求处理函数中开goroutine，并在goroutine中通过类似c.JSON的方法来返回响应，前端会无法收到响应。原因是当处理函数返回时，Gin 会自动关闭 HTTP 请求的响应，这意味着在处理函数返回后，后台的协程无法再向响应中写入数据，从而导致客户端收不到信息。（不用goroutine也不行，因为持续返回响应涉及到使用for循环，会阻塞通道，阻塞主协程，仍然无法返回响应），使用gin配合websocket解决。

于是解决方案就是，使用gin的处理函数接收图片后，把当前任务的id作为响应返回给前端。然后前端再用当前任务的id建立websocket请求，在websocket请求中，通常不会自动关闭，除非显式地由客户端或服务器关闭，所以就可以在ws的处理函数中开goroutine来持续返回响应。

func WsHandleConnect(s *melody.Session) {TaskId, _ := s.Get("task_id")// 将 TaskId 转换为字符串并解析为整数taskIdStr, _ := TaskId.(string)taskId, err := strconv.Atoi(taskIdStr)if err != nil {sendJSONResponse(s, Res{Code: 4000,Msg:  "Invalid Task ID",Data: nil,})s.Close()return}var task *model.Task// 遍历 TaskQueue 查找对应的任务for _, t := range global.TaskQueue {if t.ID == taskId {task = tbreak}}if task == nil {sendJSONResponse(s, Res{Code: 4000,Msg:  "Task ID not provided",Data: nil,})s.Close()return}logger.Log.Error("task:", task.ID)go func() {for {select {case result := <-task.Response:sendJSONResponse(s, Res{Code: global.SuccessCode,Msg:  result,Data: nil,})s.Close()returndefault:time.Sleep(3 * time.Second)if len(global.TaskQueue) == 0 {sendJSONResponse(s, Res{Code: global.SuccessCode,Msg:  "Task failed",Data: nil,})s.Close()return}position := service.FindTaskPosition(task.ID)sendJSONResponse(s, Res{Code: global.SuccessCode,Data: position,})}}}()
}