# 0.相关链接
grpc github：https://github.com/grpc/grpc-go (opens new window)
官方文档：https://grpc.io/docs/languages/go/ (opens new window)
案例代码：https://github.com/tenqaz/go-examples (opens new window)
# 1. 定义proto文件
proto文件是用来预先定义的消息格式。数据包会按照proto文件所定义的消息格式完成二进制码流的编码和解码。
代码语言：javascript复制syntax = "proto3";
// 指定生成的 go 文件存放位置及其包名
option go_package = "./;proto";
// 定义User rpc服务
service User {
// 定义rpc服务的方法
rpc AddUser (UserRequest) returns (UserResponse);
rpc GetUser (GetUserRequest) returns (GetUserResponse);
}
// 请求的结构体
message UserRequest {
string name = 1;
uint32 age = 2;
}
// 响应的结构体
message UserResponse {
string msg = 1;
int32 code = 2;
}
message GetUserRequest {
string name = 1;
}
message GetUserResponse {
string name = 1;
string age = 2;
}# 2. 编译proto文件
首先需要安装protoc的二进制文件，是用来编译proto的。
下载地址：https://github.com/protocolbuffers/protobuf/releases (opens new window)
然后再下载安装编译proto文件的插件
代码语言：javascript复制$ go install google.golang.org/protobuf/cmd/protoc-gen-go@v1.28
$ go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.2进入proto目录下使用以下命令对proto文件进行编译，会生成两个文件user.pb.go和user_grpc.pb.go两个文件
protoc --go_out=. --go-grpc_out=. user.proto- --go_out=. protobuf相关代码文件生成在该目录下
- --go-grpc_out=. grpc相关代码文件生成在该目录下
编译后:
- user.pb.go：主要是请求与相应数据包的结构体定义，客户端和服务端都可以通过该结构体进行序列化与反序列化。
- user_grpc.pb.go：主要是grpc的服务端和客户端代码，通过实现及调用接口来互相沟通。
# 3. 创建grpc服务端
我们首先要实现user_grpc.pb.go中的UserServer接口，该接口中的方法是我们在proto文件中定义的rpc服务AddUser和GetUser。
不过，你会发现其中还多了个mustEmbedUnimplementedUserServer方法，但是我们proto文件中并没有定义，这个是用来兼容使用的。你看可以文件中UnimplementedUserServer结构体实现了UserServer接口，自己定义UserService包含UnimplementedUserServer，即使没有实现UserServer所有接口，也不会报错。
package main
import (
"context"
"fmt"
"google.golang.org/grpc"
"log"
"net"
"simple_example/proto"
)
// UserService 定义结构体，实现UserServer
type UserService struct {
proto.UnimplementedUserServer
}
func NewUserService() *UserService {
return &UserService{}
}
// AddUser 实现rpc方法
func (us *UserService) AddUser(ctx context.Context, request *proto.UserRequest) (*proto.UserResponse, error) {
fmt.Printf("add user success. name = %s, age = %dn", request.GetName(), request.GetAge())
return &proto.UserResponse{Msg: "success", Code: 0}, nil
}
func (us *UserService) GetUser(ctx context.Context, request *proto.GetUserRequest) (*proto.GetUserResponse, error) {
fmt.Printf("get user. name = %sn", request.GetName())
return &proto.GetUserResponse{Name: request.GetName(), Age: 1999}, nil
}
func main() {
// 监听端口
lis, err := net.Listen("tcp", fmt.Sprintf("localhost:%d", 8000))
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
grpcServer := grpc.NewServer()
// 注册rpc服务
proto.RegisterUserServer(grpcServer, NewUserService())
grpcServer.Serve(lis)
}# 4. 创建grpc客户端
调用user_grpc.pb.go文件中的NewUserClient方法来创建调用rpc服务的客户端，调用rpc方法时使用user.pb.go中的结构体作为输入与输出。
package main
import (
"context"
"fmt"
"google.golang.org/grpc"
"google.golang.org/grpc/credentials/insecure"
"log"
"simple_example/proto"
)
func main() {
var opts []grpc.DialOption
opts = append(opts, grpc.WithTransportCredentials(insecure.NewCredentials()))
conn, err := grpc.Dial("localhost:8000", opts...)
if err != nil {
log.Fatalf("fail to dial: %v", err)
}
defer conn.Close()
client := proto.NewUserClient(conn)
// 调用rpc服务AddUser方法
resp, err := client.AddUser(context.Background(), &proto.UserRequest{Name: "zhengwenfeng", Age: 18})
if err != nil {
log.Fatalf("fail to AddUser: %v", err)
}
fmt.Printf("AddUser, msg = %s, code = %dn", resp.Msg, resp.Code)
// 调用rpc服务GetUser方法
getuserResp, err := client.GetUser(context.Background(), &proto.GetUserRequest{Name: "zhangsan"})
if err != nil {
log.Fatalf("fail to GetUser: %v", err)
}
fmt.Printf("GetUser, Name = %s, Age = %d", getuserResp.Name, getuserResp.Age)
}# 5. 运行结果
首先运行服务端监听rpc服务 然后再运行客户端调用服务。 服务端输出：
代码语言：javascript复制add user success. name = zhengwenfeng, age = 18
get user success. name = zhangsan客户端输出：
代码语言：javascript复制AddUser, msg = success, code = 0
GetUser, Name = zhangsan, Age = 1999
