zhangdeman/gin
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

切换util工具函数

Browse Source
This commit is contained in:
白茶清欢
2023-08-10 14:47:20 +08:00
parent 5e1aaaf64e
commit c93cf01fe0
4 changed files with 12 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
middleware/ip_validate.go
View File

@ -10,7 +10,7 @@ package middleware
import (
"git.zhangdeman.cn/zhangdeman/gin/define"
"git.zhangdeman.cn/zhangdeman/gin/response"
"git.zhangdeman.cn/zhangdeman/util"
networkUtil "git.zhangdeman.cn/zhangdeman/network/util"
"github.com/gin-gonic/gin"
)
@ -26,7 +26,7 @@ func ValidateBlackIPMiddleware(code interface{}, httpCode int, validateFunc defi
ctx.Next()
return
}
if validateFunc(ctx, util.IP.GetRemoteIP(ctx.Request)) {
if validateFunc(ctx, networkUtil.IP.GetRemoteIP(ctx.Request)) {
// 命中黑名单
response.Send(ctx, code, httpCode, nil)
ctx.Abort()
@ -48,7 +48,7 @@ func ValidateWhiteIPMiddleware(code interface{}, httpCode int, validateFunc defi
ctx.Next()
return
}
if !validateFunc(ctx, util.IP.GetRemoteIP(ctx.Request)) {
if !validateFunc(ctx, networkUtil.IP.GetRemoteIP(ctx.Request)) {
// 非名单
response.Send(ctx, code, httpCode, nil)
ctx.Abort()