standard/grpctest/client/grpcclient.go

package client

import (
	"context"
	"fmt"
	"io"
	pb "test/grpctest/proto"
)

//接受服务端返回的流
func getStream(client pb.ObjectStorageServiceClient){
	stream, err := client.GetStream(context.Background(),&pb.ReqData{
		Data: "shuai",
	})
	if err != nil {
		fmt.Println("error:",err.Error())
		return
	}
	for {
		repData,err := stream.Recv()
		if err != nil {
			if err == io.EOF {
				fmt.Println("get response over")
				break
			}
			fmt.Println("get stream failed",err.Error())
		}
		fmt.Println("server return:",repData.Data)
	}
}

//向服务端发送流
func setStream(client pb.ObjectStorageServiceClient){
	sendBuffer := []string {"hello","shuai","mei"}
	stream,err := client.SetStream(context.Background())
	if err != nil {
		fmt.Println("set stream failed")
	}
	if stream == nil {
		fmt.Println("stream is nil")
	}

}