Wait for follower and leader loops to finish

cluster/cluster.go

@@ -648,10 +648,6 @@ func (c *cluster) Shutdown() error {
 		c.raft.Shutdown()
 	}
 
-	// TODO: here might some situations, where the manager is still need from the synchronize loop and will run into a panic
-	c.manager = nil
-	c.raft = nil
-
 	return nil
 }
 

cluster/leader.go

@@ -180,6 +180,22 @@ func (c *cluster) monitorLeadership() {
 				c.leaderLock.Unlock()
 			}
 		case <-c.shutdownCh:
+			if weAreFollowerCh != nil {
+				close(weAreFollowerCh)
+			}
+
+			if weAreLeaderCh != nil {
+				close(weAreLeaderCh)
+			}
+
+			if weAreEmergencyLeaderCh != nil {
+				close(weAreEmergencyLeaderCh)
+			}
+
+			leaderLoop.Wait()
+			emergencyLeaderLoop.Wait()
+			followerLoop.Wait()
+
 			return
 		}
 	}