o /wi @sddlmZddlmZddlmZddlmZddlm Z ddl m Z ddl m Z er=ddlZdd lmZdd lmZn e d Ze d Ze d ZeeZdZ d"d#ddZe dddddd$d d!ZdS)%) annotations)ThreadPoolExecutor) TYPE_CHECKING)logging)experimental_func) _LazyImport) BaseStorageN)servicer) api_pb2_grpcgrpczoptuna.storages._grpc.servicerz1optuna.storages._grpc.auto_generated.api_pb2_grpcz%Y-%m-%d %H:%M:%S.%fstoragerhoststrportint thread_poolThreadPoolExecutor | Nonereturn grpc.ServercCs>t|ptdd}tt||||d||S)N ) max_workers:)r serverrr $add_StorageServiceServicer_to_server grpc_servicerOptunaStorageProxyServiceadd_insecure_portr r rrrrY/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/optuna/storages/_grpc/server.py make_servers  r z4.2.0 localhosti2)r rrNonecCsBt||||}|td|d|td|dS)aRun a gRPC server for the given storage URL, host, and port. Example: Run this server with the following way: .. code:: from optuna.storages import run_grpc_proxy_server from optuna.storages import get_storage storage = get_storage("mysql+pymysql://:@/[?]") run_grpc_proxy_server(storage, host="localhost", port=13000) Please refer to the client class :class:`~optuna.storages.GrpcStorageProxy` for the client usage. Please use :func:`~optuna.storages.get_storage` instead of :class:`~optuna.storages.RDBStorage` since ``RDBStorage`` by itself does not use cache in process and it may cause significant slowdown. Args: storage: A storage object to proxy. host: Hostname to listen on. port: Port to listen on. thread_pool: Thread pool to use for the server. If :obj:`None`, a default thread pool with 10 workers will be used. .. warning:: Currently, gRPC storage proxy does not support the :class:`~optuna.storages.JournalStorage`. This issue is tracked in https://github.com/optuna/optuna/issues/6084. Please use :class:`~optuna.storages.RDBStorage` instead. zServer started at rz Listening...N)r start_loggerinfowait_for_terminationrrrrrun_grpc_proxy_server&s *  r')N) r rr rrrrrrr) r rr rrrrrrr") __future__rconcurrent.futuresrtypingroptunaroptuna._experimentalroptuna._importsroptuna.storagesrr optuna.storages._grpcr r$optuna.storages._grpc.auto_generatedr get_logger__name__r$DATETIME_FORMATr r'rrrrs.