go - 如何为wosagger服务websocket？

Question

我正在尝试在我的可笑API中提供聊天网络套接字。我正在使用以下示例: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还是很陌生，所以我为此苦苦挣扎。

我该怎么办？

谢谢。

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