60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
// Package middleware ...
|
|
//
|
|
// Description : middleware ...
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2022-06-25 23:24
|
|
package middleware
|
|
|
|
import (
|
|
"git.zhangdeman.cn/zhangdeman/gin/define"
|
|
"git.zhangdeman.cn/zhangdeman/gin/response"
|
|
networkUtil "git.zhangdeman.cn/zhangdeman/network/util"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// ValidateBlackIPMiddleware 验证黑名单IP的中间件
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 23:24 2022/6/25
|
|
func ValidateBlackIPMiddleware(code interface{}, httpCode int, validateFunc define.IsBlackIP) gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
if nil == validateFunc {
|
|
// 不验证黑名单
|
|
ctx.Next()
|
|
return
|
|
}
|
|
if validateFunc(ctx, networkUtil.IP.GetRemoteIP(ctx.Request)) {
|
|
// 命中黑名单
|
|
response.Send(ctx, code, httpCode, nil)
|
|
ctx.Abort()
|
|
return
|
|
}
|
|
ctx.Next()
|
|
}
|
|
}
|
|
|
|
// ValidateWhiteIPMiddleware 是否白名单IP
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 23:33 2022/6/25
|
|
func ValidateWhiteIPMiddleware(code interface{}, httpCode int, validateFunc define.IsWhiteIP) gin.HandlerFunc {
|
|
return func(ctx *gin.Context) {
|
|
if nil == validateFunc {
|
|
// 不验证白名单
|
|
ctx.Next()
|
|
return
|
|
}
|
|
if !validateFunc(ctx, networkUtil.IP.GetRemoteIP(ctx.Request)) {
|
|
// 非名单
|
|
response.Send(ctx, code, httpCode, nil)
|
|
ctx.Abort()
|
|
return
|
|
}
|
|
ctx.Next()
|
|
}
|
|
}
|