72 lines
1.5 KiB
Go
72 lines
1.5 KiB
Go
package server
|
||
|
||
import (
|
||
"net/url"
|
||
"regexp"
|
||
"strings"
|
||
|
||
"git.pyer.club/kingecg/gologger"
|
||
)
|
||
|
||
type UrlParamMatcher struct {
|
||
Params []string
|
||
Reg *regexp.Regexp
|
||
}
|
||
|
||
// 解析URL函数
|
||
// 参数：
|
||
//
|
||
// u：待解析的URL字符串
|
||
//
|
||
// 返回值：
|
||
//
|
||
// 解析后的UrlParamMatcher结构体
|
||
func ParseUrl(u string) UrlParamMatcher {
|
||
ret := UrlParamMatcher{}
|
||
|
||
uo, _ := url.Parse(u)
|
||
|
||
// 判断路径是否非空且非根路径
|
||
if uo.Path != "" && uo.Path != "/" {
|
||
// 将路径按斜杠分割成切片
|
||
us := strings.Split(uo.Path, "/")
|
||
for index, v := range us {
|
||
// 判断是否以冒号开头
|
||
if strings.HasPrefix(v, ":") {
|
||
// 去除冒号前缀
|
||
param := strings.TrimPrefix(v, ":")
|
||
// 将参数添加到Params切片中
|
||
ret.Params = append(ret.Params, param)
|
||
// 将参数名作为正则表达式的命名捕获组
|
||
us[index] = "(?P<" + param + ">.*)"
|
||
}
|
||
}
|
||
// 如果存在参数，则将路径编译为正则表达式
|
||
if len(ret.Params) > 0 {
|
||
ret.Reg, _ = regexp.Compile("^" + strings.Join(us, "/") + "$")
|
||
}
|
||
}
|
||
|
||
return ret
|
||
}
|
||
|
||
func MatchUrlParam(u string, matcher *UrlParamMatcher) (map[string]string, bool) {
|
||
l := gologger.GetLogger("URLMatcher")
|
||
l.Debug("Match for ", u)
|
||
if matcher.Reg != nil {
|
||
uo, _ := url.Parse(u)
|
||
if uo.Path == "" || uo.Path == "/" {
|
||
return nil, false
|
||
}
|
||
matches := matcher.Reg.FindStringSubmatch(uo.Path)
|
||
if len(matches) > 0 {
|
||
params := make(map[string]string)
|
||
for i, name := range matcher.Params {
|
||
params[name] = matches[i+1]
|
||
}
|
||
return params, true
|
||
}
|
||
}
|
||
return nil, false
|
||
}
|