fix: not wait for MaxHoldTime if AlwaysHold is true

hsldymq · Jan 30, 2024 · 11fd20d · 11fd20d
1 parent b637607
commit 11fd20d
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1 +1,2 @@
-vendor/
+vendor/
+.idea/
diff --git a/keeper.go b/keeper.go
@@ -145,12 +145,13 @@ func (k *ShutdownKeeper) Wait() {
 	}
 	<-k.shutdownEventChan
 
-	if !k.alwaysHold && k.getHoldTokenNum() == 0 {
-		return
-	}
-	select {
-	case <-time.After(k.maxHoldTime):
-	case <-k.tokenReleaseChan:
+	if k.alwaysHold {
+		<-time.After(k.maxHoldTime)
+	} else if k.getHoldTokenNum() > 0 {
+		select {
+		case <-time.After(k.maxHoldTime):
+		case <-k.tokenReleaseChan:
+		}
 	}
 }