diff --git a/fun.go b/fun.go index 33872e4..bad5a0d 100644 --- a/fun.go +++ b/fun.go @@ -9,11 +9,19 @@ import ( ) type Fun struct { - methods map[string]methodInfo - routes map[string]RouteHandler // 自定义路由:"GET /path" → 处理器 - boxes *sync.Map // 依赖容器:reflect.Type → reflect.Value - guards []*any // 全局 Guard - serviceGuards map[string][]*any // 服务级 Guard,按服务名 + methods map[string]methodInfo + routes map[string]RouteHandler // 自定义路由:"GET /path" → 处理器(精确匹配) + wildcardRoutes map[string][]wildcardRoute + boxes *sync.Map // 依赖容器:reflect.Type → reflect.Value + guards []*any // 全局 Guard + serviceGuards map[string][]*any // 服务级 Guard,按服务名 +} + +// wildcardRoute 通配符路由(BindRoute path 以 "/*" 结尾注册): +// prefix 如 "/image",匹配 prefix 与 prefix 下任意子路径 +type wildcardRoute struct { + prefix string + handler RouteHandler } var ( @@ -33,10 +41,11 @@ type methodInfo struct { func New() *Fun { f := &Fun{ - methods: map[string]methodInfo{}, - routes: map[string]RouteHandler{}, - boxes: &sync.Map{}, - serviceGuards: map[string][]*any{}, + methods: map[string]methodInfo{}, + routes: map[string]RouteHandler{}, + wildcardRoutes: map[string][]wildcardRoute{}, + boxes: &sync.Map{}, + serviceGuards: map[string][]*any{}, } if fun == nil { fun = f diff --git a/handle.go b/handle.go index 7afef44..8f6aa40 100644 --- a/handle.go +++ b/handle.go @@ -7,19 +7,27 @@ import ( "fmt" "reflect" "runtime/debug" + "strings" "github.com/valyala/fasthttp" ) -// handle 处理 HTTP 请求:先匹配自定义路由(BindRoute),未命中走 /cell RPC +// handle 处理 HTTP 请求:先匹配自定义路由(BindRoute 精确/通配),未命中走 /cell RPC func (f *Fun) handle(fastCtx *fasthttp.RequestCtx) { ctx := &Ctx{RequestCtx: fastCtx} defer f.handlePanic(ctx) - if handler, ok := f.routes[string(fastCtx.Method())+" "+string(fastCtx.Path())]; ok { - f.handleRoute(fastCtx, handler) + method, path := string(fastCtx.Method()), string(fastCtx.Path()) + if handler, ok := f.routes[method+" "+path]; ok { + f.handleRoute(fastCtx, handler, "") return } + for _, r := range f.wildcardRoutes[method] { + if path == r.prefix || strings.HasPrefix(path, r.prefix+"/") { + f.handleRoute(fastCtx, r.handler, strings.TrimPrefix(path, r.prefix+"/")) + return + } + } if ctx.path() != "/cell" { ctx.setStatusCode(fasthttp.StatusNotFound) diff --git a/route.go b/route.go index 229e04f..a9f83b6 100644 --- a/route.go +++ b/route.go @@ -16,10 +16,12 @@ type RouteHandler func(ctx *RouteCtx) error // RouteCtx 自定义路由上下文:Data 合并了 URL 查询参数与 POST 表单参数(表单优先), // 支付回调等第三方以 form-urlencoded 回调的场景可直接 Param 取值。 +// Wildcard 为通配符路由(/prefix/*)匹配到的剩余路径(不含前导 "/")。 // 独立于服务内嵌的 Ctx:后者辅助方法刻意全小写以防混入 RPC 方法集,路由不复用该类型。 type RouteCtx struct { RequestCtx *fasthttp.RequestCtx Data map[string]string + Wildcard string } // Param 取查询/表单参数,不存在返回空串 @@ -27,10 +29,12 @@ func (c *RouteCtx) Param(name string) string { return c.Data[name] } -// BindRoute 注册自定义路由(方法大小写不敏感 + 精确路径匹配),用于 GET 直链、 -// 健康检查、支付回调等无法走 POST /cell RPC 的场景。 +// BindRoute 注册自定义路由(方法大小写不敏感;path 精确匹配,或以 "/*" 结尾做前缀通配), +// 用于 GET 直链、健康检查、支付回调等无法走 POST /cell RPC 的场景。 // // - path 必须以 "/" 开头;/cell 为 RPC 保留路径,不可注册 +// - 通配符形式如 "/image/*":匹配 "/image/a/b.png" 等任意子路径, +// 匹配到的剩余路径(去掉前导 "/",如 "a/b.png")经 RouteCtx.Wildcard 取出 // - 同一 方法+路径 重复注册直接 panic // - 与 BindService 一致,需在 Start 前完成注册(启动阶段单线程) func (f *Fun) BindRoute(method, path string, handler RouteHandler) { @@ -44,9 +48,21 @@ func (f *Fun) BindRoute(method, path string, handler RouteHandler) { if !strings.HasPrefix(path, "/") { panic(fmt.Sprintf("fun: BindRoute path %q must start with '/'", path)) } - if path == "/cell" { + if path == "/cell" || path == "/cell/*" { panic("fun: /cell is reserved for RPC") } + if prefix, ok := strings.CutSuffix(path, "/*"); ok { + if prefix == "" || strings.HasSuffix(prefix, "/") { + panic(fmt.Sprintf("fun: BindRoute wildcard path %q invalid (no trailing '/' allowed before /*)", path)) + } + for _, r := range f.wildcardRoutes[method] { + if r.prefix == prefix { + panic(fmt.Sprintf("fun: route %s %s/* already bound", method, prefix)) + } + } + f.wildcardRoutes[method] = append(f.wildcardRoutes[method], wildcardRoute{prefix: prefix, handler: handler}) + return + } key := method + " " + path if _, exists := f.routes[key]; exists { panic(fmt.Sprintf("fun: route %s already bound", key)) @@ -55,8 +71,8 @@ func (f *Fun) BindRoute(method, path string, handler RouteHandler) { } // handleRoute 执行自定义路由:合并查询与表单参数(application/x-www-form-urlencoded), -// 处理器返回 error 时按统一 Result 格式输出错误响应 -func (f *Fun) handleRoute(fastCtx *fasthttp.RequestCtx, handler RouteHandler) { +// 处理器返回 error 时按统一 Result 格式输出错误响应;wildcard 为通配路由匹配的剩余路径 +func (f *Fun) handleRoute(fastCtx *fasthttp.RequestCtx, handler RouteHandler, wildcard string) { data := map[string]string{} fastCtx.QueryArgs().VisitAll(func(k, v []byte) { data[string(k)] = string(v) @@ -64,7 +80,7 @@ func (f *Fun) handleRoute(fastCtx *fasthttp.RequestCtx, handler RouteHandler) { fastCtx.PostArgs().VisitAll(func(k, v []byte) { data[string(k)] = string(v) }) - if err := handler(&RouteCtx{RequestCtx: fastCtx, Data: data}); err != nil { + if err := handler(&RouteCtx{RequestCtx: fastCtx, Data: data, Wildcard: wildcard}); err != nil { (&Ctx{RequestCtx: fastCtx}).sendError(err) } }