高效URL路径模式匹配与变量提取教程

package main

import (
    "fmt"
    "regexp"
    "strings"
)

// MatchURLPattern 匹配URL路径并提取变量
// pattern: 包含占位符的URL模式，例如 "/some/{tag}/here"
// urlPath: 待匹配的实际URL路径，例如 "/some/text/here"
// 返回值: 包含提取变量的map和是否匹配成功的布尔值
func MatchURLPattern(pattern string, urlPath string) (map[string]string, bool) {
    // 1. 将模式字符串转换为正则表达式
    // 查找所有 {variable} 形式的占位符
    rePlaceholder := regexp.MustCompile(`\{([a-zA-Z0-9_]+)\}`)

    // 将模式中的 {variable} 替换为命名捕获组 `(?P<variable>[^/]+)`
    // `[^/]+` 表示匹配一个或多个非斜杠字符，通常用于匹配URL路径段
    regexPattern := rePlaceholder.ReplaceAllStringFunc(pattern, func(s string) string {
        // s 的形式为 "{tag}"
        variableName := s[1 : len(s)-1] // 提取 "tag"
        return fmt.Sprintf("(?P<%s>[^/]+)", variableName)
    })

    // 确保正则表达式匹配整个字符串，加上 ^ 和 $
    finalRegexPattern := "^" + regexPattern + "$"

    // 编译正则表达式
    re, err := regexp.Compile(finalRegexPattern)
    if err != nil {
        fmt.Printf("Error compiling regex for pattern '%s': %v\n", pattern, err)
        return nil, false
    }

    // 2. 执行匹配
    matches := re.FindStringSubmatch(urlPath)
    if matches == nil {
        return nil, false // 不匹配
    }

    // 3. 提取变量
    params := make(map[string]string)
    for i, name := range re.SubexpNames() {
        // 索引0是整个匹配的字符串，空名称的组是未命名的捕获组，我们只关心命名组
        if i != 0 && name != "" { 
            params[name] = matches[i]
        }
    }

    return params, true
}

func main() {
    patternA := "/some/{tag}/here"
    urlB := "/some/text/here"
    urlC := "/some/another-text/here"
    urlD := "/some/text/not-here" // 不匹配的URL
    patternE := "/users/{userID}/posts/{postID}"
    urlF := "/users/123/posts/abc"

    fmt.Printf("匹配模式 '%s' 与 URL '%s':\n", patternA, urlB)
    params, matched := MatchURLPattern(patternA, urlB)
    if matched {
        fmt.Printf("  匹配成功！提取参数: %v\n", params) // 输出: map[tag:text]
    } else {
        fmt.Println("  不匹配。")
    }

    fmt.Printf("\n匹配模式 '%s' 与 URL '%s':\n", patternA, urlC)
    params, matched = MatchURLPattern(patternA, urlC)
    if matched {
        fmt.Printf("  匹配成功！提取参数: %v\n", params) // 输出: map[tag:another-text]
    } else {
        fmt.Println("  不匹配。")
    }

    fmt.Printf("\n匹配模式 '%s' 与 URL '%s':\n", patternA, urlD)
    params, matched = MatchURLPattern(patternA, urlD)
    if matched {
        fmt.Printf("  匹配成功！提取参数: %v\n", params)
    } else {
        fmt.Println("  不匹配。") // 输出: 不匹配。
    }

    fmt.Printf("\n匹配模式 '%s' 与 URL '%s':\n", patternE, urlF)
    params, matched = MatchURLPattern(patternE, urlF)
    if matched {
        fmt.Printf("  匹配成功！提取参数: %v\n", params) // 输出: map[postID:abc userID:123]
    } else {
        fmt.Println("  不匹配。")
    }
}