Explorar o código

fix rpc client examle (#81)

Zhang Hao %!s(int64=4) %!d(string=hai) anos
pai
achega
0f8140031a
Modificáronse 1 ficheiros con 7 adicións e 1 borrados
  1. 7 1
      example/rpc/client/stream/client.go

+ 7 - 1
example/rpc/client/stream/client.go

@@ -33,8 +33,12 @@ func main() {
 		log.Fatal(err)
 	}
 
+	done := make(chan struct{})
 	go func() {
-		for {
+		defer func() {
+			done <- struct{}{}
+		}()
+		for i := 0; i < 3; i++ {
 			resp, err := stm.Recv()
 			if err != nil {
 				log.Fatal(err)
@@ -52,4 +56,6 @@ func main() {
 			log.Fatal(err)
 		}
 	}
+
+	<-done
 }