def create_media_server(port, threads=8): media_server = grpc.server( futures.ThreadPoolExecutor(threads), interceptors=[ TracingInterceptor(), get_media_auth_interceptor( config.config["MEDIA_SERVER_BEARER_TOKEN"]) ], ) media_server.add_insecure_port(f"[::]:{port}") media_pb2_grpc.add_MediaServicer_to_server(Media(), media_server) return media_server
def media_session(bearer_token): """ Create a fresh Media API for testing, uses the bearer token for media auth """ media_auth_interceptor = get_media_auth_interceptor(bearer_token) with futures.ThreadPoolExecutor(1) as executor: server = grpc.server(executor, interceptors=[media_auth_interceptor]) port = server.add_secure_port("localhost:0", grpc.local_server_credentials()) servicer = Media() media_pb2_grpc.add_MediaServicer_to_server(servicer, server) server.start() call_creds = grpc.access_token_call_credentials(bearer_token) comp_creds = grpc.composite_channel_credentials(grpc.local_channel_credentials(), call_creds) try: with grpc.secure_channel(f"localhost:{port}", comp_creds) as channel: yield media_pb2_grpc.MediaStub(channel) finally: server.stop(None).wait()
requests_pb2_grpc.add_RequestsServicer_to_server(Requests(), server) search_pb2_grpc.add_SearchServicer_to_server(Search(), server) threads_pb2_grpc.add_ThreadsServicer_to_server(Threads(), server) server.start() media_server = grpc.server( futures.ThreadPoolExecutor(2), interceptors=[ LoggingInterceptor(), get_media_auth_interceptor( config.config["MEDIA_SERVER_BEARER_TOKEN"]) ], ) media_server.add_insecure_port("[::]:1753") media_pb2_grpc.add_MediaServicer_to_server(Media(), media_server) media_server.start() logger.info( f"Serving on 1751 (secure), 1752 (auth), 1753 (media), and 1754 (jailed)" ) if config.config["ROLE"] in ["scheduler", "all"]: scheduler = start_jobs_scheduler() if config.config["ROLE"] in ["worker", "all"]: worker = start_jobs_worker() logger.info("App waiting for signal...") signal.pause()