Make it done when Push to the notify channel if given

make sure it done,Push to the notify channel if given when shutdown sign arrvied.
hashicorp · Sep 5, 2023 · 6afcd70 · 6afcd70
1 parent 3b0f2e5
commit 6afcd70
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/raft.go b/raft.go
@@ -434,6 +434,11 @@ func (r *Raft) runLeader() {
 		select {
 		case notify <- true:
 		case <-r.shutdownCh:
+			// make sure push to the notify channel ( if given )
+			select {
+			case notify <- true:
+			default:
+			}
 		}
 	}