Есть ли способ установить тайм-аут на стороне сервера grpc? - PullRequest
Новая
       36

Есть ли способ установить тайм-аут на стороне сервера grpc?

0 голосов
/ 29 мая 2019

Невозможно тайм-аут соединения GRPC со стороны сервера. Возможно, что клиент устанавливает соединение, но остается в режиме удержания / сна, что приводит к зависанию соединения с сервером grpc. Есть ли способ на стороне сервера отключить соединение через определенное время или установить тайм-аут?

Мы попытались отключить соединение со стороны клиента, но не смогли сделать это со стороны сервера. По этой ссылке Проблема с настройкой gRPC. Получив прерывистую ошибку RPC недоступно , Ангад говорит, что это возможно, но невозможно определить эти параметры в python.

Мой фрагмент кода: 

def serve():
server = grpc.server(thread_pool=futures.ThreadPoolExecutor(max_workers=2), maximum_concurrent_rpcs=None, options=(('grpc.so_reuseport', 1),('grpc.GRPC_ARG_KEEPALIVE_TIME_MS', 1000)))
stt_pb2_grpc.add_ListenerServicer_to_server(Listener(), server)
server.add_insecure_port("localhost:50051")
print("Server starting in port "+str(50051))
server.start()
try:
    while True:
        time.sleep(60 * 60 * 24)
except KeyboardInterrupt:
    server.stop(0)
if __name__ == '__main__':
serve()

Я ожидаю, что время ожидания соединения со стороны сервера grpc также должно быть в python.

1 Ответ

0 голосов
/ 04 июня 2019

Короче говоря, вы можете найти context.abort(...) полезным, см. Справочник по API .Время ожидания обработчика сервера не поддерживается базовым C-Core API gRPC Python.Итак, вам нужно реализовать свой собственный механизм тайм-аута в Python.

Вы можете попробовать какое-то решение из других вопросов StackOverflow.

Или использовать простые, но большие накладные расходы, чтобы прерватьсоединение через определенный промежуток времени.Это может выглядеть так: 

_DEFAULT_TIME_LIMIT_S = 5

class FooServer(FooServicer):

  def RPCWithTimeLimit(self, request, context):
    rpc_ended = threading.Condition()
    work_finished = threading.Event()

    def wrapper(...):
      YOUR_ACTUAL_WORK(...)
      work_finished.set()
      rpc_ended.notify_all()

    def timer():
      time.sleep(_DEFAULT_TIME_LIMIT_S)
      rpc_ended.notify_all()

    work_thread = threading.Thread(target=wrapper, ...)
    work_thread.daemon = True
    work_thread.start()

    timer_thread = threading.Thread(target=timer)
    timer_thread.daemon = True
    timer_thread.start()

    rpc_ended.wait()
    if work_finished.is_set():
      return NORMAL_RESPONSE
    else:
      context.abort(grpc.StatusCode.DEADLINE_EXCEEDED, 'RPC Time Out!')
...