「GoCN酷Go推荐」使用 cmux 实现服务端连接多路复用
如果一个应用需要同时对外提供 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 main
import (
"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 http
httpL := mux.Match(cmux.Any())
// gRPC server
grpcS := grpc.NewServer()
pb.RegisterGreeterServer(grpcS, &server{})
// HTTP server
httpS := &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-8
greet 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 相关的内容，欢迎扫描下方👇 关注 公众号，回复关键词 [实战群] ,就有机会进群和我们进行交流~