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
31
32
33
34
35
36
37
38
39
40
41
|
import os
import subprocess
import sys
import time
from concurrent import futures
import grpc
from molotov.tests._grpc import helloworld_pb2, helloworld_pb2_grpc
HERE = os.path.dirname(os.path.abspath(__file__))
PATH = os.path.join(HERE, "service.py")
class Greeter(helloworld_pb2_grpc.GreeterServicer):
def SayHello(self, request, context):
response = helloworld_pb2.HelloReply()
response.message = f"Hello, {request.name}!"
return response
def run_server():
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
helloworld_pb2_grpc.add_GreeterServicer_to_server(Greeter(), server)
server.add_insecure_port("[::]:50051")
server.start()
print("Server started. Listening on 0.0.0.0:50051")
server.wait_for_termination()
class GRPCServer:
def __init__(self):
self.p = subprocess.Popen([sys.executable, PATH])
time.sleep(1)
def stop(self):
self.p.terminate()
if __name__ == "__main__":
run_server()
|
