七叶笔记 » golang编程 » go grpc学习

go grpc学习

分类: golang编程 | 浏览: 252

1.windows下安装protoc

下载protoc-3.19.4-win64.zip

解压bin下面protoc.exe，复制到Gobin目录下。

2.安装 Go protobuf 插件 

 go get -u github.com/golang/protobuf/proto

go get -u github.com/golang/protobuf/protoc-gen-go

3.安装 grpc-go 

 go get google.golang.org/grpc

写个 Hello World 服务

  • 编写服务端 .proto 文件
  • 生成服务端 .pb.go 文件并同步给客户端
  • 编写服务端提供接口的代码
  • 编写客户端调用接口的代码 
 syntax = "proto3"; // 指定 proto 版本

package hello;     // 指定包名
option go_package = "/hello";

// 定义 Hello 服务
service Hello {
// 定义 SayHello 方法
rpc SayHello(HelloRequest) returns (HelloResponse) {}

// 定义 LotsOfReplies 方法
rpc LotsOfReplies(HelloRequest) returns (stream HelloResponse){}
}

// HelloRequest 请求结构
message HelloRequest {
string name = 1;//第一个参数
}

// HelloResponse 响应结构
message HelloResponse {
    string message = 1;//第一个参数
}

生成go客户端文件：

protoc -I . –go_out=plugins=grpc:. .\hello\hello.proto

提示：protoc-gen-go不存在，需要把GoPath的bin路径添加到环境变量bin下面

调用示例

编写服务端代码： 

 package service

import (
"fmt"
"golang.org/x/net/context"
"grpc/api/hello"
)

type HelloController struct{}

func (h *HelloController) SayHello(ctx context.Context, in *hello.HelloRequest) (*hello.HelloResponse, error) {
return &hello.HelloResponse{Message: fmt.Sprintf("%s", in.Name)}, nil
}

func (h *HelloController) LotsOfReplies(in *hello.HelloRequest, stream hello.Hello_LotsOfRepliesServer) error {
for i := 0; i < 10; i++ {
stream.Send(&hello.HelloResponse{Message: fmt.Sprintf("%s %s %d", in.Name, "Reply", i)})
}
return nil
}

服务端启动main: 

 package main

import (
   "google.golang.org/grpc"
   "grpc/api/hello"//调用hello.pb.go文件
   "grpc/internal/service"
   "log"
   "net"
)

const (
   Address = "0.0.0.0:9090"
)

func main() {
   listen, err := net.Listen("tcp", Address)
   if err != nil {
      log.Fatalf("Failed to listen: %v", err)
   }

   s := grpc.NewServer()

   // 服务注册
   hello.RegisterHelloServer(s, &service.HelloController{})

   log.Println("Listen on " + Address)

   if err := s.Serve(listen); err != nil {
      log.Fatalf("Failed to serve: %v", err)
   }
}

客户端调用： 

 package main

import (
"golang.org/x/net/context"
"google.golang.org/grpc"
"grpc/api/hello"//调用客户端hello.pb.go文件
"io"
"log"
)

const Address string = "0.0.0.0:9090"

func main() {
conn, err := grpc.Dial(Address, grpc.WithInsecure())
if err != nil {
log.Fatalln(err)
}
defer conn.Close()

// 初始化客户端
c := hello.NewHelloClient(conn)

// 调用 SayHello 方法
res, err := c.SayHello(context.Background(), &hello.HelloRequest{Name: "Hello World"})

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

log.Println(res.Message)

// 调用 LotsOfReplies 方法
stream, err := c.LotsOfReplies(context.Background(), &hello.HelloRequest{Name: "Hello World"})
if err != nil {
log.Fatalln(err)
}

for {
res, err := stream.Recv()
if err == io.EOF {
break
}

if err != nil {
log.Printf("stream.Recv: %v", err)
}

log.Printf("%s", res.Message)
}
}

相关文章