1. 配置protobuf
新建hello.proto文件
syntax = "proto3"; // 指定proto版本
package hello; // 指定默认包名
// 指定golang包名
option go_package = "./;hello";
// 定义Hello服务
service Hello {
// 定义SayHello方法
rpc SayHello(HelloRequest) returns (HelloResponse) {}
}
// HelloRequest 请求结构
message HelloRequest {
string name = 1;
}
// HelloResponse 响应结构
message HelloResponse {
string message = 1;
}
hello.proto文件中定义了一个Hello Service,该服务包含一个SayHello方法,同时声明了HelloRequest和HelloResponse消息结构用于请求和响应。客户端使用HelloRequest参数调用SayHello方法请求服务端,服务端响应HelloResponse消息。一个最简单的服务就定义好了。
2. 编译golang用的protobuf
提前安装好protoc
在hello.proto当前目录运行,编译后得到文件:hello.pb.go、hello_grpc.pb.go
protoc --go_out=. --go_opt=paths=source_relative --go-grpc_out=. --go-grpc_opt=paths=source_relative hello.proto
3. 新建服务端 server.go、客户端 client.go
server.go
package main
import (
"fmt"
pb "go_mod/proto/hello"
"golang.org/x/net/context"
"google.golang.org/grpc"
"google.golang.org/grpc/reflection"
"net"
)
// 定义helloService并实现约定的接口
type helloService struct {
pb.UnimplementedHelloServer
}
// SayHello 实现Hello服务接口
func (h helloService) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloResponse, error) {
fmt.Println("request:", in.Name)
resp := new(pb.HelloResponse)
resp.Message = fmt.Sprintf("Hello %s.", in.Name)
return resp, nil
}
func main() {
// Address gRPC服务地址
var Address = "127.0.0.1:8001"
listen, err := net.Listen("tcp", Address)
if err != nil {
fmt.Println("Failed to listen: %v", err)
}
// 实例化grpc Server
s := grpc.NewServer()
// 注册HelloService
pb.RegisterHelloServer(s, &helloService{})
// 往grpc服务端注册反射服务
reflection.Register(s)
fmt.Println("Listen on " + Address)
// 启动grpc服务
if err := s.Serve(listen); err != nil {
fmt.Println(err)
}
}
client.go
package main
import (
"fmt"
pb "go_mod/proto/hello"
"golang.org/x/net/context"
"google.golang.org/grpc"
)
func main() {
// Address gRPC服务地址
var Address = "127.0.0.1:8001"
// 连接
conn, err := grpc.Dial(Address, grpc.WithInsecure())
if err != nil {
fmt.Println(err)
}
defer conn.Close()
// 初始化客户端
c := pb.NewHelloClient(conn)
// 调用方法
req := &pb.HelloRequest{Name: "gRPC"}
res, err := c.SayHello(context.Background(), req)
if err != nil {
fmt.Println(err)
}
//fmt.Println(res.Message)
fmt.Println(res)
}