如果一个应用需要同时对外提供 HTTP 和 gRPC 服务，通常情况下我们会为两个服务绑定不同的监听端口，而本文要介绍的 cmux 为我们提供了一种连接多路复用的新选择，使用 cmux 可以将不同服务绑定在同一个网络端口上！

✦

简介

✦

多路复用是个很常见的概念，我们在编写 HTTP 服务时通常会用 http.ServeMux 或类似的三方工具实现请求与 handler 的匹配和绑定，这类多路复用一般作用在某个服务内部。而 cmux（github.com/soheilhy/cmux） 则提供了基于请求内容对连接进行多路复用的能力，使用 cmux 我们可以在同一个 TCP 监听（端口）上提供包括 gRPC、SSH、HTTPS、HTTP、Go RPC 在内的几乎任何协议。

cmux 通过扩展 net.Listener 的方式实现了连接多路复用能力，在接收到客户端请求后，cmux 会根据注册的规则对客户端请求进行鉴别和匹配，并根据匹配结果将请求转发给相应的服务。

✦

使用举例

✦

cmux 的使用特别简单，我们只需要为每个服务指定相应的匹配规则即可：

package mainimport ("context""log""net""net/http""google.golang.org/grpc"pb "google.golang.org/grpc/examples/helloworld/helloworld""github.com/soheilhy/cmux"
)func main() {lis, err := net.Listen("tcp", "localhost:50051")if err != nil {log.Fatalf("failed to listen: %v", err)}mux := cmux.New(lis)// gRPC 匹配规则grpcL := mux.MatchWithWriters(cmux.HTTP2MatchHeaderFieldSendSettings("content-type", "application/grpc"),)// otherwise serve httphttpL := mux.Match(cmux.Any())// gRPC servergrpcS := grpc.NewServer()pb.RegisterGreeterServer(grpcS, &server{})// HTTP serverhttpS := &http.Server{Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {w.Write([]byte("greet from HTTP\n"))}),}// start serving!go grpcS.Serve(grpcL)go httpS.Serve(httpL)log.Printf("serve both grpc and http at %v", lis.Addr())if err := mux.Serve(); err != nil {log.Fatalf("failed to serve: %v", err)}
}type server struct {pb.UnimplementedGreeterServer
}func (*server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {return &pb.HelloReply{Message: "greet from gRPC"}, nil
}

先来试试 http 访问是否正常

$ curl -D - http://localhost:50051
HTTP/1.1 200 OK
Date: Mon, 21 Mar 2022 08:20:15 GMT
Content-Length: 16
Content-Type: text/plain; charset=utf-8greet from HTTP

再来试试 gRPC 访问

$ ${GOPATH}/bin/greeter_client -addr localhost:50051
2022/03/21 16:12:11 Greeting: greet from gRPC

测试客户端安装：go install google.golang.org/grpc/examples/helloworld/greeter_client

如上，我们通过 50051 这个端口同时承载了 HTTP 和 gRPC 两个服务，是不是特别神奇！

✦

总结

✦

使用 cmux 我们可以轻松实现对服务端连接的多路复用，cmux 仅会在连接建立之初读取少量请求样本进行路由匹配，因此在客户端长连接场景下的性能损失几乎可以忽略不计。大家来试试吧！

✦

参考资料

✦

• https://github.com/soheilhy/cmux

《酷Go推荐》招募：

各位Gopher同学，最近我们社区打算推出一个类似GoCN每日新闻的新栏目《酷Go推荐》，主要是每周推荐一个库或者好的项目，然后写一点这个库使用方法或者优点之类的，这样可以真正的帮助到大家能够学习到

新的库，并且知道怎么用。

大概规则和每日新闻类似，如果报名人多的话每个人一个月轮到一次，欢迎大家报名！戳「阅读原文」，即可报名

想要了解更多 Golang 相关的内容，欢迎扫描下方