hare3/compat: exit if hare channel got closed (#5247)

on close empty value will be emitted, this commit checks that and exit if channel was closed
spacemeshos · Nov 9, 2023 · 6c63aad · 6c63aad
1 parent c43676f
commit 6c63aad
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/hare3/compat/result.go b/hare3/compat/result.go
@@ -15,7 +15,10 @@ func ReportResult(ctx context.Context, logger *zap.Logger, from <-chan hare3.Con
 		case <-ctx.Done():
 			logger.Info("hare3 results reporter exited")
 			return
-		case out := <-from:
+		case out, open := <-from:
+			if !open {
+				return
+			}
 			select {
 			case to <- hare.LayerOutput{
 				Ctx:       ctx,

diff --git a/hare3/compat/weakcoin.go b/hare3/compat/weakcoin.go
@@ -19,7 +19,10 @@ func ReportWeakcoin(ctx context.Context, logger *zap.Logger, from <-chan hare3.W
 		case <-ctx.Done():
 			logger.Info("weak coin reporter exited")
 			return
-		case out := <-from:
+		case out, open := <-from:
+			if !open {
+				return
+			}
 			if err := to.Set(out.Layer, out.Coin); err != nil {
 				logger.Error("failed to update weakcoin",
 					zap.Uint32("lid", out.Layer.Uint32()),