通过中间件统一捕获错误,提升Go http服务的可维护性:定义返回Error的HandlerFunc类型,创建ErrorMiddleware捕获错误并根据自定义错误类型(如NotFoundError、ValidationError)返回对应json响应,业务逻辑中直接返回错误,路由注册时用中间件包装,实现错误处理集中化与API响应标准化。
在golang中构建HTTP服务时,统一处理错误是提升代码可维护性和API一致性的关键。通过中间件捕获路由处理函数中的错误,可以避免在每个handler中重复写错误处理逻辑。
使用中间件统一捕获HTTP错误
Go的HTTP处理函数(http.HandlerFunc)本身不支持返回error,但我们可以封装一个返回error的处理函数类型,并通过中间件来捕获这些错误。
定义一个能返回错误的处理函数类型:
type HandlerFunc func(w http.ResponseWriter, r *http.Request) error
然后编写中间件,将这种函数包装成标准的
http.HandlerFunc
,并在发生错误时统一处理:
实现错误捕获中间件:
func ErrorMiddleware(next HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { if err := next(w, r); err != nil { // 可根据错误类型做不同处理 var apiError struct { Code int `json:"code"` Message string `json:"message"` } switch e := err.(type) { case *NotFoundError: apiError.Code = 404 apiError.Message = e.Error() case *ValidationError: apiError.Code = 400 apiError.Message = e.Error() default: apiError.Code = 500 apiError.Message = "Internal server error" } w.Header().Set("Content-Type", "application/json") w.WriteHeader(apiError.Code) json.NewEncoder(w).Encode(apiError) } } }
自定义错误类型便于分类处理
定义不同的错误类型,可以让中间件做出更精确的响应。
立即学习“go语言免费学习笔记(深入)”;
type NotFoundError struct{ Message string } func (e *NotFoundError) Error() string { return e.Message } type ValidationError struct{ Message string } func (e *ValidationError) Error() string { return e.Message }
在业务逻辑中直接返回这些错误:
func GetUser(w http.ResponseWriter, r *http.Request) error { id := r.URL.Query().Get("id") if id == "" { return &ValidationError{Message: "missing user id"} } // 模拟用户不存在 if id == "999" { return &NotFoundError{Message: "user not found"} } // 正常响应 w.Header().Set("Content-Type", "application/json") json.NewEncoder(w).Encode(map[string]string{"id": id, "name": "Alice"}) return nil }
注册路由时使用中间件包装
将自定义的HandlerFunc通过ErrorMiddleware包装后注册到路由。
http.HandleFunc("/user", ErrorMiddleware(GetUser)) log.Println("Server starting on :8080") log.Fatal(http.ListenAndServe(":8080", nil))
这样,无论是在参数校验、数据查询还是内部逻辑中返回错误,都会被中间件捕获并转换为统一格式的JSON响应。
基本上就这些。这种方式让错误处理集中化,API响应更规范,也方便后续接入日志、监控等系统。不复杂但容易忽略。
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END