Dubbo-go中Triple协议HTTP头信息缺失问题解析

背景介绍

在微服务架构中，Dubbo-go作为一款高性能的Go语言RPC框架，支持多种协议通信。其中Triple协议是基于HTTP/2的gRPC协议实现，但在实际使用过程中，开发者发现Triple协议存在HTTP头信息缺失的问题，特别是无法获取请求的基本信息如scheme(http/https)、host等。

问题现象

当开发者尝试通过Dubbo-go的过滤器获取Triple协议的请求头信息时，发现只能获取到部分标准头信息，而缺少了一些常见的HTTP头字段：

user-agent grpc-go-triple/0.1.0 (go1.21.6)
te trailers
content-type application/grpc+proto
interface greet.GreetService
grpc-accept-encoding gzip
grpc-timeout 2999023u
accept-encoding identity

这些信息中缺失了关键的scheme(http/https)、host等字段，而这些字段对于实现某些功能(如安全认证、请求路由等)是必要的。

技术分析

HTTP/2与h2c协议

Triple协议底层使用HTTP/2进行通信。在非TLS环境下，会使用h2c(HTTP/2 Cleartext)协议。h2c协议有一个特殊的协商过程：客户端会首先发送一个PRI方法的请求来协商协议升级。

问题根源

1. HTTP头设置时机问题：在h2c协商阶段(PRI方法请求)，此时不能随意设置自定义头，否则会破坏协议协商过程。

2. 协议实现差异：HTTPS和HTTP在协议处理上存在差异，HTTPS可以直接获取TLS相关信息，而HTTP则不能。

3. 头信息传递机制：Dubbo-go的Triple协议实现中，没有将完整的HTTP头信息传递到业务层。

解决方案

双端口方案

针对这一问题，Dubbo-go社区提出了双端口解决方案：

1. 同时监听HTTP和HTTPS端口：根据配置决定是否启用HTTPS端口。

2. 差异化处理头信息：

   • 对于HTTPS请求：可以安全地设置和获取各种头信息，包括TLS证书相关信息
   • 对于HTTP请求：避免在h2c协商阶段设置自定义头

func (s *Server) Run() error {
    httpAddr := s.addr
    httpsAddr := s.getHTTPSAddress(s.addr)
    
    // 根据配置决定是否启用HTTPS
    httpOn := true
    httpsOn := false
    if s.tlsConfigProvider != nil {
        httpsOn = true
    }
    
    // HTTP处理器
    setHTTPHeaders := func(h http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            // 避免在h2c协商阶段设置头
            if r.Method != "PRI" {
                headers := make(map[string]interface{})
                headers[constant.HttpHeaderXSchemeName] = "http"
                headers[constant.HttpHeaderXHostName] = r.Host
                // 其他头信息...
                ctx := context.WithValue(r.Context(), constant.AttachmentKey, headers)
                request := r.WithContext(ctx)
                h.ServeHTTP(w, request)
                return
            }
            h.ServeHTTP(w, r)
        })
    }
    
    // HTTPS处理器
    setHTTPSHeaders := func(h http.Handler) http.Handler {
        return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
            headers := make(map[string]interface{})
            headers[constant.HttpHeaderXSchemeName] = "https"
            headers[constant.HttpHeaderXHostName] = r.Host
            // 可以安全设置TLS相关信息
            if certs := r.TLS.PeerCertificates; len(certs) > 0 {
                if uris := certs[0].URIs; len(uris) > 0 {
                    headers[constant.HttpHeaderXSpiffeName] = uris[0].String()
                }
            }
            ctx := context.WithValue(r.Context(), constant.AttachmentKey, headers)
            request := r.WithContext(ctx)
            h.ServeHTTP(w, request)
        })
    }
    
    // 初始化HTTP和HTTPS服务器
    if httpOn {
        s.httpSrv = &http.Server{Handler: setHTTPHeaders(handler)}
    }
    if httpsOn {
        s.httpsSrv = &http.Server{Handler: setHTTPSHeaders(handler)}
    }
    
    // 启动服务...
}

实现细节

1. h2c处理：使用标准库的h2c处理器来处理HTTP/2明文连接。

2. 头信息传递：通过context传递头信息，避免直接修改请求对象。

3. TLS信息获取：在HTTPS处理器中，可以安全地获取客户端证书等信息。

4. 协议协商保护：在HTTP处理器中，特别检查PRI方法，避免破坏h2c协商。

最佳实践建议

1. 统一头信息处理：建议在框架层面统一处理HTTP头信息的传递，而不是让每个应用自行处理。

2. 安全考虑：对于敏感信息(如TLS证书信息)，应该提供明确的访问控制机制。

3. 性能优化：头信息的处理应该尽量减少内存分配和复制操作。

4. 兼容性保证：新版本应该保持对旧版本行为的兼容，避免破坏现有应用。

总结

Dubbo-go的Triple协议在HTTP头信息处理上存在一些限制，主要是由于HTTP/2协议本身的特性导致的。通过双端口方案和差异化的头信息处理策略，可以有效地解决这一问题。开发者在使用时需要注意协议差异，并根据实际需求选择合适的实现方式。未来Dubbo-go可能会在框架层面提供更完善的解决方案，简化开发者的使用体验。