47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
from concurrent import futures
|
||
import grpc
|
||
import time
|
||
|
||
from grpc_proto.check_grpc import check_grpc_pb2_grpc, check_grpc_pb2
|
||
|
||
|
||
class TaskServiceServicer(check_grpc_pb2_grpc.TaskServiceServicer):
|
||
def ProcessTask(self, request, context):
|
||
print(f"Received task_id: {request.task_id}")
|
||
|
||
return check_grpc_pb2.TaskResponse(
|
||
task_id=request.task_id,
|
||
success=True,
|
||
message="Task processed successfully"
|
||
)
|
||
|
||
|
||
class HealthCheckServicer(check_grpc_pb2_grpc.HealthCheckServicer):
|
||
def Check(self, request, context):
|
||
# 简单实现：总是返回SERVING状态
|
||
# 实际应用中可以根据服务状态返回不同值
|
||
return check_grpc_pb2.HealthCheckResponse(
|
||
status=check_grpc_pb2.HealthCheckResponse.ServingStatus.SERVING
|
||
)
|
||
|
||
|
||
def serve():
|
||
server = grpc.server(futures.ThreadPoolExecutor(max_workers=10))
|
||
|
||
# 添加服务实现
|
||
check_grpc_pb2_grpc.add_TaskServiceServicer_to_server(TaskServiceServicer(), server)
|
||
check_grpc_pb2_grpc.add_HealthCheckServicer_to_server(HealthCheckServicer(), server)
|
||
|
||
server.add_insecure_port('[::]:50051')
|
||
server.start()
|
||
print("Server started, listening on port 50051...")
|
||
|
||
try:
|
||
while True:
|
||
time.sleep(86400) # 保持运行
|
||
except KeyboardInterrupt:
|
||
server.stop(0)
|
||
|
||
|
||
if __name__ == '__main__':
|
||
serve() |