如何使用 go-swagger 服务 websocket？

Question

我正在尝试在我的 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)
}

但也意识到为什么这行不通。

Error during WebSocket handshake: Unexpected response code: 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 还很陌生，所以我正在为此苦苦挣扎。

我该怎么做？

谢谢。

Answer 1

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

如何使用 go-swagger 服务 websocket？

How to serve websocket with go-swagger?

go

gorilla

go-swagger