Fix data race around leader handling (#20)

Fixes #19.
twitchscience · Feb 23, 2019 · 6d05cb3 · 6d05cb3
1 parent b6682f9
commit 6d05cb3
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/kinsumer.go b/kinsumer.go
@@ -357,11 +357,7 @@ func (k *Kinsumer) Run() error {
 			if err != nil {
 				k.errors <- fmt.Errorf("error deregistering client: %s", err)
 			}
-			if k.isLeader {
-				close(k.leaderLost)
-				k.leaderLost = nil
-				k.isLeader = false
-			}
+			k.unbecomeLeader()
 			// Do this outside the k.isLeader check in case k.isLeader was false because
 			// we lost leadership but haven't had time to shutdown the goroutine yet.
 			k.leaderWG.Wait()