xing/wp-go
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

optimize code

Browse Source
This commit is contained in:
xing 2025-03-19 23:21:34 +08:00
parent 092f30b644
commit afe73c81e7
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
app/middleware/iplimit.go
View File

@ -158,23 +158,25 @@ func IpLimit(num int64, clearNum ...int64) (func(ctx *gin.Context), func(int64,
return CustomFlowLimit[string](a, num, clearNum...) return CustomFlowLimit[string](a, num, clearNum...)
} }
const minute = "2006-01-02 15:04"
func IpMinuteLimit(num int64, clearNum ...int64) (func(ctx *gin.Context), func(int64, ...int64)) { func IpMinuteLimit(num int64, clearNum ...int64) (func(ctx *gin.Context), func(int64, ...int64)) {
a := NewFlowLimits(func(c *gin.Context) string { a := NewFlowLimits(func(c *gin.Context) string {
return str.Join(c.ClientIP(), "|", time.Now().Format("2006-01-02 15:04")) return str.Join(c.ClientIP(), "|", time.Now().Format(minute))
}, ToManyRequest(), IpMinuteLimitDeferFn, }, ToManyRequest(), IpMinuteLimitDeferFn,
) )
return CustomFlowLimit(a, num, clearNum...) return CustomFlowLimit(a, num, clearNum...)
} }
func IpMinuteLimitDeferFn(_ *gin.Context, m LimitMap[string], k string, _ *int64, currentTotalFlow int64) { func IpMinuteLimitDeferFn(_ *gin.Context, m LimitMap[string], _ string, _ *int64, currentTotalFlow int64) {
cNum := atomic.LoadInt64(m.ClearNum) cNum := atomic.LoadInt64(m.ClearNum)
minu := strings.Split(k, "|")[1]
if currentTotalFlow <= cNum { if currentTotalFlow <= cNum {
now, _ := time.Parse(minute, time.Now().Format(minute))
m.Mux.Lock() m.Mux.Lock()
for key := range m.Map { for key := range m.Map {
t := strings.Split(key, "|")[1] minu, _ := time.Parse(minute, strings.Split(key, "|")[1])
if minu != t { if now.Sub(minu) > 0 {
delete(m.Map, key) delete(m.Map, key)
} }
} }