当前位置： 技术文章>> 100道Go语言面试题之-Go语言的net/http包是如何处理HTTP请求的？如何编写一个处理HTTP请求的中间件？

文章标题：100道Go语言面试题之-Go语言的net/http包是如何处理HTTP请求的？如何编写一个处理HTTP请求的中间件？
  • 文章分类: 后端
  • 9538 阅读

Go语言的net/http包是如何处理HTTP请求的？

Go语言的net/http包是Go标准库的一部分，它提供了一个非常强大且易用的HTTP服务器和客户端实现。net/http包处理HTTP请求的过程大致可以概括为以下几个步骤：

  1. 监听端口：通过http.ListenAndServe或http.Serve函数启动HTTP服务器，并监听指定端口上的TCP连接。

  2. 接收请求：当客户端发起HTTP请求时，服务器接收请求并解析请求头、请求体等信息。

  3. 路由请求：通过URL路径匹配找到对应的处理器（Handler）。在Go的net/http包中，这通常是通过http.HandleFunc、http.Handle或自定义的ServeMux实现的。

  4. 处理请求：调用注册的处理器（Handler）来处理请求。处理器是一个实现了http.Handler接口的类型，它必须有一个ServeHTTP(ResponseWriter, *Request)方法。

  5. 发送响应：处理器处理完请求后，通过ResponseWriter接口向客户端发送HTTP响应，包括状态码、响应头和响应体。

  6. 关闭连接：请求处理完毕后，TCP连接可能会根据HTTP协议或服务器配置被关闭或保持活动状态以供后续请求使用。

如何编写一个处理HTTP请求的中间件？

在Go中，中间件是一个函数，它接受一个http.Handler作为参数并返回一个新的http.Handler。这样，你可以将多个中间件串联起来，每个中间件都可以对请求进行预处理或对响应进行后处理，然后将请求传递给下一个中间件或最终的处理器。

下面是一个简单的中间件示例，该中间件记录每个请求的到达时间，并在响应中添加一个自定义的HTTP头：

package main

import (
    "fmt"
    "net/http"
    "time"
)

// LoggingMiddleware 是一个中间件，它记录了请求的到达时间
func LoggingMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 记录请求到达时间
        startTime := time.Now()

        // 调用下一个处理器（可能是另一个中间件或最终的处理器）
        next.ServeHTTP(w, r)

        // 记录请求处理时间
        fmt.Printf("Request %s %s took %v\n", r.Method, r.URL.Path, time.Since(startTime))

        // 可以在这里添加更多的后处理逻辑，比如修改响应头等
        // 例如，添加一个自定义的HTTP头
        w.Header().Set("X-Processed-By", "MyMiddleware")
    })
}

func mainHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Hello, world!")
}

func main() {
    http.Handle("/", LoggingMiddleware(http.HandlerFunc(mainHandler)))
    http.ListenAndServe(":8080", nil)
}

在这个例子中，LoggingMiddleware中间件记录了请求的到达时间，并在请求处理完毕后打印了处理时间。然后，它向响应中添加了一个名为X-Processed-By的自定义HTTP头。最后，它调用next.ServeHTTP(w, r)将请求传递给下一个处理器（在这个例子中是mainHandler）。

推荐文章