fix: stop chain-sync client in bulk mode in test program (#598)

blinklabs-io · Apr 25, 2024 · 62885c9 · 62885c9
1 parent df1a18b
commit 62885c9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/cmd/gouroboros/chainsync.go b/cmd/gouroboros/chainsync.go
@@ -247,6 +247,11 @@ func testChainSync(f *globalFlags) {
 			fmt.Printf("ERROR: failed to get available block range: %s\n", err)
 			os.Exit(1)
 		}
+		// Stop the chain-sync client to prevent the connection getting closed due to chain-sync idle timeout
+		if err := oConn.ChainSync().Client.Stop(); err != nil {
+			fmt.Printf("ERROR: failed to shutdown chain-sync: %s\n", err)
+			os.Exit(1)
+		}
 		if err := oConn.BlockFetch().Client.GetBlockRange(start, end); err != nil {
 			fmt.Printf("ERROR: failed to request block range: %s\n", err)
 			os.Exit(1)