go实现grpc通信

2022-04-25 09:28:08 浏览数 (1)

准备环境

代码语言:javascript复制
go get google.golang.org/grpc
go get google.golang.org/protobuf

代码实现

新增 chat/chat.proto 文件:

代码语言:javascript复制
syntax = "proto3";

package template;
option go_package = "../chat";
message Message{
  string body = 1;
}

service ChatService{
  rpc SayHello(Message) returns (Message){}
  rpc BroadcastMessage2(Message) returns (Message) {}
}

命令生成pb.go

代码语言:javascript复制
protoc --go_out=plugins=grpc:chat ./chat/chat.proto

编写chat 服务文件 chat.go:

代码语言:javascript复制
package chat

import (
   "context"
   "log"
)

type Server struct {
}

func (receiver *Server) SayHello(ctx context.Context, in *Message) (*Message, error) {
   log.Printf("Receive message body from client: %s", in.Body)
   return &Message{Body: "Hello From the Server!"}, nil
}

func (receiver *Server) Test(ctx context.Context, in *Message) (*Message, error) {
   log.Printf("test new message from a client: %s", in.Body)
   return &Message{Body: "test message!"}, nil
}

编写grpc-server服务端:

代码语言:javascript复制
package main

import (
   "fmt"
   "google.golang.org/grpc"
   "grpc-study/chat"
   "log"
   "net"
)

func main() {
   fmt.Println("Go gRPC Beginners Tutorial!")

   listen, err := net.Listen("tcp", ":9000")
   if err != nil {
      log.Fatalf("failed to listen: %v", err)
   }

   grpcServer := grpc.NewServer()
   s := chat.Server{}

   chat.RegisterChatServiceServer(grpcServer, &s)

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

grpc-client 客户端调用

代码语言:javascript复制
package main

import (
   "context"
   "google.golang.org/grpc"
   "grpc-study/chat"
   "log"
   "time"
)

func main() {
   var conn *grpc.ClientConn
   conn, err := grpc.Dial(":9000", grpc.WithInsecure())
   if err != nil {
      log.Fatalf("did not connect: %s n", err)
   }
   defer conn.Close()

   c := chat.NewChatServiceClient(conn)
   for true {
      response, err := c.SayHello(context.Background(), &chat.Message{Body: "66666"})
      if err != nil {
         log.Fatalf("Error when calling SayHello: %s", err)
      }
      log.Printf("Response from server: %s", response.Body)
      broadcastMessage, err := c.Test(context.Background(), &chat.Message{Body: "123"})
      if err != nil {
         log.Fatalf("Error when calling Broadcast Message: %s", err)
      }
      log.Printf("Response from server: %s", broadcastMessage.Body)

      time.Sleep(1*time.Second)
   }
}

启动

本文为仙士可原创文章,转载无需和我联系,但请注明来自仙士可博客www.php20.cn

0 人点赞