1.7 Go HTTP服务器进阶：路由、中间件、错误处理完整实现

Go HTTP服务器进阶：路由、中间件、错误处理完整实现

引言

在掌握了 HTTP 服务器基础后，我们需要学习更高级的特性来构建生产级的应用。本文将深入讲解路由管理、中间件设计、错误处理、优雅关闭等进阶特性，让你能够构建健壮、可维护的 HTTP 服务器。

一、高级路由管理

1.1 使用第三方路由库（Gorilla Mux）

虽然标准库足够强大，但第三方路由库提供了更多功能：

go get github.com/gorilla/mux

packagemainimport("fmt""net/http""github.com/gorilla/mux")funcmain(){r:=mux.NewRouter()// 基本路由r.HandleFunc("/",homeHandler).Methods("GET")// 路径参数r.HandleFunc("/users/{id}",getUserHandler).Methods("GET")// 正则表达式匹配r.HandleFunc("/users/{id:[0-9]+}",getUserHandler).Methods("GET")// 查询参数r.HandleFunc("/search",searchHandler).Queries("q","{query}")// 子路由api:=r.PathPrefix("/api").Subrouter()api.HandleFunc("/users",listUsersHandler).Methods("GET")api.HandleFunc("/users",createUserHandler).Methods("POST")http.ListenAndServe(":8080",r)}funchomeHandler(w http.ResponseWriter,r*http.Request){fmt.Fprintf(w,"首页")}funcgetUserHandler(w http.ResponseWriter,r*http.Request){vars:=mux.Vars(r)id:=vars["id"]fmt.Fprintf(w,"用户 ID: %s",id)}funcsearchHandler(w http.ResponseWriter,r*http.Request){vars:=mux.Vars(r)query:=vars["query"]fmt.Fprintf(w,"搜索: %s",query)}funclistUsersHandler(w http.ResponseWriter,r*http.Request){fmt.Fprintf(w,"用户列表")}funccreateUserHandler(w http.ResponseWriter,r*http.Request){fmt.Fprintf(w,"创建用户")}

1.2 路由分组和版本控制

packagemainimport("fmt""net/http""github.com/gorilla/mux")funcmain(){r:=mux.NewRouter()// API 版本 1v1:=r.PathPrefix("/api/v1").Subrouter()v1.HandleFunc("/users",v1UsersHandler).Methods("GET")// API 版本 2v2:=r.PathPrefix("/api/v2").Subrouter()v2.HandleFunc("/users",v2UsersHandler).Methods("GET")http.ListenAndServe(":8080",r)}funcv1UsersHandler(w http.ResponseWriter,r*http.Request){fmt.Fprintf(w,"API v1 用户列表")}funcv2UsersHandler(w http.ResponseWriter,r*http.Request){fmt.Fprintf(w,"API v2 用户列表（新格式）")}

二、中间件设计模式

2.1 中间件接口设计

packagemainimport("context""net/http")typeMiddlewarefunc(http.Handler)http.Handler// 链式组合中间件funcChain(middlewares...Middleware)Middleware{returnfunc(next http.Handler)http.Handler{fori:=len(middlewares)-1;i>=0;i--{next=middlewares[i](next)}returnnext}}// 应用中间件到 Handlerfunc(m Middleware)Apply(next http.Handler)http.Handler{returnm(next)}

2.2 请求 ID 中间件

packagemainimport("context""net/http""github.com/google/uuid")constRequestIDKey="request_id"funcRequestIDMiddleware(next http.Handler)http.Handler{returnhttp.HandlerFunc(func(w http.ResponseWriter,r*http.Request){// 生成或获取请求 IDrequestID:=r.Header.Get("X-Request-ID")ifrequestID==""{requestID=uuid.New().String()}// 设置到响应头w.Header().Set("X-Request-ID",requestID)// 设置到上下文ctx:=cont