使用Golang的x/net/http2实现单TCP连接发送多请求

我的服务器支持 HTTP/1.1 升级到 h2c，这意味着无需 TLS。

哦。我不确定你能否使用标准库实现这一点。看起来你已经找到了这个变通方案，但也许它现在不能按预期工作了。或许你应该在 Github 上提个 issue，也许有人会有更好的主意？

要实现单TCP连接发送多请求，需要使用HTTP/2的连接复用特性。你的代码已经接近正确，但需要确保服务器支持HTTP/2，并且请求要复用同一个连接。以下是完整的示例：

package main

import (
    "context"
    "crypto/tls"
    "fmt"
    "net"
    "net/http"
    "time"

    "golang.org/x/net/http2"
)

func main() {
    // 创建支持HTTP/2的Transport
    transport := &http2.Transport{
        AllowHTTP: true, // 允许HTTP（非TLS）连接
        DialTLS: func(network, addr string, cfg *tls.Config) (net.Conn, error) {
            // 使用普通TCP连接（非TLS）
            return net.Dial(network, addr)
        },
    }

    // 创建HTTP客户端
    client := &http.Client{
        Transport: transport,
        Timeout:   10 * time.Second,
    }

    // 创建多个请求
    req1, _ := http.NewRequest("GET", "http://localhost:8080/api/1", nil)
    req2, _ := http.NewRequest("GET", "http://localhost:8080/api/2", nil)
    req3, _ := http.NewRequest("POST", "http://localhost:8080/api/3", nil)

    // 使用相同的客户端发送请求（会自动复用连接）
    var wg sync.WaitGroup
    wg.Add(3)

    go func() {
        defer wg.Done()
        resp, err := client.Do(req1)
        if err != nil {
            fmt.Printf("Request 1 error: %v\n", err)
            return
        }
        defer resp.Body.Close()
        fmt.Println("Request 1 completed")
    }()

    go func() {
        defer wg.Done()
        resp, err := client.Do(req2)
        if err != nil {
            fmt.Printf("Request 2 error: %v\n", err)
            return
        }
        defer resp.Body.Close()
        fmt.Println("Request 2 completed")
    }()

    go func() {
        defer wg.Done()
        resp, err := client.Do(req3)
        if err != nil {
            fmt.Printf("Request 3 error: %v\n", err)
            return
        }
        defer resp.Body.Close()
        fmt.Println("Request 3 completed")
    }()

    wg.Wait()
}

如果你需要更精确地控制连接复用，可以使用http2.Transport的DialTLSContext方法：

transport := &http2.Transport{
    AllowHTTP: true,
    DialTLSContext: func(ctx context.Context, network, addr string, cfg *tls.Config) (net.Conn, error) {
        dialer := &net.Dialer{
            Timeout:   30 * time.Second,
            KeepAlive: 30 * time.Second,
        }
        return dialer.DialContext(ctx, network, addr)
    },
}

要验证是否真正复用了连接，可以添加连接统计：

// 添加连接状态监控
var connectionCount int32
transport = &http2.Transport{
    AllowHTTP: true,
    DialTLS: func(network, addr string, cfg *tls.Config) (net.Conn, error) {
        atomic.AddInt32(&connectionCount, 1)
        fmt.Printf("Creating new connection #%d to %s\n", connectionCount, addr)
        return net.Dial(network, addr)
    },
}

确保服务器端也支持HTTP/2。对于Go的标准HTTP服务器，需要这样启用：

server := &http.Server{
    Addr:    ":8080",
    Handler: yourHandler,
}
// 启用HTTP/2支持
http2.ConfigureServer(server, &http2.Server{})
server.ListenAndServe()

关键点：

1. 所有请求使用同一个http.Client实例
2. 目标服务器必须支持HTTP/2
3. 并发请求会自动复用连接
4. 连接复用由HTTP/2协议层自动处理