在Go HTTP路由中实现基于正则表达式的灵活匹配

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

// route 结构体用于存储正则表达式模式和对应的HTTP处理程序。
type route struct {
    pattern *regexp.Regexp // 编译后的正则表达式模式
    handler http.Handler   // 对应的HTTP处理程序
}

// 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 方法用于注册一个 http.HandlerFunc 到指定的正则表达式模式。
// 它会将 http.HandlerFunc 适配为 http.Handler。
func (h *RegexpHandler) HandleFunc(pattern *regexp.Regexp, handler func(http.ResponseWriter, *http.Request)) {
    h.routes = append(h.routes, &route{pattern, http.HandlerFunc(handler)})
}

// ServeHTTP 方法实现了 http.Handler 接口。
// 它遍历注册的路由，使用正则表达式匹配请求的 URL 路径。
// 找到第一个匹配的路由后，调用其对应的处理程序。
// 如果没有匹配的路由，则返回 404 Not Found。
func (h *RegexpHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
    for _, route := range h.routes {
        if route.pattern.MatchString(r.URL.Path) { // 使用正则表达式匹配请求路径
            route.handler.ServeHTTP(w, r) // 匹配成功，调用对应的处理函数
            return
        }
    }
    // 没有模式匹配，发送 404 响应
    http.NotFound(w, r)
}

// 示例处理函数
func groupPeopleHandler(w http.ResponseWriter, r *http.Request) {
    // 假设模式是 ^/groups/([^/]+)/people$
    // 这里需要再次使用正则表达式来提取路径参数
    re := regexp.MustCompile("^/groups/([^/]+)/people$")
    matches := re.FindStringSubmatch(r.URL.Path)

    if len(matches) > 1 {
        groupID := matches[1] // 获取第一个捕获组，即通配符匹配到的部分
        fmt.Fprintf(w, "Handling request for people in group: %s\n", groupID)
    } else {
        http.NotFound(w, r) // 理论上不会发生，因为只有匹配的请求才会进入此处理函数
    }
}

func userProfileHandler(w http.ResponseWriter, r *http.Request) {
    // 假设模式是 ^/users/([0-9]+)$
    re := regexp.MustCompile("^/users/([0-9]+)$")
    matches := re.FindStringSubmatch(r.URL.Path)

    if len(matches) > 1 {
        userID := matches[1]
        fmt.Fprintf(w, "Handling request for user ID: %s\n", userID)
    } else {
        http.NotFound(w, r)
    }
}

func homeHandler(w http.ResponseWriter, r *http.Request) {
    fmt.Fprintf(w, "Welcome to the home page!\n")
}

func main() {
    // 创建自定义的正则表达式路由器
    mux := new(RegexpHandler)

    // 注册路由和处理函数
    // 注意：正则表达式需要完整匹配整个路径，所以通常以 ^ 开头和 $ 结尾
    mux.HandleFunc(regexp.MustCompile("^/$"), homeHandler) // 匹配根路径
    mux.HandleFunc(regexp.MustCompile("^/groups/([^/]+)/people$"), groupPeopleHandler) // 匹配 /groups/anything/people
    mux.HandleFunc(regexp.MustCompile("^/users/([0-9]+)$"), userProfileHandler) // 匹配 /users/123

    fmt.Println("Server listening on :8080...")
    log.Fatal(http.ListenAndServe(":8080", mux)) // 将自定义路由器作为参数传入
}