Commit 92ab0171 authored by shajiaiming's avatar shajiaiming

fix

parent 62178740
...@@ -8,20 +8,16 @@ import ( ...@@ -8,20 +8,16 @@ import (
"strings" "strings"
) )
func AUTH(handlerTableName map[string]string) gin.HandlerFunc { func AUTH() gin.HandlerFunc {
return func(ctx *gin.Context) { return func(ctx *gin.Context) {
token := ctx.Request.Header.Get("Token") token := ctx.Request.Header.Get("Token")
user, _ := util.ParseToken(token) user, _ := util.ParseToken(token)
if 81 != user.UserInfo.Uid { if !strings.Contains(strings.ToLower(user.UserInfo.Username), "kefu") {
ctx.Next() return
} }
handlerName := strings.Split(ctx.HandlerName(), ".")[1] handlerName := strings.Split(ctx.HandlerName(), ".")[1]
if _, ok := handlerTableName[handlerName]; !ok {
ctx.Next()
}
handleAlowed := []string{"GetCoinChains", "GetUserChains"} handleAlowed := []string{"GetCoinChains", "GetUserChains"}
_, handle := util.Contains(handleAlowed, handlerName) _, handle := util.Contains(handleAlowed, handlerName)
......
package routers package routers
import ( import (
"bwallet/middleware/auth"
"bwallet/middleware/jwt" "bwallet/middleware/jwt"
"bwallet/middleware/log" "bwallet/middleware/log"
"bwallet/pkg/e" "bwallet/pkg/e"
...@@ -52,11 +53,10 @@ func InitRouter() *gin.Engine { ...@@ -52,11 +53,10 @@ func InitRouter() *gin.Engine {
api := r.Group("/api") api := r.Group("/api")
api.Use(jwt.JWT()) api.Use(jwt.JWT()).Use(auth.AUTH())
api.POST("/log", backend.AddOperationLog) api.POST("/log", backend.AddOperationLog)
api.GET("/logs", backend.GetOperationLogs) api.GET("/logs", backend.GetOperationLogs)
api.Use(log.LogMiddleware(e.HandleTableName)) api.Use(log.LogMiddleware(e.HandleTableName))
//api.Use(auth.AUTH(e.HandleTableName))
{ {
//api.POST("/upload",backend.Upload) //api.POST("/upload",backend.Upload)
api.GET("/coins", backend.GetCoins) api.GET("/coins", backend.GetCoins)
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment