grpc的一元调用（简单模式）

hello.proto文件

syntax = "proto3";

option go_package = "./;proto";

service Greeter {
  rpc SayHello(HelloReqData) returns (HelloResData);
}

message HelloReqData{
  string name = 1;
}

message HelloResData{
  string msg = 1;
}

生成go文件命令

protoc --go_out=. --go-grpc_out=. hello.proto

client.go

package main

import (
	"Go_Bible/metadata_test/proto"
	"context"
	"fmt"
	"google.golang.org/grpc"
	"google.golang.org/grpc/metadata"
)

func main() {
	// 拨号
	conn, err := grpc.Dial("127.0.0.1:15001", grpc.WithInsecure())
	if err != nil {
		panic("客户端拨号失败:" + err.Error())
	}
	// 关闭连接
	defer conn.Close()
	// 创建客户端
	client := proto.NewGreeterClient(conn)

	// 创建metadata元数据
	md := metadata.New(map[string]string{
		"token" : "1234",
	})
	// 创建上下文
	ctx := metadata.NewOutgoingContext(context.Background(), md)

	// 传入上下文调用接口
	data, err := client.SayHello(ctx, &proto.HelloReqData{Name: "kevin"})
	if err != nil {
		panic("调用SayHello接口失败:" + err.Error())
	}
	// 打印结果
	fmt.Println("调用接口成功:" + data.Msg)
}

server.go

package main

import (
	"Go_Bible/metadata_test/proto"
	"context"
	"fmt"
	"google.golang.org/grpc"
	"google.golang.org/grpc/metadata"
	"net"
	"strconv"
)

type Server struct {
	proto.UnimplementedGreeterServer
}

func (s *Server) SayHello(ctx context.Context, in *proto.HelloReqData) (*proto.HelloResData, error) {
	// 在调用的接口处，获取上下文，获取metadata元数据
	// md是一个map，key为string类型，值为[]string类型
	md, ok := metadata.FromIncomingContext(ctx)
	if !ok {
		fmt.Println("服务端SayHello接口获取metadata失败")
	}
	// 枚举元数据
	for k, v := range md {
		fmt.Println(k + ":")
		for i, val := range v {
			fmt.Println(strconv.Itoa(i) + `:` + val)
		}
	}
	// 返回一个响应
	return &proto.HelloResData{
		Msg: "Hello:" + in.Name,
	}, nil
}

func main() {
	// 创建服务
	server := grpc.NewServer()
	// 注册服务
	proto.RegisterGreeterServer(server, &Server{})
	// 监听端口
	lis, err := net.Listen("tcp", "0.0.0.0:15001")
	if err != nil {
		panic("服务端监听失败:" + err.Error())
	}
	// 开启服务
	if err = server.Serve(lis); err != nil {
		panic("开启服务失败:" + err.Error())
	}
}