From: qydysky <qydysky@foxmail.com>
Date: Mon, 12 Apr 2021 08:25:42 +0000 (+0800)
Subject: fix block in limit
X-Git-Tag: v0.5.6
X-Git-Url: http://127.0.0.1:8081/?a=commitdiff_plain;h=4f8c72d0378c0b7b12b3da009caf8cc3f6e0343b;p=part%2F.git

fix block in limit
---

diff --git a/limit/Limit.go b/limit/Limit.go
index 410a2c8..c7d55af 100644
--- a/limit/Limit.go
+++ b/limit/Limit.go
@@ -23,8 +23,12 @@ func New(maxNum_in_period,ms_in_period,ms_to_timeout int) (*Limit) {
 
 	go func(object *Limit){
 		for object.maxNum_in_period > 0 {
-			for i:=1;i<=object.maxNum_in_period;i++{
-				object.bucket <- struct{}{}
+			object.bucket <- struct{}{}
+			for i:=1;i<object.maxNum_in_period;i++{
+				select {
+				case object.bucket <- struct{}{}:;
+				default :i = object.maxNum_in_period
+				}
 			}
 			time.Sleep(time.Duration(object.ms_in_period)*time.Millisecond)
 		}
diff --git a/limit/Limit_test.go b/limit/Limit_test.go
index 7441449..688a803 100644
--- a/limit/Limit_test.go
+++ b/limit/Limit_test.go
@@ -54,10 +54,10 @@ func Test_4(t *testing.T){
 }
 
 func Test_5(t *testing.T){
-	l := New(100,5000,0)
-	for i:=0;i<50;i+=1{
+	l := New(100,3000,0)
+	t.Log(l.TK())
+	for i:=1;i<=50;i+=1{
 		l.TO()
-		time.Sleep(time.Millisecond)
 	}
 	if l.TK() != 50 {t.Error(`5`,l.TK())}
 }
\ No newline at end of file