Golang：HttpRouter 源码分析（二）

2023-10-24 09:02:25

感谢大家阅读我上篇文章：《Golang：HttpRouter 源码分析（一）》，今天继续为大家带来第二篇文章。本篇文章我们将一起继续剖析 Golang 中 HttpRouter 库的源码，并通过一个完整的示例来演示路由树的构建过程。

HttpRouter 是一个 Golang 中非常受欢迎的 HTTP 路由库，以其高效和强大的功能而闻名。上篇讲到了如何创建一个 HttpRouter 路由器，以及如何注册路由。本篇文章我们将进一步深入了解路由树的构建过程，以及如何使用路由树来处理 HTTP 请求。

深入路由树

路由树是一个二叉树，每个节点都存储了一个路由。当一个 HTTP 请求到达时，路由器会根据请求的路径在路由树中进行查找。如果找到匹配的路由，则将请求转发给对应的处理器。如果没有找到匹配的路由，则会返回 404 Not Found 错误。

路由树的构建过程是一个递归的过程。首先，路由器会创建一个根节点。然后，对于每个注册的路由，路由器会从根节点开始，根据路由的路径逐层向下查找。如果找到一个匹配的节点，则将路由添加到该节点。如果没有找到匹配的节点，则会创建一个新的节点，并将路由添加到该节点。

例如，我们注册了一个路由 /user/{id}，其中 {id} 是一个占位符。当一个 HTTP 请求到达时，路由器会从根节点开始，根据请求的路径 /user/123 逐层向下查找。如果找到一个匹配的节点，则将请求转发给对应的处理器。如果没有找到匹配的节点，则会创建一个新的节点，并将路由添加到该节点。

路由树的构建过程非常高效，因为它只会在需要时才创建节点。因此，即使注册了大量路由，路由树也不会变得非常庞大。

处理 HTTP 请求

当一个 HTTP 请求到达时，路由器会根据请求的路径在路由树中进行查找。如果找到匹配的路由，则将请求转发给对应的处理器。如果没有找到匹配的路由，则会返回 404 Not Found 错误。

路由器的处理器是一个函数，它接受一个 HTTP 请求和一个 HTTP 响应作为参数。处理器可以对请求进行处理，并生成一个 HTTP 响应。例如，我们可以创建一个处理器来处理 /user/{id} 路由，该处理器可以根据请求的路径参数 {id} 来获取用户信息，然后将用户信息输出到 HTTP 响应中。

func UserHandler(w http.ResponseWriter, r *http.Request) {
  // 获取请求的路径参数 {id}
  id := mux.Vars(r)["id"]

  // 根据 {id} 获取用户信息
  user, err := GetUser(id)
  if err != nil {
    http.Error(w, "User not found", http.StatusNotFound)
    return
  }

  // 将用户信息输出到 HTTP 响应中
  fmt.Fprintf(w, "User: %s", user.Name)
}