client

LdDl · Jun 10, 2022 · 02f582e · 02f582e
1 parent ea0174b
commit 02f582e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/cmd/client/main.go b/cmd/client/main.go
@@ -124,14 +124,17 @@ func main() {
 		for {
 			serverResp, err := privateStreamData.Recv()
 			if err == io.EOF {
+				done <- true
 				return
 			}
 			if err != nil {
 				fmt.Println(err)
+				done <- true
 				break
 			}
 			if serverResp == nil {
 				fmt.Println("Nil data")
+				done <- true
 				return
 			}
 			fmt.Printf("Message from server: %v\n", serverResp.Message)