当前位置: 代码迷 >> 综合 >> grpc-python 与 grpc-go 互传图片文件
  详细解决方案

grpc-python 与 grpc-go 互传图片文件

热度:34   发布时间:2023-10-17 22:16:45.0

将go语言作为客户端,python作为服务器端

go语言部分:

go_client.proto:

// Copyright 2015 gRPC authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.syntax = "proto3";option java_multiple_files = true;
option java_package = "io.grpc.examples.helloworld";
option java_outer_classname = "HelloWorldProto";package helloworld;// The greeting service definition.
service Greeter {// Sends a greetingrpc SayHello (HelloRequest) returns (HelloReply) {}
}// The request message containing the user's name.
message HelloRequest {string name = 1;
//下面添加定义photo的二进制字段名bytes photo = 2;
}// The response message containing the greetings
message HelloReply {string message = 1;
}

go_client/main.go:

/*** Copyright 2015 gRPC authors.** Licensed under the Apache License, Version 2.0 (the "License");* you may not use this file except in compliance with the License.* You may obtain a copy of the License at**     http://www.apache.org/licenses/LICENSE-2.0** Unless required by applicable law or agreed to in writing, software* distributed under the License is distributed on an "AS IS" BASIS,* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.* See the License for the specific language governing permissions and* limitations under the License.**/// Package main implements a client for Greeter service.
package mainimport ("context""io/ioutil""log""os""time""google.golang.org/grpc"pb "google.golang.org/grpc/examples/helloworld/helloworld"
)const (address     = "localhost:50051"defaultName = "world"
)func main() {// Set up a connection to the server.conn, err := grpc.Dial(address, grpc.WithInsecure())if err != nil {log.Fatalf("did not connect: %v", err)}defer conn.Close()c := pb.NewGreeterClient(conn)// Contact the server and print out its response.name := defaultNameif len(os.Args) > 1 {name = os.Args[1]}ctx, cancel := context.WithTimeout(context.Background(), time.Second)defer cancel()//读取一张图片并发送到python服务端photo, err := ioutil.ReadFile("C:\\Users\\zhilo\\Desktop\\桌面的东西\\prologue.png")r, err := c.SayHello(ctx, &pb.HelloRequest{Name: name, Photo:photo})if err != nil {log.Fatalf("could not greet: %v", err)}log.Printf("Greeting: %s", r.Message)
}

接着,在根目录执行:

$protoc -I helloworld/ helloworld/helloworld.proto --go_out=plugins=grpc:helloworld

生成helloworld.pb.go

 

python部分:

python_server.protoc:

// Copyright 2015 gRPC authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.syntax = "proto3";option java_multiple_files = true;
option java_package = "io.grpc.examples.helloworld";
option java_outer_classname = "HelloWorldProto";
option objc_class_prefix = "HLW";package helloworld;// The greeting service definition.
service Greeter {// Sends a greetingrpc SayHello (HelloRequest) returns (HelloReply) {}
}// The request message containing the user's name.
message HelloRequest {string name = 1;
//下面定义photo的二进制字段名bytes photo = 2;
}// The response message containing the greetings
message HelloReply {string message = 1;
}

 

python_server.py:

# Copyright 2015 gRPC authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""The Python implementation of the GRPC helloworld.Greeter server."""from concurrent import futures
import time
import loggingimport grpc
from PIL import Image
import ioimport helloworld_pb2
import helloworld_pb2_grpc_ONE_DAY_IN_SECONDS = 60 * 60 * 24class Greeter(helloworld_pb2_grpc.GreeterServicer):def SayHello(self, request, context):#接受到go传过来的图片之后的处理部分:data_stream = request.photodata = io.BytesIO(data_stream)img = Image.open(data)img.save("test.png")return helloworld_pb2.HelloReply(message='Hello, %s! from python' % request.name)def SayHelloAgain(self, request, context):return helloworld_pb2.HelloReply(message="Hello, %s! Again!!!" % request.name)def serve():server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server)server.add_insecure_port('[::]:50051')server.start()try:while True:time.sleep(_ONE_DAY_IN_SECONDS)except KeyboardInterrupt:server.stop(0)if __name__ == '__main__':logging.basicConfig()serve()

然后在根目录运行下面命令:

$python -m grpc_tools.protoc -I../../protos --python_out=. --grpc_python_out=. ../../protos/helloworld.proto

 

这样,先运行python_server.py,然后再运行go_client/main.go,就可以将go端的照片传送到python端并保存啦!