compare.py 881 Bytes
Newer Older
Nane Kratzke's avatar
Deploy  
Nane Kratzke committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
import requests, time, os
from statistics import median

import grpc
import helloworld_pb2
import helloworld_pb2_grpc


grpc_svc = os.environ.get("GRPC_SVC", "localhost")
grpc_port = os.environ.get("GRPC_PORT", "5555")
rest_svc = os.environ.get("REST_SVC", "localhost")
rest_port = os.environ.get("REST_PORT", "5000")

channel = grpc.insecure_channel(f"{ grpc_svc }:{ grpc_port }")
stub = helloworld_pb2_grpc.GreeterStub(channel)

grpc_s = []
rest_s = []

while True:
    for i in range(0, 100):
        start = time.time()
        response = stub.SayHello(helloworld_pb2.HelloRequest(name='you'))
        grpc_s.append(time.time() - start)

        start = time.time()
        x = requests.get(f"http://{ rest_svc }:{ rest_port }/hello/you")
        rest_s.append(time.time() - start)

    print(f"gRPC: { median(grpc_s) * 1000 :05.2f}ms REST: { median(rest_s) * 1000 :05.2f}ms")