80 lines
2.2 KiB
Go
80 lines
2.2 KiB
Go
|
// Package router ...
|
||
|
|
//
|
||
|
|
// Description : router ...
|
||
|
|
//
|
||
|
|
// Author : go_developer@163.com<白茶清欢>
|
||
|
|
//
|
||
|
|
// Date : 2023-03-03 16:48
|
||
|
|
package router
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"net/http"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"github.com/gin-gonic/gin"
|
||
|
|
)
|
||
|
|
|
||
|
|
// HandleRegisterRouter 注册gin路由
|
||
|
|
//
|
||
|
|
// Author : go_developer@163.com<白茶清欢>
|
||
|
|
//
|
||
|
|
// Date : 8:36 下午 2021/3/9
|
||
|
|
func HandleRegisterRouter(router *gin.Engine, method string, uri string, handler gin.HandlerFunc, middlewareList []gin.HandlerFunc) error {
|
||
|
|
if nil == middlewareList {
|
||
|
|
middlewareList = make([]gin.HandlerFunc, 0)
|
||
|
|
}
|
||
|
|
switch strings.ToUpper(method) {
|
||
|
|
case http.MethodGet:
|
||
|
|
router.GET(uri, handler).Use(middlewareList...)
|
||
|
|
case http.MethodPost:
|
||
|
|
router.POST(uri, handler).Use(middlewareList...)
|
||
|
|
case http.MethodDelete:
|
||
|
|
router.DELETE(uri, handler).Use(middlewareList...)
|
||
|
|
case http.MethodHead:
|
||
|
|
router.HEAD(uri, handler).Use(middlewareList...)
|
||
|
|
case http.MethodOptions:
|
||
|
|
router.OPTIONS(uri, handler).Use(middlewareList...)
|
||
|
|
case http.MethodPatch:
|
||
|
|
router.PATCH(uri, handler).Use(middlewareList...)
|
||
|
|
case http.MethodPut:
|
||
|
|
router.PUT(uri, handler).Use(middlewareList...)
|
||
|
|
case "ANY": // 一次性注册全部请求方法的路由
|
||
|
|
router.Any(uri, handler).Use(middlewareList...)
|
||
|
|
default:
|
||
|
|
// 不是一个函数,数名method配置错误
|
||
|
|
return fmt.Errorf("uri=%s method=%s 请求方法配置错误", uri, method)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
// RegisterRouterGroup 注册gin路由
|
||
|
|
//
|
||
|
|
// Author : go_developer@163.com<白茶清欢>
|
||
|
|
//
|
||
|
|
// Date : 8:36 下午 2021/3/9
|
||
|
|
func RegisterRouterGroup(router *gin.RouterGroup, method string, uri string, handler gin.HandlerFunc) error {
|
||
|
|
switch strings.ToUpper(method) {
|
||
|
|
case http.MethodGet:
|
||
|
|
router.GET(uri, handler)
|
||
|
|
case http.MethodPost:
|
||
|
|
router.POST(uri, handler)
|
||
|
|
case http.MethodDelete:
|
||
|
|
router.DELETE(uri, handler)
|
||
|
|
case http.MethodHead:
|
||
|
|
router.HEAD(uri, handler)
|
||
|
|
case http.MethodOptions:
|
||
|
|
router.OPTIONS(uri, handler)
|
||
|
|
case http.MethodPatch:
|
||
|
|
router.PATCH(uri, handler)
|
||
|
|
case http.MethodPut:
|
||
|
|
router.PUT(uri, handler)
|
||
|
|
case "ANY": // 一次性注册全部请求方法的路由
|
||
|
|
router.Any(uri, handler)
|
||
|
|
default:
|
||
|
|
// 不是一个函数,数名method配置错误
|
||
|
|
return fmt.Errorf("uri=%s method=%s 请求方法配置错误", uri, method)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|