gRPC框架 & ProtoBuf
安装相关工具:
pip3 install grpcio
pip3 install grpcio-tools
protobuf3有自己专门的定义的格式，基于此可以生成不同的脚本
编写示例的protobuf3:
syntax = "proto3";
message HelloRequest {
string name = 1;
}
生成代码:
python3 -m grpc_tools.protoc --python_out=. --grpc_python_out=. -I. hello.proto
生成文件:
hello_pb2_grpc.py hello_pb2.py
调用示例:
import hello_pb2
request = hello_pb2.HelloRequest()
request.name = "David"
# 对象生成字符串
res = request.SerializeToString()
print(res)
# 通过字符串反向生成对象
request2 = hello_pb2.HelloRequest()
request2.ParseFromString(res)
print(request2.name)
定义proto文件
syntax = "proto3";
service Greeter {
rpc SayHello(HelloRequest) returns (HelloReply) {}
}
message HelloRequest {
string name = 1;
}
message HelloReply {
string message = 1;
}
生成文件:
python3 -m grpc_tools.protoc --python_out=. --grpc_python_out=. -I. hello.proto
示例代码
服务器端:
import hello_pb2
import hello_pb2_grpc
import grpc
from concurrent import futures
class Greeter(hello_pb2_grpc.GreeterServicer):
def SayHello(self, request, context):
return hello_pb2.HelloReply(message=f"Hello, {request.name}")
if __name__ == "__main__":
# 实例化server
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
# 注册逻辑到server
hello_pb2_grpc.add_GreeterServicer_to_server(Greeter(),server)
server.add_insecure_port('localhost:50051')
server.start()
server.wait_for_termination()
客户端:
import grpc
import hello_pb2,hello_pb2_grpc
if __name__ == "__main__":
with grpc.insecure_channel("localhost:50051") as channel:
stub = hello_pb2_grpc.GreeterStub(channel)
rsp: hello_pb2.HelloReply = stub.SayHello(hello_pb2.HelloRequest(name="David"))
print(rsp.message)
客户端直接如同调用本地函数一样调用远程函数SayHello，只要写好规则就能自动生成代码
在Go语言中也是同理，将以上proto文件放到一个目录下，使用protoc生成:
protoc --go_out=. --go-grpc_out=. ./*.proto
生成了hello.pb.go文件
服务端代码:
type Server struct{}
func (s *Server) SayHello(ctx context.Context, request *proto.HelloRequest) (*proto.HelloReply, error) {
return &proto.HelloReply{
Message: "hello " + request.Name,
}, nil
}
func main() {
s := grpc.NewServer()
proto.RegisterGreeterServer(s, &Server{})
lis, err := net.Listen("tcp", "0.0.0.0:8080")
if err != nil {
panic("failed to listen:" + err.Error())
}
err = s.Serve(lis)
if err != nil {
panic("failed to start grpc:" + err.Error())
}
}
客户端代码:
func main() {
conn, err := grpc.Dial("127.0.0.1:8080", grpc.WithInsecure())
if err != nil {
panic(err)
}
defer conn.Close()
c := proto.NewGreeterClient(conn)
r, err := c.SayHello(context.Background(), &proto.HelloRequest{Name: "David"})
if err != nil {
panic(err)
}
fmt.Println(r.Message)
}
同时Python和Go是可以互相调用的
也就是说，Python开启服务器时，Go的客户端通过服务器的地址和端口可以使用RPC调用，反之亦然
只要基于同一份proto文件即可实现互相调用
