Go语言Web服务中错误处理与响应封装的模式化实践

// 假设 ponyService 和 rainbowService 是处理业务逻辑的服务
// 它们各自的 getAll() 方法返回 (interface{}, error)

func listPonies(w http.ResponseWriter, r *http.Request) {
  ponies, err := ponyService.getAll()
  if err != nil {
      // 重复的错误处理逻辑
      w.Write([]byte(err.Error())) // 直接写入错误信息
      return
  }
  w.Write([]byte(string(ponies))) // 重复的成功响应写入逻辑
}

func listRainbows(w http.ResponseWriter, r *http.Request) {
  rainbows, err := rainbowService.getAll()
  if err != nil {
      // 重复的错误处理逻辑
      w.Write([]byte(err.Error()))
      return
  }
  w.Write([]byte(string(rainbows))) // 重复的成功响应写入逻辑
}

// 辅助函数尝试封装错误处理和响应写入
func handleErrorAndWriteResponse(w http.ResponseWriter, obj interface{}, err error) {
  if err != nil {
      w.Write([]byte(err.Error()))
      return
  }
  // 假设 obj 可以被转换为字符串
  w.Write([]byte(string(obj)))
}

// 尝试调用辅助函数
// func listPonies(w http.ResponseWriter, r *http.Request) {
//   // 这里会遇到问题：ponyService.getAll() 返回两个值 (interface{}, error)，
//   // 但 handleErrorAndWriteResponse 需要三个参数 (http.ResponseWriter, interface{}, error)。
//   // Go语言不允许直接将多返回值作为多个参数传递。
//   handleErrorAndWriteResponse(w, ponyService.getAll())
// }

// appHandler 是一个自定义的HTTP处理函数类型，它返回一个 error。
type appHandler func(http.ResponseWriter, *http.Request) error

// ServeHTTP 方法是 http.Handler 接口的一部分。
// 它包装了实际的业务逻辑处理函数 (fn)，并在 fn 返回错误时进行统一处理。
func (fn appHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    // 调用实际的业务逻辑处理函数
    if err := fn(w, r); err != nil {
        // 如果业务逻辑返回错误，则在这里进行统一的错误处理。
        // 例如，记录错误日志，并向客户端返回一个统一的错误响应。
        // http.Error 会设置 HTTP 状态码为 500 Internal Server Error
        // 并将错误信息写入响应体。
        http.Error(w, err.Error(), http.StatusInternalServerError)
        // 实际应用中，这里还可以添加日志记录，例如：
        // log.Printf("Error processing request %s: %v", r.URL.Path, err)
    }
}

// listPonies 改造后，只关注业务逻辑，并在出错时返回 error。
func listPonies(w http.ResponseWriter, r *http.Request) error {
    ponies, err := ponyService.getAll()
    if err != nil {
        // 业务逻辑函数只返回错误，不直接处理 HTTP 响应
        return err
    }
    // 成功时写入响应
    w.Write([]byte(string(ponies)))
    return nil // 没有错误发生
}

// listRainbows 改造后也遵循同样的模式
func listRainbows(w http.ResponseWriter, r *http.Request) error {
    rainbows, err := rainbowService.getAll()
    if err != nil {
        return err
    }
    w.Write([]byte(string(rainbows)))
    return nil
}

import (
    "log"
    "net/http"
    "github.com/gorilla/mux" // 假设使用 mux 路由器
)

// 模拟服务层
var ponyService = &struct{
    getAll func() (interface{}, error)
}{
    getAll: func() (interface{}, error) {
        // 模拟业务逻辑，可能返回错误
        // return nil, errors.New("failed to get ponies from upstream")
        return "some ponies data", nil
    },
}

var rainbowService = &struct{
    getAll func() (interface{}, error)
}{
    getAll: func() (interface{}, error) {
        // 模拟业务逻辑，可能返回错误
        return "some rainbows data", nil
    },
}


func init() {
    m := mux.NewRouter()
    // 使用 appHandler 包装业务逻辑函数
    m.Handle("/ponies", appHandler(listPonies))
    m.Handle("/rainbows", appHandler(listRainbows))
    http.Handle("/", m) // 将 mux 路由器注册到默认 HTTP ServeMux
}

func main() {
    log.Println("Server starting on :8080")
    // 启动 HTTP 服务器
    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatalf("Server failed to start: %v", err)
    }
}