0

我正在尝试在我的 go-swagger API 中提供聊天 websocket。我正在使用这个例子：https ://github.com/gorilla/websocket/tree/master/examples/chat

我尝试在服务器启动时将它附加到 http，但在进一步阅读后我发现这不起作用：

func configureServer(s *http.Server, scheme, addr string) {
  hub := chat.NewHub()

  go hub.Run()
  http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
    chat.ServeWs(hub, w, r)
  })
}

然后我尝试使用自定义服务器并将 server.Serve() 更改为：

hub := chat.NewHub()
go hub.Run()

http.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
  chat.ServeWs(hub, w, r)
})

if err := http.ListenAndServe(":8080", api.Serve(nil)); err != nil {
  log.Fatalln(err)
}

但也意识到为什么这不起作用。

WebSocket 握手期间出错：意外的响应代码：404

经过一番阅读，我开始理解为什么从未找到 /ws ：

hub := chat.NewHub()
go hub.Run()

mux := http.NewServeMux()
mux.HandleFunc("/ws", func(w http.ResponseWriter, r *http.Request) {
  chat.ServeWs(hub, w, r)
})

server.SetHandler(mux)
if err := server.Serve(); err != nil {
  log.Fatalln(err)
}

现在 websockets 工作但其余 api 没有，我可以明白为什么但我不知道如何让它们都工作。我对golang还很陌生，所以我正在为此苦苦挣扎。

我怎么能那样做？

谢谢。

4

1 回答 1

0

casualjim 在 Github 上帮助了我：

你可以把它放在一个中间件中，然后把它挂在 configure_XXX

func mountWebSockets(next http.Handler) http.HandlerFunc {
   return func(w http.ResponseWriter, r *http.Request) {
       if r.URL.Path == "/ws" {
            chat.ServeWs(hub, w, r)
           return
       }
       next.ServeHTTP(w, r)
   }
}

来源：https ://github.com/go-swagger/go-swagger/issues/2083#issuecomment-542764897

于 2019-10-17T10:54:22.247 回答