少提交的东西

internal/routes/registry.go

@@ -0,0 +1,43 @@
+package routes
+
+import (
+	"regexp"
+
+	"github.com/gofiber/fiber/v2"
+
+	"github.com/break/junhong_cmp_fiber/pkg/openapi"
+)
+
+// RouteSpec 定义接口文档元数据
+type RouteSpec struct {
+	Summary string
+	Input   interface{} // 请求参数结构体 (Query/Path/Body)
+	Output  interface{} // 响应参数结构体
+	Tags    []string
+	Auth    bool // 是否需要认证图标 (预留)
+}
+
+// pathParamRegex 用于匹配 Fiber 的路径参数格式 /:param
+var pathParamRegex = regexp.MustCompile(`/:([a-zA-Z0-9_]+)`)
+
+// Register 封装后的注册函数
+// router: Fiber 路由组
+// doc: 文档生成器 (如果在运行 Web 服务时为 nil，在生成文档时为非 nil)
+// basePath: 当前路由组的基础路径 (用于文档生成)
+// method, path: HTTP 方法和路径
+// handler: Fiber Handler
+// spec: 文档元数据
+func Register(router fiber.Router, doc *openapi.Generator, basePath, method, path string, handler fiber.Handler, spec RouteSpec) {
+	// 1. 注册实际的 Fiber 路由
+	router.Add(method, path, handler)
+
+	// 2. 注册文档 (如果 doc 不为空 - 也就是在生成文档模式下)
+	if doc != nil {
+		// 简单的路径拼接
+		fullPath := basePath + path
+		// 将 Fiber 路由参数格式 /:id 转换为 OpenAPI 格式 /{id}
+		openapiPath := pathParamRegex.ReplaceAllString(fullPath, "/{$1}")
+
+		doc.AddOperation(method, openapiPath, spec.Summary, spec.Input, spec.Output, spec.Tags...)
+	}
+}