如何使用 go-swagger 提供 websocket 服务？

有志者，事竟成！如果你在学习Golang，那么本文《如何使用 go-swagger 提供 websocket 服务？》，就很适合你！文章讲解的知识点主要包括，若是你对本文感兴趣，或者是想搞懂其中某个知识点，就请你继续往下看吧~

我正在尝试在我的 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 相当陌生，所以我正在努力解决这个问题。

我怎样才能做到这一点？

谢谢。

解决方案

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 服务？》文章吧，也可关注golang学习网公众号了解相关技术文章。