handle eof error

streamingfast · Aug 9, 2024 · 3509f83 · 3509f83
1 parent 9f0a708
commit 3509f83
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/injective/cmd/fireinjective/tools_fix_unknown_type_blocks.go b/injective/cmd/fireinjective/tools_fix_unknown_type_blocks.go
@@ -77,7 +77,7 @@ func fixUnknownTypeBlocksRunE(zlog *zap.Logger, tracer logging.Tracer) firecore.
 		}
 
 		var lastFilename string
-		var blockCount int
+		var blockCount = 0
 		err = srcStore.WalkFrom(ctx, "", fmt.Sprintf("%010d", firstStreamableBlock), func(filename string) error {
 			var fileReader io.Reader
 			fileReader, err = srcStore.OpenObject(ctx, filename)
@@ -135,6 +135,10 @@ func fixUnknownTypeBlocksRunE(zlog *zap.Logger, tracer logging.Tracer) firecore.
 				fmt.Println("done")
 				return nil
 			}
+			if errors.Is(err, io.EOF) {
+				fmt.Println("done")
+				return nil
+			}
 			return fmt.Errorf("walking source store: %w", err)
 		}