mirror of
https://github.com/gmsec/gmsec.git
synced 2025-10-03 07:02:21 +08:00
73 lines
2.5 KiB
Go
73 lines
2.5 KiB
Go
package routers
|
|
|
|
import (
|
|
"example/internal/api"
|
|
"fmt"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
"github.com/xxjwxc/ginrpc"
|
|
"github.com/xxjwxc/public/dev"
|
|
"github.com/xxjwxc/public/tools"
|
|
)
|
|
|
|
// OnInitRouter 默认初始化
|
|
func OnInitRouter(router gin.IRouter, objs ...interface{}) {
|
|
InitFunc(router)
|
|
InitObj(router, objs...)
|
|
}
|
|
|
|
// InitFunc 默认初始化函数
|
|
func InitFunc(router gin.IRouter) {
|
|
router.StaticFS("/file", http.Dir(tools.GetCurrentDirectory()+"/file")) //加载静态资源，一般是上传的资源，例如用户上传的图片
|
|
}
|
|
|
|
// InitObj 初始化对象
|
|
func InitObj(router gin.IRouter, objs ...interface{}) {
|
|
base := ginrpc.New(ginrpc.WithCtx(api.NewAPIFunc), ginrpc.WithOutDoc(dev.IsDev()), ginrpc.WithDebug(dev.IsDev()), ginrpc.WithOutPath("internal/routers"),
|
|
ginrpc.WithBeforeAfter(&ginrpc.DefaultGinBeforeAfter{})) // 基础信息注册
|
|
// objs = append(objs, new(hello.Hello)) // protoc service 注册
|
|
|
|
// base.OutDoc(true) // 输出文档
|
|
base.Register(router, objs...) // 对象注册
|
|
}
|
|
|
|
// Cors 跨域问题
|
|
func Cors() gin.HandlerFunc {
|
|
return func(c *gin.Context) {
|
|
method := c.Request.Method
|
|
|
|
origin := c.Request.Header.Get("Origin")
|
|
var headerKeys []string
|
|
for k := range c.Request.Header {
|
|
headerKeys = append(headerKeys, k)
|
|
}
|
|
headerStr := strings.Join(headerKeys, ", ")
|
|
if headerStr != "" {
|
|
headerStr = fmt.Sprintf("access-control-allow-origin, access-control-allow-headers, %s", headerStr)
|
|
} else {
|
|
headerStr = "access-control-allow-origin, access-control-allow-headers"
|
|
}
|
|
if origin != "" {
|
|
//下面的都是乱添加的-_-~
|
|
// c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
|
|
c.Header("Access-Control-Allow-Origin", "*")
|
|
c.Header("Access-Control-Allow-Headers", headerStr)
|
|
c.Header("Access-Control-Allow-Methods", "POST, GET, OPTIONS, PUT, DELETE")
|
|
// c.Header("Access-Control-Allow-Headers", "Authorization, Content-Length, X-CSRF-Token, Accept, Origin, Host, Connection, Accept-Encoding, Accept-Language,DNT, X-CustomHeader, Keep-Alive, User-Agent, X-Requested-With, If-Modified-Since, Cache-Control, Content-Type, Pragma")
|
|
c.Header("Access-Control-Expose-Headers", "Content-Length, Access-Control-Allow-Origin, Access-Control-Allow-Headers, Content-Type")
|
|
// c.Header("Access-Control-Max-Age", "172800")
|
|
c.Header("Access-Control-Allow-Credentials", "true")
|
|
c.Set("content-type", "application/json")
|
|
}
|
|
|
|
//放行所有OPTIONS方法
|
|
if method == "OPTIONS" {
|
|
c.JSON(http.StatusOK, "Options Request!")
|
|
}
|
|
|
|
c.Next()
|
|
}
|
|
}
|