in context/triple/go-client/cmd/client.go [44:95]
func main() {
err := config.Load()
if err != nil {
panic(err)
}
logger.Info("start to test triple unary context attachment transport")
req := &api.HelloRequest{
Name: "laurence",
}
ctx := context.Background()
// set user defined context attachment, map value can be string or []string, otherwise it is not to be transferred
userDefinedValueMap := make(map[string]interface{})
userDefinedValueMap["key1"] = "user defined value 1"
userDefinedValueMap["key2"] = "user defined value 2"
userDefinedValueMap["key3"] = []string{"user defined value 3.1", "user defined value 3.2"}
userDefinedValueMap["key4"] = []string{"user defined value 4.1", "user defined value 4.2"}
ctx = context.WithValue(ctx, constant.AttachmentKey, userDefinedValueMap)
reply, err := grpcGreeterImpl.SayHello(ctx, req)
if err != nil {
logger.Error(err)
}
logger.Infof("client response result: %v\n", reply)
//stream rpc
logger.Info("start to test triple streaming rpc context attachment transport")
request := &api.HelloRequest{
Name: "laurence",
}
stream, err := grpcGreeterImpl.SayHelloStream(ctx)
if err != nil {
logger.Error(err)
}
// stream grpc双向流式发送
err = stream.Send(request)
if err != nil {
logger.Error(err)
}
logger.Infof("client stream send request: %v\n", request)
var wg sync.WaitGroup
wg.Add(1)
go func() {
defer wg.Done()
reply, err := stream.Recv()
if err != nil {
logger.Error(err)
}
logger.Infof("client stream received result: %v\n", reply)
}()
wg.Wait()
}