本文介绍了在 Go HTTP 服务器中处理带有请求体的 GET 请求的方法。虽然 HTTP GET 请求通常不应包含请求体，但如果遇到需要处理此类请求的场景，本文将提供解决方案，包括检查 Content-Length 头部、修改标准库以及使用 Hijack 连接等方法。
在 HTTP 协议中，GET 方法通常用于获取资源，按照规范不应该包含请求体。然而，在某些特殊情况下，客户端可能会发送带有请求体的 GET 请求。Go 的 net/http 包默认情况下会忽略 GET 请求中的请求体，这使得直接读取请求体变得困难。本文将探讨如何在 Go HTTP 服务器中处理这类非标准的 GET 请求。
理解 net/http 包的行为
Go 的 net/http 包在处理 GET 请求时，会检查 Content-Length 头部。如果存在该头部，则会尝试读取请求体；否则，会默认认为 GET 请求没有请求体。这个行为在 net/http 包的 transfer.go 文件中的 fixLength 函数中有所体现。
if !isResponse && requestMethod == "GET" {
// RFC 2616 doesn't explicitly permit nor forbid an
// entity-body on a GET request so we permit one if
// declared, but we default to 0 here (not -1 below)
// if there's no mention of a body.
return 0, nil
}解决方案
-
确保客户端发送 Content-Length 头部
最简单的解决方案是确保客户端在发送 GET 请求时包含 Content-Length 头部，即使请求体为空。这样，net/http 包就会尝试读取请求体。
例如，使用 curl 发送带有请求体的 GET 请求：
curl -X GET -d '{"key":"value"}' -H "Content-Type: application/json" -H "Content-Length: 16" http://localhost:8080/Go 服务器代码：
package main import ( "fmt" "io" "log" "net/http" ) func handler(w http.ResponseWriter, r *http.Request) { defer r.Body.Close() body, err := io.ReadAll(r.Body) if err != nil { http.Error(w, "Error reading request body", http.StatusInternalServerError) log.Printf("Error reading body: %v", err) return } log.Printf("body: %v", string(body)) fmt.Fprintf(w, "Received: %s", string(body)) } func main() { http.HandleFunc("/", handler) log.Fatal(http.ListenAndServe(":8080", nil)) } -
修改 net/http 标准库
如果无法控制客户端的行为，并且客户端不发送 Content-Length 头部，则需要修改 net/http 标准库。
- 将 net/http 包从标准库复制到你的项目目录中，例如 myhttp。
- 修改 myhttp/transfer.go 文件中的 fixLength 函数，移除对 GET 请求的特殊处理。
- 在你的代码中，将 import "net/http" 替换为 import "myhttp"。
这种方法不推荐，因为它会增加维护成本，并且可能与未来的 Go 版本不兼容。
-
使用 Hijack 连接
如果客户端不使用 Keep-Alive 连接，可以使用 Hijack 方法获取底层的 TCP 连接，并直接从 Socket 中读取数据。
package main import ( "bufio" "fmt" "log" "net/http" ) func handler(w http.ResponseWriter, r *http.Request) { conn, bufrw, err := hijacker(w) if err != nil { log.Printf("Error hijacking connection: %v", err) http.Error(w, "Error hijacking connection", http.StatusInternalServerError) return } defer conn.Close() // 读取请求行，但忽略它，因为我们已经知道它是 GET 请求 _, err = bufrw.ReadString('\n') if err != nil { log.Printf("Error reading request line: %v", err) return } // 读取头部，直到空行 for { line, err := bufrw.ReadString('\n') if err != nil { log.Printf("Error reading headers: %v", err) return } if line == "\r\n" { break // 空行表示头部结束 } log.Printf("Header: %s", line) // 打印头部信息 } // 读取剩余的数据作为请求体 body, err := bufrw.ReadString('\n') // 这里假设请求体以换行符结束 if err != nil { log.Printf("Error reading body: %v", err) return } log.Printf("Body: %v", body) fmt.Fprintf(bufrw, "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\n\r\nReceived: %s", body) bufrw.Flush() } func hijacker(w http.ResponseWriter) (conn net.Conn, bufrw *bufio.ReadWriter, err error) { h, ok := w.(http.Hijacker) if !ok { return nil, nil, fmt.Errorf("type assertion to http.Hijacker failed") } conn, bufrw, err = h.Hijack() if err != nil { return nil, nil, err } return conn, bufrw, nil } func main() { http.HandleFunc("/", handler) log.Fatal(http.ListenAndServe(":8080", nil)) }注意事项:
- Hijack 会接管连接，因此需要手动处理响应。
- 这种方法只适用于非 Keep-Alive 连接。
总结
处理带有请求体的 GET 请求是一种非标准的做法，应尽量避免。如果必须处理此类请求，可以考虑以下方法：
- 确保客户端发送 Content-Length 头部。
- 修改 net/http 标准库（不推荐）。
- 使用 Hijack 连接（仅适用于非 Keep-Alive 连接）。
在选择解决方案时，需要权衡其复杂性、维护成本以及对现有代码的影响。最好的做法是修复客户端，使其遵循 HTTP 规范。
