gRPC 통신

syntax = "proto3";
package demo;

service Chat {
  // 단항: 요청 1개, 응답 1개
  rpc GetUser (UserRequest) returns (User);
  // 서버 스트리밍: 요청 1개, 응답 여러 개
  rpc WatchPrice (PriceRequest) returns (stream Price);
  // 클라이언트 스트리밍: 요청 여러 개, 응답 1개
  rpc Upload (stream Chunk) returns (UploadResult);
  // 양방향 스트리밍
  rpc Talk (stream Msg) returns (stream Msg);
}

message User {
  int32 id = 1;
  string name = 2;
}

# server.py
from concurrent import futures
import grpc
import helloworld_pb2, helloworld_pb2_grpc

class Greeter(helloworld_pb2_grpc.GreeterServicer):
    def SayHello(self, request, context):
        return helloworld_pb2.HelloReply(message="Hello, " + request.name)

server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server)
server.add_insecure_port("localhost:50051")
server.start()
server.wait_for_termination()

# client.py
with grpc.insecure_channel("localhost:50051") as channel:
    stub = helloworld_pb2_grpc.GreeterStub(channel)
    reply = stub.SayHello(helloworld_pb2.HelloRequest(name="kim"))
    print(reply.message)             # Hello, kim

import json

def varint(n):
    out = bytearray()
    while True:
        b = n & 0x7F
        n >>= 7
        if n:
            out.append(b | 0x80)       # 뒤에 바이트가 더 있음
        else:
            out.append(b)
            return bytes(out)

def encode_user(uid, name):
    data = name.encode()
    return (bytes([(1 << 3) | 0]) + varint(uid)      # 필드 1: varint
            + bytes([(2 << 3) | 2]) + varint(len(data)) + data)   # 필드 2: 길이 + 문자열

msg = encode_user(150, "kim")
print(msg.hex(" "), len(msg))      # 08 96 01 12 03 6b 69 6d 8
print(len(json.dumps({"id": 150, "name": "kim"}, separators=(",", ":"))))   # 23