Return if is canceld context

Author: anhthii

pkg/eventconsumer/keygen_consumer.go

@@ -60,6 +60,9 @@ func (sc *keygenConsumer) waitForAllPeersReadyToGenKey(ctx context.Context) erro
 	for {
 		select {
 		case <-ctx.Done():
+			if ctx.Err() == context.Canceled {
+				return nil
+			}
 			return ctx.Err()
 		case <-ticker.C:
 			allPeersReady := sc.peerRegistry.ArePeersReady()

pkg/eventconsumer/sign_consumer.go

@@ -70,6 +70,9 @@ func (sc *signingConsumer) waitForSufficientPeers(ctx context.Context) error {
 	for {
 		select {
 		case <-ctx.Done():
+			if ctx.Err() == context.Canceled {
+				return nil
+			}
 			return ctx.Err()
 		case <-ticker.C:
 			readyPeers := sc.peerRegistry.GetReadyPeersCount()