Go 框架的灵活性与可扩展性

import (
    "fmt"
    "net/http"
)

// AuthMiddleware 检查请求中的授权标头是否有效。
func AuthMiddleware(next http.Handler) http.Handler {
    return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        // 检查授权标头
        token := r.Header.Get("Authorization")
        if !IsValidToken(token) {
            // 未授权，返回错误响应
            http.Error(w, "Unauthorized", http.StatusUnauthorized)
            return
        }

        // 授权有效，调用下一个中间件或处理器
        next.ServeHTTP(w, r)
    })
}

import (
    "database/sql"
    "fmt"
)

// UserDB 是一个实现了 UserDAO 接口的数据库存储后端。
type UserDB struct {
    db *sql.DB
}

// NewUserDB 返回一个新的 UserDB 实例。
func NewUserDB(db *sql.DB) *UserDB {
    return &UserDB{db}
}

// FindByID 通过其 ID 查找用户。
func (u *UserDB) FindByID(id int) (*User, error) {
    // 从数据库中查询用户
    row := u.db.QueryRow("SELECT * FROM users WHERE id = ?", id)
    var user User
    err := row.Scan(&user.ID, &user.Name)
    return &user, err
}

package main

import (
    "fmt"
    "net/http"

    "github.com/gorilla/mux"
    "github.com/your-module/user-db"
)

// User 是用户模型的表示。
type User struct {
    ID   int    `json:"id"`
    Name string `json:"name"`
}

// UserHandler 处理用户 API 请求。
type UserHandler struct {
    db userdb.UserDAO
}

// NewUserHandler 返回一个新的 UserHandler 实例。
func NewUserHandler(db userdb.UserDAO) *UserHandler {
    return &UserHandler{db}
}

// FindByID 处理用于获取指定 ID 的用户的 GET 请求。
func (h *UserHandler) FindByID(w http.ResponseWriter, r *http.Request) {
    // 从请求参数中获取 ID
    params := mux.Vars(r)
    id, err := strconv.Atoi(params["id"])
    if err != nil {
        http.Error(w, "Invalid ID", http.StatusBadRequest)
        return
    }

    // 使用存储后端查找用户
    user, err := h.db.FindByID(id)
    if err != nil {
        http.Error(w, "User not found", http.StatusNotFound)
        return
    }

    // 响应找到的用户
    json.NewEncoder(w).Encode(user)
}

// main 函数初始化路由器、中间件和存储后端。
func main() {
    // 初始化路由器
    r := mux.NewRouter()

    // 添加用于身份验证的中间件
    r.Use(AuthMiddleware)

    // 创建用户存储后端
    db := NewUserDB(connectToDatabase())

    // 创建用户处理器
    handler := NewUserHandler(db)

    // 注册用户 API 路由
    r.HandleFunc("/users/{id}", handler.FindByID).Methods("GET")

    // 启动 HTTP 服务器
    http.ListenAndServe(":8080", r)
}