如何在gRPC服务器中配置自定义Handler（使用chi路由）

要让gRPC服务器同时处理gRPC请求和Chi路由的HTTP请求，核心是实现请求分流——根据请求类型将流量转发给对应的处理逻辑。以下是两种可行方案：

方案一：自定义Handler分流（无额外依赖） #

直接通过net/http的Handler判断请求类型，将gRPC请求交给gRPC服务器处理，普通HTTP请求交给Chi路由。

实现步骤 #

1. 初始化gRPC服务器并注册服务
2. 生成你的Chi路由实例（复用你已有的routes()方法）
3. 编写分流逻辑，通过请求头和HTTP版本区分gRPC请求

示例代码 #

import (
    "log"
    "net"
    "net/http"
    "strings"
    "google.golang.org/grpc"
)

// 注册你的gRPC服务（根据实际业务实现）
func registerGRPCServer(s *grpc.Server) {
    // pb.RegisterYourServiceServer(s, &yourServiceImpl{})
}

func main() {
    // 初始化Chi路由
    app := &Config{}
    chiHandler := app.routes()

    // 创建gRPC服务器
    grpcServer := grpc.NewServer()
    registerGRPCServer(grpcServer)

    // 自定义分流Handler
    combinedHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 判断是否为gRPC请求：HTTP/2协议 + Content-Type为application/grpc
        if r.ProtoMajor == 2 && strings.HasPrefix(r.Header.Get("Content-Type"), "application/grpc") {
            grpcServer.ServeHTTP(w, r)
        } else {
            // 普通HTTP请求交给Chi处理
            chiHandler.ServeHTTP(w, r)
        }
    })

    // 启动服务器
    listener, err := net.Listen("tcp", ":50051")
    if err != nil {
        log.Fatalf("监听端口失败: %v", err)
    }
    log.Printf("服务器启动，监听地址: %v", listener.Addr())
    
    if err := http.Serve(listener, combinedHandler); err != nil {
        log.Fatalf("服务启动失败: %v", err)
    }
}

方案二：使用cmux实现协议分流（更优雅） #

借助cmux库（专门用于多协议端口复用），根据请求协议类型将流量分流到gRPC服务器和HTTP服务器，逻辑更严谨，适合复杂场景。

实现步骤 #

1. 安装cmux依赖：go get github.com/soheilhy/cmux
2. 创建端口监听器并初始化cmux实例
3. 分别为gRPC和HTTP请求创建匹配规则
4. 启动两个服务器（gRPC和Chi HTTP）并运行cmux

示例代码 #

import (
    "log"
    "net"
    "net/http"
    "google.golang.org/grpc"
    "github.com/soheilhy/cmux"
)

// 注册你的gRPC服务（根据实际业务实现）
func registerGRPCServer(s *grpc.Server) {
    // pb.RegisterYourServiceServer(s, &yourServiceImpl{})
}

func main() {
    // 初始化Chi路由
    app := &Config{}
    chiHandler := app.routes()

    // 监听端口
    lis, err := net.Listen("tcp", ":50051")
    if err != nil {
        log.Fatalf("监听端口失败: %v", err)
    }

    // 创建cmux实例
    mux := cmux.New(lis)

    // 匹配gRPC请求（HTTP/2协议）
    grpcListener := mux.Match(cmux.HTTP2())
    // 匹配普通HTTP请求（HTTP/1.x协议）
    httpListener := mux.Match(cmux.HTTP1())

    // 启动gRPC服务器
    grpcServer := grpc.NewServer()
    registerGRPCServer(grpcServer)
    go func() {
        if err := grpcServer.Serve(grpcListener); err != nil {
            log.Fatalf("gRPC服务启动失败: %v", err)
        }
    }()

    // 启动HTTP服务器（使用Chi路由）
    httpServer := &http.Server{Handler: chiHandler}
    go func() {
        if err := httpServer.Serve(httpListener); err != nil {
            log.Fatalf("HTTP服务启动失败: %v", err)
        }
    }()

    // 启动cmux分流器
    log.Printf("服务器启动，监听地址: %v", lis.Addr())
    if err := mux.Serve(); err != nil {
        log.Fatalf("分流服务启动失败: %v", err)
    }
}

注意事项 #

• 路径冲突：确保gRPC服务的路径（格式为/package.Service/Method）与Chi路由的路径无重叠，避免请求处理混乱。
• TLS支持：如果需要加密传输，两种方案都可以配置TLS证书，cmux支持在TLS层之上进行协议分流。
• 测试验证：分别测试gRPC请求和HTTP请求，确认两者都能被正确处理。

内容的提问来源于stack exchange，提问作者Girish Bhutiya