Go HTTP 路由进阶：实现自定义正则表达式路由

package main

import (
    "log"
    "net/http"
    "regexp"
    "fmt"
)

// route 结构体定义了路由规则，包含一个正则表达式模式和对应的处理器
type route struct {
    pattern *regexp.Regexp
    handler http.Handler
}

// RegexpHandler 是一个自定义的 HTTP 处理器，它维护一个路由规则列表
type RegexpHandler struct {
    routes []*route
}

// Handler 方法用于注册一个带有正则表达式模式和 http.Handler 的路由
func (h *RegexpHandler) Handler(pattern *regexp.Regexp, handler http.Handler) {
    h.routes = append(h.routes, &route{pattern, handler})
}

// HandleFunc 方法用于注册一个带有正则表达式模式和处理函数（func）的路由
func (h *RegexpHandler) HandleFunc(pattern *regexp.Regexp, handler func(http.ResponseWriter, *http.Request)) {
    h.routes = append(h.routes, &route{pattern, http.HandlerFunc(handler)})
}

// ServeHTTP 是 RegexpHandler 实现 http.Handler 接口的核心方法
// 它负责根据请求的 URL 路径匹配注册的正则表达式，并调用相应的处理器
func (h *RegexpHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    for _, route := range h.routes {
        // 检查当前路由的正则表达式是否匹配请求的 URL 路径
        if route.pattern.MatchString(r.URL.Path) {
            // 如果匹配，则调用对应的处理器的 ServeHTTP 方法，并立即返回
            route.handler.ServeHTTP(w, r)
            return
        }
    }
    // 如果所有模式都未匹配，则返回 404 Not Found 响应
    http.NotFound(w, r)
}

func main() {
    // 创建一个 RegexpHandler 实例
    router := new(RegexpHandler)

    // 注册一个处理 /groups/{groupID}/people 模式的路由
    // 例如：/groups/123/people, /groups/abc/people
    router.HandleFunc(regexp.MustCompile("^/groups/([a-zA-Z0-9]+)/people$"), func(w http.ResponseWriter, r *http.Request) {
        // 从 URL 路径中提取 groupID
        matches := regexp.MustCompile("^/groups/([a-zA-Z0-9]+)/people$").FindStringSubmatch(r.URL.Path)
        if len(matches) > 1 {
            groupID := matches[1]
            fmt.Fprintf(w, "Fetching people for group: %s\n", groupID)
        } else {
            http.Error(w, "Invalid group ID", http.StatusBadRequest)
        }
    })

    // 注册一个处理 /users/{userID} 模式的路由
    // 例如：/users/456
    router.HandleFunc(regexp.MustCompile("^/users/([0-9]+)$"), func(w http.ResponseWriter, r *http.Request) {
        matches := regexp.MustCompile("^/users/([0-9]+)$").FindStringSubmatch(r.URL.Path)
        if len(matches) > 1 {
            userID := matches[1]
            fmt.Fprintf(w, "Fetching user with ID: %s\n", userID)
        } else {
            http.Error(w, "Invalid user ID", http.StatusBadRequest)
        }
    })

    // 注册一个简单的根路径路由
    router.HandleFunc(regexp.MustCompile("^/$"), func(w http.ResponseWriter, r *http.Request) {
        fmt.Fprintf(w, "Welcome to the homepage!\n")
    })

    // 将自定义路由器绑定到 HTTP 服务器
    log.Println("Server starting on :8080")
    err := http.ListenAndServe(":8080", router)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}