Fix ip limit

DaoCloud · Jun 14, 2024 · fdc69bd · fdc69bd
1 parent 8b17fdb
commit fdc69bd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/crproxy.go b/crproxy.go
@@ -696,10 +696,12 @@ func (c *CRProxy) checkLimit(rw http.ResponseWriter, r *http.Request, info *Path
 				c.logger.Println("exceed limit", address, aver, *c.ipsSpeedLimit)
 			}
 			if c.limitDelay {
-				select {
-				case <-r.Context().Done():
-					return false
-				case <-time.After(bps.Next().Sub(time.Now())):
+				for bps.Aver() > *c.ipsSpeedLimit {
+					select {
+					case <-r.Context().Done():
+						return false
+					case <-time.After(bps.Next().Sub(time.Now())):
+					}
 				}
 			} else {
 				err := ErrorCodeTooManyRequests