github_mirrors/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2025-01-24 01:30:25 +08:00
Code Issues Packages Projects Releases Wiki Activity

update rpc example

Browse Source
This commit is contained in:
kevin 2020-09-18 18:15:39 +08:00
parent 0f8140031a
commit 974c19d6d3
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
example/rpc/client/stream/client.go
View File

@ -5,6 +5,7 @@ import (
"flag" "flag"
"fmt" "fmt"
"log" "log"
"sync"
"github.com/tal-tech/go-zero/core/discov" "github.com/tal-tech/go-zero/core/discov"
"github.com/tal-tech/go-zero/example/rpc/remote/stream" "github.com/tal-tech/go-zero/example/rpc/remote/stream"
@ -33,22 +34,21 @@ func main() {
log.Fatal(err) log.Fatal(err)
} }
done := make(chan struct{}) var wg sync.WaitGroup
go func() { go func() {
defer func() { for {
done <- struct{}{}
}()
for i := 0; i < 3; i++ {
resp, err := stm.Recv() resp, err := stm.Recv()
if err != nil { if err != nil {
log.Fatal(err) log.Fatal(err)
} }
fmt.Println("=>", resp.Greet) fmt.Println("=>", resp.Greet)
wg.Done()
} }
}() }()
for i := 0; i < 3; i++ { for i := 0; i < 3; i++ {
wg.Add(1)
fmt.Println("<=", name) fmt.Println("<=", name)
if err = stm.Send(&stream.StreamReq{ if err = stm.Send(&stream.StreamReq{
Name: name, Name: name,
@ -57,5 +57,5 @@ func main() {
} }
} }
<-done wg.Wait()
} }