go集成grpc

可以去 https://github.com/protocolbuffers/protobuf 下载

通过命令 go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.26 得到

通过命令 go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.1 得到

grpc参考官网 https://www.grpc.io/。把这些exe文件都加到环境变量

syntax = "proto3";

package service;

option go_package = "go-test/pb/hello;hello";

message Req{
  int32 prod_id=1;
}

message Res{
  int32 prod_stock=1;
}

service Hello{
  rpc Hello(Req) returns (Res);
}

//go:generate protoc --go_out=. --go_opt=paths=source_relative   --go-grpc_out=. --go-grpc_opt=paths=source_relative     ./pb/hello.proto

//生成或者更新 proto的go代码
//go:generate protoc --go_out=. --go_opt=paths=source_relative   --go-grpc_out=. --go-grpc_opt=paths=source_relative     ./pb/hello.proto

func main()  {
	startServer()
}



func startServer(){
	l, err := net.Listen("tcp", ":9999")
	if err != nil{
		panic(err)
	}
	s := grpc.NewServer()
	hello.RegisterHelloServer(s,&rpcServer{})
	s.Serve(l)
}

type  rpcServer struct{
	hello.UnimplementedHelloServer
}

func (r *rpcServer) Hello(ctx context.Context,req *hello.Req) (res *hello.Res,err error){
	res = &hello.Res{ProdStock: 99}
	return res,nil
}

//测试 提供的服务
func TestClient(t *testing.T) {
	log.Println("开始启动客户端")
	conn, err := grpc.Dial("localhost:9999",grpc.WithInsecure())
	if err !=nil{
		log.Fatalln(err)
	}
	defer conn.Close()
	client := hello.NewHelloClient(conn)
	res, err := client.Hello(context.Background(), &hello.Req{ProdId: 1})

	if err != nil{
		log.Fatalln(err)
	}

	fmt.Println(res)

}