本教程详细阐述了在go语言中如何利用http.hijacker接口实现http流式双工通信。针对标准http处理器可能在写入响应后关闭请求体的限制，文章通过劫持底层tcp连接，允许开发者同时读写请求和响应数据，从而实现高效、实时的双向数据流处理，并提供了详细的代码示例和注意事项。
理解HTTP双工通信的挑战
在go的net/http包中，标准的HTTP处理器通常遵循请求-响应模型。当服务器开始向http.ResponseWriter写入响应数据时，底层的http.Request.Body流可能会被关闭，这使得在响应开始发送后，无法继续从请求体中读取数据。这种行为对于需要进行流式处理，即在读取请求体的同时向客户端发送部分响应的场景（例如代理、实时数据转换服务或某些长连接应用）构成了挑战。传统的HTTP处理器设计旨在一次性处理请求并发送完整响应，不适合需要客户端和服务器之间持续双向数据流的“双工”通信模式。
Go中实现双工通信的核心机制：http.Hijacker
为了克服标准HTTP处理器的限制，Go语言提供了http.Hijacker接口。这个接口允许HTTP服务器的处理器“劫持”底层的TCP连接，从而获得对原始net.Conn的完全控制权。一旦连接被劫持，HTTP服务器将不再管理该连接的生命周期，包括读写操作和关闭。这为实现自定义的协议、长连接或如本教程所述的流式双工通信提供了极大的灵活性。
http.Hijacker接口定义如下：
type Hijacker interface { Hijack() (net.Conn, *bufio.ReadWriter, Error) }
Hijack()方法返回三个值：
立即学习“go语言免费学习笔记（深入）”；
- net.Conn: 原始的TCP连接，允许直接进行字节流读写。
- *bufio.ReadWriter: 一个带有缓冲的读写器，包含了在劫持前HTTP服务器可能已经读取或写入的数据。
- error: 劫持过程中可能发生的错误。
使用http.Hijacker实现流式双工处理器
实现流式双工处理器的关键步骤是获取并管理被劫持的连接。以下是具体的实现：
-
获取Hijacker接口： 在HTTP处理器中，首先尝试将http.ResponseWriter断言为http.Hijacker接口。如果服务器不支持劫持（尽管大多数Go HTTP服务器都支持），则应返回错误。
package main import ( "bufio" "fmt" "io" "log" "net/http" "time" ) func duplexHandler(w http.ResponseWriter, r *http.Request) { // 1. 检查并劫持连接 hj, ok := w.(http.Hijacker) if !ok { http.Error(w, "服务器不支持连接劫持", http.StatusInternalServerError) return } conn, bufrw, err := hj.Hijack() if err != nil { http.Error(w, fmt.Sprintf("劫持连接失败: %v", err), http.StatusInternalServerError) return } defer conn.Close() // 确保连接最终被关闭 log.Println("连接已劫持") // 2. 手动发送HTTP响应头 // 劫持后，我们需要手动构造并发送HTTP响应的起始行和头部。 // 这里我们发送一个简单的200 OK响应，并设置Connection: keep-alive以保持连接。 _, err = bufrw.WriteString("HTTP/1.1 200 OKrn") _, err = bufrw.WriteString("Content-Type: text/plainrn") _, err = bufrw.WriteString("Connection: keep-alivern") // 保持连接 _, err = bufrw.WriteString("rn") // 空行表示头部结束 if err != nil { log.Printf("写入响应头失败: %v", err) return } // 立即刷新缓冲区，确保响应头发送到客户端 if err := bufrw.Flush(); err != nil { log.Printf("刷新响应头失败: %v", err) return } log.Println("响应头已发送") // 3. 实现双向数据流 // 现在我们可以同时从请求体读取数据并向响应体写入数据。 // 这里我们创建一个简单的示例：从请求体读取，进行一些处理，然后写入响应体。 // 使用goroutine同时处理读写，以实现双工通信 done := make(chan struct{}) // 读取请求体并处理 go func() { defer func() { done <- struct{}{} }() // 读取完成后通知主goroutine // r.Body 包含原始请求体，bufrw.Reader 可能包含劫持前已缓冲的请求数据。 // 使用io.MultiReader确保我们能读取所有请求数据。 reader := io.Reader(r.Body) if bufrw.Reader.Buffered() > 0 { reader = io.MultiReader(bufrw.Reader, r.Body