时间戳

首页 > 解决方案 > 如何使用 grpc-web 进行流式传输并在同一端口上使用 REST API？

javascript - 如何使用 grpc-web 进行流式传输并在同一端口上使用 REST API？

问题描述

我在 Go 中有一个服务器，用于cmux允许多个协议在同一个端口上运行，但是 README 中指定了一个限制，说

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

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

标签： javascripthttpgogrpcmuxer

解决方案

这是一个非常棘手的问题。由于浏览器更喜欢使用现有的 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 服务器处理程序，然后我们可以显式检查标头并直接调用处理程序。

推荐阅读