Go中实现gRPC

38次阅读

共计 1354 个字符,预计需要花费 4 分钟才能阅读完成。

简介

gRPCgoogle 开发的远程过程调用系统,基于 http2.0 协议标准开发,编码协议使用 protobuf

编写 proto

实现一个计算两个数之和的服务。

syntax = "proto3";

package proto;

option go_package = "/cal;cal";

message RequestInfo {
  int64 number1 = 1;
  int64 number2 = 2;
}

message ResponseInfo {int64 res = 1;}

service Cal {rpc Add (RequestInfo) returns (ResponseInfo) {}}

生成 go 文件

使用 protoc 生成 go 文件。

protoc --go_out=. --go-grpc_out=. ./proto/cal.proto

生成 cal.pb.gocal_grpc.pb.go 两个文件。

cal.pb.go 文件中主要包含 proto 中的 message 结构体。

cal_grpc.pb.go 文件中包含注册服务端的方法 RegisterCalServer、实例化客户端的方法 NewCalClient、未实现服务的结构体 UnimplementedCalServer

编写服务端

package main

import (
    "context"
    "google.golang.org/grpc"
    "net"
    "test/cal"
)

type Cal struct {cal.UnimplementedCalServer}

// 实现 Add 方法
func (c *Cal) Add(ctx context.Context, req *cal.RequestInfo) (res *cal.ResponseInfo, err error) {return &cal.ResponseInfo{Res: req.Number1 + req.Number2}, nil
}

func main() {
    // 监听
    listen, _ := net.Listen("tcp", ":8080")

    // 实例化 grpc 服务
    s := grpc.NewServer()

    // 注册服务
    cal.RegisterCalServer(s, &Cal{})

    // 启动
    s.Serve(listen)
}
  1. 结构体需包含 cal.UnimplementedCalServer
  2. 需实现 Add 方法,如未实现会调用 cal.UnimplementedCalServer 中的 Add 方法。

编写客户端

package main

import (
    "context"
    "fmt"
    "google.golang.org/grpc"
    "google.golang.org/grpc/credentials/insecure"
    "test/cal"
)

func main() {
    // 建立连接
    conn, _ := grpc.Dial("127.0.0.1:8080", grpc.WithTransportCredentials(insecure.NewCredentials()))

    // 实例化客户端
    client := cal.NewCalClient(conn)

    // 调用服务
    res, _ := client.Add(context.Background(), &cal.RequestInfo{
        Number1: 1,
        Number2: 2,
    })

    fmt.Println(res)
}
正文完
 
dkp
版权声明:本站原创文章,由 dkp 2024-01-03发表,共计1354字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请注明出处。