87 lines
3.4 KiB
Go
87 lines
3.4 KiB
Go
package fun
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// RouteHandler 自定义 HTTP 路由处理器。
|
|
//
|
|
// 返回 error 时框架统一输出错误响应；返回 nil 视为已自行写回响应——
|
|
// 可直接操作 RouteCtx.RequestCtx 完全自定义状态码与内容
|
|
// （如支付回调要求的纯文本 "success" 应答）。
|
|
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 取查询/表单参数，不存在返回空串
|
|
func (c *RouteCtx) Param(name string) string {
|
|
return c.Data[name]
|
|
}
|
|
|
|
// 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) {
|
|
if handler == nil {
|
|
panic("fun: BindRoute handler cannot be nil")
|
|
}
|
|
method = strings.ToUpper(strings.TrimSpace(method))
|
|
if method == "" {
|
|
panic("fun: BindRoute method cannot be empty")
|
|
}
|
|
if !strings.HasPrefix(path, "/") {
|
|
panic(fmt.Sprintf("fun: BindRoute path %q must start with '/'", path))
|
|
}
|
|
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))
|
|
}
|
|
f.routes[key] = handler
|
|
}
|
|
|
|
// handleRoute 执行自定义路由：合并查询与表单参数（application/x-www-form-urlencoded），
|
|
// 处理器返回 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)
|
|
})
|
|
fastCtx.PostArgs().VisitAll(func(k, v []byte) {
|
|
data[string(k)] = string(v)
|
|
})
|
|
if err := handler(&RouteCtx{RequestCtx: fastCtx, Data: data, Wildcard: wildcard}); err != nil {
|
|
(&Ctx{RequestCtx: fastCtx}).sendError(err)
|
|
}
|
|
}
|