支持解析路由前缀

router/register.go

@@ -1,115 +1,69 @@
 // Package router ...
 //
-// Description : 注册路由
+// Description : router ...
 //
 // Author : go_developer@163.com<白茶清欢>
 //
-// Date : 2021-03-26 2:13 下午
+// Date : 2024-07-20 21:39
 package router
 
 import (
-	"log"
+	"fmt"
 	"reflect"
 
-	"github.com/pkg/errors"
-
 	"github.com/gin-gonic/gin"
 )
 
 var (
-	// DebugLogEnable 默认打开debug日志
-	DebugLogEnable = true
+	ginRouter = gin.Default()
 )
 
-// DisableDebugLog 禁用debug日志
-//
-// Author : go_developer@163.com<白茶清欢>
-//
-// Date : 2:17 下午 2021/3/26
-func DisableDebugLog() {
-	DebugLogEnable = false
+func init() {
+
 }
 
-// RegisterRouter 注册一个路由
+// Register 注册路由
 //
 // Author : go_developer@163.com<白茶清欢>
 //
-// Date : 2:14 下午 2021/3/26
-func RegisterRouter(router *gin.Engine, apiInstanceList ...any) error {
-	for _, apiInstance := range apiInstanceList {
-		if nil == apiInstance {
+// Date : 21:40 2024/7/20
+func Register(port int, controllerList ...any) {
+	for _, controller := range controllerList {
+		if nil == controller {
+			// 忽略空指针
 			continue
 		}
-		val := reflect.ValueOf(apiInstance)
-		switch val.Kind() {
-		case reflect.Struct:
-			fallthrough
-		case reflect.Ptr:
-			api, ok := apiInstance.(IApi)
-			if ok {
-				if err := HandleRegisterRouter(router, api.GetMethod(), api.GetURI(), api.GetHandler(), api.GetMiddleWareList()); nil != err {
-					routerLog(err.Error())
-					return err
-				}
-				continue
-			}
-			routerLog(val.String() + "结构体或者结构体指针, 自动识别函数是否包含RouterFunc")
-			// 不是IApi接口,自动识别函数列表 RouterFunc 函数自动注册
-			methodCnt := val.NumMethod()
-			for i := 0; i < methodCnt; i++ {
-				// TODO : 识别函数本身是不是 RouterFunc
-				af, o := val.Method(i).Interface().(func() (string, string, gin.HandlerFunc, []gin.HandlerFunc))
-				if o {
-					method, uri, handler, middlewareList := af()
-					if err := HandleRegisterRouter(router, method, uri, handler, middlewareList); nil != err {
-						routerLog(err.Error())
-						return err
-					}
-					continue
-				}
-				apiFuncList := val.Method(i).Call(nil)
-				for _, apiFuncVal := range apiFuncList {
-					apiFunc, ok := apiFuncVal.Interface().(RegisterFunc)
-					if !ok {
-						continue
-					}
-					method, uri, handler, middlewareList := apiFunc()
-					if err := HandleRegisterRouter(router, method, uri, handler, middlewareList); nil != err {
-						routerLog(err.Error())
-						return err
-					}
-				}
-
-			}
-		case reflect.Func:
-			api, ok := apiInstance.(RegisterFunc)
-			if !ok {
-				err := errors.New("函数方式注册路由必须是 RouterFunc")
-				routerLog(err.Error())
-				return err
-			}
-			method, uri, handler, middlewareList := api()
-			if err := HandleRegisterRouter(router, method, uri, handler, middlewareList); nil != err {
-				routerLog(err.Error())
-				return err
-			}
-		default:
-			err := errors.New("注册的路由必须是 IApi 或者 RouterFunc 或者 包含 RouterFunc 的结构体")
-			routerLog(err.Error())
-			return err
-		}
 	}
-	return nil
 }
 
-// routerLog 记录日志
+// parseController 解析controller
 //
 // Author : go_developer@163.com<白茶清欢>
 //
-// Date : 2:28 下午 2021/3/26
-func routerLog(msg string) {
-	if !DebugLogEnable || len(msg) == 0 {
-		return
+// Date : 22:10 2024/7/20
+func parseController(controller any) {
+	controllerType := reflect.TypeOf(controller)
+	controllerValue := reflect.ValueOf(controller)
+	routerPrefix := "/"
+	// 解析路由前缀函数
+	// routerPrefix 不能有任何入参, 并且只能有一个返回值,
+	// 返回值类型为字符串, 为具体路由前缀
+	routerPrefixFunc, routerPrefixFuncExist := controllerType.MethodByName(PrefixFuncName)
+	if routerPrefixFuncExist {
+		routerPrefixFuncType := routerPrefixFunc.Type
+		if routerPrefixFuncType.NumIn() == 1 && // 无任何入参, 正在没有如何入参情况下, 第一个参数是结构体指针
+			routerPrefixFuncType.NumOut() == 1 && // 只能有一个返回值
+			routerPrefixFuncType.Out(0).Kind() == reflect.String { // 返回值必须是字符串
+			routerPrefix = routerPrefixFunc.Func.Call([]reflect.Value{controllerValue})[0].String()
+			// .Call(nil)[0].String()
+		}
+		// 请求组的中间件
+		/*routerMiddlewareFunc := controllerType.MethodByName(MiddlewareFuncName)
+		routerMiddlewareFuncType := routerMiddlewareFunc.Type()
+		if routerMiddlewareFuncType.NumIn() == 0 && // 无需任何参数
+			routerMiddlewareFuncType.NumOut() == 1 && // 只能有一个返回值
+			routerMiddlewareFuncType.Out(0).String() == "[]gin.HandlerFunc" { // 返回值必须是gin.HandlerFunc
+		}*/
 	}
-	log.Print(msg)
+	fmt.Println(routerPrefix)
 }