diff --git a/main.go b/main.go index 1a67da2..0c26844 100644 --- a/main.go +++ b/main.go @@ -36,7 +36,7 @@ var DequeueHandler = triggered.RegisterExecute(func(params *triggered.Params[boo return } - log.Println(item) + // log.Println(item) PopChannel <- item } }) diff --git a/websocket.go b/websocket.go index ee28295..496f32b 100644 --- a/websocket.go +++ b/websocket.go @@ -68,6 +68,14 @@ func queueHandler(w http.ResponseWriter, r *http.Request) { } } + err = conn.SetWriteDeadline(time.Now().Add(time.Second * 15)) + if err != nil { + log.Println(err) + log.Println("重新回队") + stateClient.PushItem(nil, item) + return + } + // 写回消息 err = conn.WriteMessage(websocket.BinaryMessage, wdata) if err != nil { @@ -78,15 +86,6 @@ func queueHandler(w http.ResponseWriter, r *http.Request) { } // 打印消息 fmt.Printf("%s 处理完成. len\n", conn.RemoteAddr()) - - // 读取消息 - // msgType, msg, err := conn.ReadMessage() - - // if err != nil { - // log.Println(msgType, msg, err) - // return - // } - } }