From 08befa532610678f650047e003a5f2387997e975 Mon Sep 17 00:00:00 2001 From: kishori82 Date: Mon, 17 Nov 2025 07:36:41 -0500 Subject: [PATCH 1/2] fix wg.Done() --- grpc_p2p_client/p2p_client_multi_streams_subscribe.go | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/grpc_p2p_client/p2p_client_multi_streams_subscribe.go b/grpc_p2p_client/p2p_client_multi_streams_subscribe.go index 3b954de..1815f28 100644 --- a/grpc_p2p_client/p2p_client_multi_streams_subscribe.go +++ b/grpc_p2p_client/p2p_client_multi_streams_subscribe.go @@ -96,7 +96,6 @@ func main() { if *outputData != "" { dataDone = make(chan bool) go func() { - defer wg.Done() header := fmt.Sprintf("receiver\tsender\tsize\tsha256(msg)") go writeToFile(ctx, dataCh, dataDone, *outputData, header) }() @@ -105,7 +104,6 @@ func main() { if *outputTrace != "" { traceDone = make(chan bool) go func() { - defer wg.Done() header := "" //fmt.Sprintf("sender\tsize\tsha256(msg)") go writeToFile(ctx, traceCh, traceDone, *outputTrace, header) }() @@ -148,15 +146,16 @@ func receiveMessages(ctx context.Context, ip string, writeData bool, dataCh chan grpc.MaxCallSendMsgSize(math.MaxInt), ), ) + defer conn.Close() + + fmt.Printf("IP - %v\n", ip) if err != nil { log.Fatalf("failed to connect to node %v", err) } - defer conn.Close() client := protobuf.NewCommandStreamClient(conn) stream, err := client.ListenCommands(ctx) - if err != nil { log.Fatalf("ListenCommands: %v", err) } From 1a448b74cbd7facc279db6e916a4f40caf8d215e Mon Sep 17 00:00:00 2001 From: kishori82 Date: Mon, 17 Nov 2025 10:25:20 -0500 Subject: [PATCH 2/2] fixed the done --- grpc_p2p_client/p2p_client_multi_streams_subscribe.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/grpc_p2p_client/p2p_client_multi_streams_subscribe.go b/grpc_p2p_client/p2p_client_multi_streams_subscribe.go index 1815f28..79c0d72 100644 --- a/grpc_p2p_client/p2p_client_multi_streams_subscribe.go +++ b/grpc_p2p_client/p2p_client_multi_streams_subscribe.go @@ -146,12 +146,12 @@ func receiveMessages(ctx context.Context, ip string, writeData bool, dataCh chan grpc.MaxCallSendMsgSize(math.MaxInt), ), ) - defer conn.Close() - fmt.Printf("IP - %v\n", ip) + fmt.Printf("IP - %v\n", ip) if err != nil { log.Fatalf("failed to connect to node %v", err) } + defer conn.Close() client := protobuf.NewCommandStreamClient(conn)