在同一端口上使用 REST API 并利用 grpc-web 进行流式传输的方法

利用 cmux 在同一端口上运行 REST API 和 grpc-web 流式传输时，通常会遇到浏览器重复使用连接导致多路复用失败的问题。为了解决这个问题，可以通过在所有传入的 HTTP 请求上注册一个迷你 HTTP 服务器处理程序，并根据请求标头显式调用处理程序来避免 cmux 的默认多路复用机制。这样，浏览器可以同时进行流式传输和调用 REST API，而不会影响多路复用。

我有一个用 Go 编写的服务器，它使用 cmux 允许多个协议在同一端口上运行，但自述文件中指定了一个限制

cmux 在接受时匹配连接。例如，一个连接可以是 gRPC 或 REST，但不能同时是两者。也就是说，我们假设客户端连接用于 gRPC 或 REST。

我需要浏览器能够从 grpc-web 进行流式传输并在同一端口上调用 REST API，但浏览器重复使用相同的连接并导致多路复用无法工作。

解决方案

这是一个很难识别的问题。由于浏览器更喜欢使用现有的 tcp 连接进行管道传输，因此多路复用器往往会将数据包发送到错误的协议处理程序，例如，它可能会将 grpc-web 数据包发送到 rest，反之亦然。幸运的是，有一个非常简单的解决方案：

package main

listener := net.Listen("tcp", ":2289")
multiplexer := cmux.New(listener)
grpcListener := multiplexer.Match(cmux.HTTP2())
httpListener := multiplexer.Match(cmux.HTTP1Fast())
grpcServer := grpc.Server()
wrappedGrpc := grpcweb.WrapServer(grpcServer)
go func() {
  httpServer := echo.New()
  (&http.Server{
      Handler: http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
        if strings.Contains(req.Header.Get("Access-Control-Request-Headers"), "x-grpc-web") || req.Header.Get("x-grpc-web") == "1" || req.Header.Get("Sec-Websocket-Protocol") == "grpc-websockets" {
          inst.API.GrpcWebServer.ServeHTTP(resp, req)
        } else {
          httpServer.ServeHTTP(resp, req)
      }
    }),
  }).Serve(httpListener)
}()

go func() {
  grpcServer.Serve(grpcListener)
}()

go func() {
  multiplexer.Serve()
}()

这是如何工作的？

本质上，我们没有使用 cmux 的默认多路复用（仅对每个连接进行多路复用），而是在所有传入的 http 请求上注册了一个新的迷你 http 服务器处理程序，然后让我们显式检查标头并直接调用处理程序。

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。