搭建自定义错误处理框架以增强Golang函数的错误处理能力

type DatabaseError struct {
    Message string
    Query string
    Code int
}

func (e DatabaseError) Error() string {
    return fmt.Sprintf("DatabaseError: %s (query: %s, code: %d)", e.Message, e.Query, e.Code)
}

func HandleDatabaseError(w http.ResponseWriter, r *httpRequest, err error) {
    if dbErr, ok := err.(DatabaseError); ok {
        switch dbErr.Code {
        case 404:
            http.NotFound(w, r)
        case 500:
            http.InternalServerError(w, r)
        default:
            http.BadRequest(w, r)
        }
    } else {
        http.Error(w, "Internal Server Error", 500)
    }
}

func QueryDatabase(query string) (result []string, err error) {
    // 执行数据库查询

    if err != nil {
        if dbErr, ok := err.(DatabaseError); ok {
            return nil, HandleDatabaseError(err)
        }
        return nil, err
    }

    // 处理成功的结果

    return result, nil
}

package main

import (
    "log"
    "net/http"
    // 自定義的錯誤處理器及其他代碼
)

func main() {
    http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
        result, err := QueryDatabase("SELECT * FROM users")
        if err != nil {
            // 根據錯誤，產生不同的 HTTP 狀態碼
            HandleDatabaseError(w, r, err)
        } else {
            // 處理成功的結果
            fmt.Fprintf(w, "Users: %+v", result)
        }
    })
    log.Fatal(http.ListenAndServe(":8080", nil))
}