From 4f8c72d0378c0b7b12b3da009caf8cc3f6e0343b Mon Sep 17 00:00:00 2001
From: qydysky <qydysky@foxmail.com>
Date: Mon, 12 Apr 2021 16:25:42 +0800
Subject: [PATCH] fix block in limit

---
 limit/Limit.go      | 8 ++++++--
 limit/Limit_test.go | 6 +++---
 2 files changed, 9 insertions(+), 5 deletions(-)

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
-- 
2.39.2