indexify 0.3.5__py3-none-any.whl → 0.3.7__py3-none-any.whl

indexify/function_executor/proto/function_executor_pb2.pyi

@@ -1,247 +0,0 @@
-from typing import ClassVar as _ClassVar
-from typing import Iterable as _Iterable
-from typing import Mapping as _Mapping
-from typing import Optional as _Optional
-from typing import Union as _Union
-
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import message as _message
-from google.protobuf.internal import containers as _containers
-
-DESCRIPTOR: _descriptor.FileDescriptor
-
-class SerializedObject(_message.Message):
-    __slots__ = ("bytes", "string", "content_type")
-    BYTES_FIELD_NUMBER: _ClassVar[int]
-    STRING_FIELD_NUMBER: _ClassVar[int]
-    CONTENT_TYPE_FIELD_NUMBER: _ClassVar[int]
-    bytes: bytes
-    string: str
-    content_type: str
-    def __init__(
-        self,
-        bytes: _Optional[bytes] = ...,
-        string: _Optional[str] = ...,
-        content_type: _Optional[str] = ...,
-    ) -> None: ...
-
-class InitializeRequest(_message.Message):
-    __slots__ = ("namespace", "graph_name", "graph_version", "function_name", "graph")
-    NAMESPACE_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_NAME_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_VERSION_FIELD_NUMBER: _ClassVar[int]
-    FUNCTION_NAME_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_FIELD_NUMBER: _ClassVar[int]
-    namespace: str
-    graph_name: str
-    graph_version: str
-    function_name: str
-    graph: SerializedObject
-    def __init__(
-        self,
-        namespace: _Optional[str] = ...,
-        graph_name: _Optional[str] = ...,
-        graph_version: _Optional[str] = ...,
-        function_name: _Optional[str] = ...,
-        graph: _Optional[_Union[SerializedObject, _Mapping]] = ...,
-    ) -> None: ...
-
-class InitializeResponse(_message.Message):
-    __slots__ = ("success", "customer_error")
-    SUCCESS_FIELD_NUMBER: _ClassVar[int]
-    CUSTOMER_ERROR_FIELD_NUMBER: _ClassVar[int]
-    success: bool
-    customer_error: str
-    def __init__(
-        self, success: bool = ..., customer_error: _Optional[str] = ...
-    ) -> None: ...
-
-class SetInvocationStateRequest(_message.Message):
-    __slots__ = ("key", "value")
-    KEY_FIELD_NUMBER: _ClassVar[int]
-    VALUE_FIELD_NUMBER: _ClassVar[int]
-    key: str
-    value: SerializedObject
-    def __init__(
-        self,
-        key: _Optional[str] = ...,
-        value: _Optional[_Union[SerializedObject, _Mapping]] = ...,
-    ) -> None: ...
-
-class SetInvocationStateResponse(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class GetInvocationStateRequest(_message.Message):
-    __slots__ = ("key",)
-    KEY_FIELD_NUMBER: _ClassVar[int]
-    key: str
-    def __init__(self, key: _Optional[str] = ...) -> None: ...
-
-class GetInvocationStateResponse(_message.Message):
-    __slots__ = ("key", "value")
-    KEY_FIELD_NUMBER: _ClassVar[int]
-    VALUE_FIELD_NUMBER: _ClassVar[int]
-    key: str
-    value: SerializedObject
-    def __init__(
-        self,
-        key: _Optional[str] = ...,
-        value: _Optional[_Union[SerializedObject, _Mapping]] = ...,
-    ) -> None: ...
-
-class InvocationStateRequest(_message.Message):
-    __slots__ = ("request_id", "task_id", "set", "get")
-    REQUEST_ID_FIELD_NUMBER: _ClassVar[int]
-    TASK_ID_FIELD_NUMBER: _ClassVar[int]
-    SET_FIELD_NUMBER: _ClassVar[int]
-    GET_FIELD_NUMBER: _ClassVar[int]
-    request_id: str
-    task_id: str
-    set: SetInvocationStateRequest
-    get: GetInvocationStateRequest
-    def __init__(
-        self,
-        request_id: _Optional[str] = ...,
-        task_id: _Optional[str] = ...,
-        set: _Optional[_Union[SetInvocationStateRequest, _Mapping]] = ...,
-        get: _Optional[_Union[GetInvocationStateRequest, _Mapping]] = ...,
-    ) -> None: ...
-
-class InvocationStateResponse(_message.Message):
-    __slots__ = ("request_id", "success", "set", "get")
-    REQUEST_ID_FIELD_NUMBER: _ClassVar[int]
-    SUCCESS_FIELD_NUMBER: _ClassVar[int]
-    SET_FIELD_NUMBER: _ClassVar[int]
-    GET_FIELD_NUMBER: _ClassVar[int]
-    request_id: str
-    success: bool
-    set: SetInvocationStateResponse
-    get: GetInvocationStateResponse
-    def __init__(
-        self,
-        request_id: _Optional[str] = ...,
-        success: bool = ...,
-        set: _Optional[_Union[SetInvocationStateResponse, _Mapping]] = ...,
-        get: _Optional[_Union[GetInvocationStateResponse, _Mapping]] = ...,
-    ) -> None: ...
-
-class FunctionOutput(_message.Message):
-    __slots__ = ("outputs",)
-    OUTPUTS_FIELD_NUMBER: _ClassVar[int]
-    outputs: _containers.RepeatedCompositeFieldContainer[SerializedObject]
-    def __init__(
-        self, outputs: _Optional[_Iterable[_Union[SerializedObject, _Mapping]]] = ...
-    ) -> None: ...
-
-class RouterOutput(_message.Message):
-    __slots__ = ("edges",)
-    EDGES_FIELD_NUMBER: _ClassVar[int]
-    edges: _containers.RepeatedScalarFieldContainer[str]
-    def __init__(self, edges: _Optional[_Iterable[str]] = ...) -> None: ...
-
-class RunTaskRequest(_message.Message):
-    __slots__ = (
-        "namespace",
-        "graph_name",
-        "graph_version",
-        "function_name",
-        "graph_invocation_id",
-        "task_id",
-        "function_input",
-        "function_init_value",
-    )
-    NAMESPACE_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_NAME_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_VERSION_FIELD_NUMBER: _ClassVar[int]
-    FUNCTION_NAME_FIELD_NUMBER: _ClassVar[int]
-    GRAPH_INVOCATION_ID_FIELD_NUMBER: _ClassVar[int]
-    TASK_ID_FIELD_NUMBER: _ClassVar[int]
-    FUNCTION_INPUT_FIELD_NUMBER: _ClassVar[int]
-    FUNCTION_INIT_VALUE_FIELD_NUMBER: _ClassVar[int]
-    namespace: str
-    graph_name: str
-    graph_version: str
-    function_name: str
-    graph_invocation_id: str
-    task_id: str
-    function_input: SerializedObject
-    function_init_value: SerializedObject
-    def __init__(
-        self,
-        namespace: _Optional[str] = ...,
-        graph_name: _Optional[str] = ...,
-        graph_version: _Optional[str] = ...,
-        function_name: _Optional[str] = ...,
-        graph_invocation_id: _Optional[str] = ...,
-        task_id: _Optional[str] = ...,
-        function_input: _Optional[_Union[SerializedObject, _Mapping]] = ...,
-        function_init_value: _Optional[_Union[SerializedObject, _Mapping]] = ...,
-    ) -> None: ...
-
-class RunTaskResponse(_message.Message):
-    __slots__ = (
-        "task_id",
-        "function_output",
-        "router_output",
-        "stdout",
-        "stderr",
-        "is_reducer",
-        "success",
-    )
-    TASK_ID_FIELD_NUMBER: _ClassVar[int]
-    FUNCTION_OUTPUT_FIELD_NUMBER: _ClassVar[int]
-    ROUTER_OUTPUT_FIELD_NUMBER: _ClassVar[int]
-    STDOUT_FIELD_NUMBER: _ClassVar[int]
-    STDERR_FIELD_NUMBER: _ClassVar[int]
-    IS_REDUCER_FIELD_NUMBER: _ClassVar[int]
-    SUCCESS_FIELD_NUMBER: _ClassVar[int]
-    task_id: str
-    function_output: FunctionOutput
-    router_output: RouterOutput
-    stdout: str
-    stderr: str
-    is_reducer: bool
-    success: bool
-    def __init__(
-        self,
-        task_id: _Optional[str] = ...,
-        function_output: _Optional[_Union[FunctionOutput, _Mapping]] = ...,
-        router_output: _Optional[_Union[RouterOutput, _Mapping]] = ...,
-        stdout: _Optional[str] = ...,
-        stderr: _Optional[str] = ...,
-        is_reducer: bool = ...,
-        success: bool = ...,
-    ) -> None: ...
-
-class HealthCheckRequest(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class HealthCheckResponse(_message.Message):
-    __slots__ = ("healthy",)
-    HEALTHY_FIELD_NUMBER: _ClassVar[int]
-    healthy: bool
-    def __init__(self, healthy: bool = ...) -> None: ...
-
-class InfoRequest(_message.Message):
-    __slots__ = ()
-    def __init__(self) -> None: ...
-
-class InfoResponse(_message.Message):
-    __slots__ = ("version", "sdk_version", "sdk_language", "sdk_language_version")
-    VERSION_FIELD_NUMBER: _ClassVar[int]
-    SDK_VERSION_FIELD_NUMBER: _ClassVar[int]
-    SDK_LANGUAGE_FIELD_NUMBER: _ClassVar[int]
-    SDK_LANGUAGE_VERSION_FIELD_NUMBER: _ClassVar[int]
-    version: str
-    sdk_version: str
-    sdk_language: str
-    sdk_language_version: str
-    def __init__(
-        self,
-        version: _Optional[str] = ...,
-        sdk_version: _Optional[str] = ...,
-        sdk_language: _Optional[str] = ...,
-        sdk_language_version: _Optional[str] = ...,
-    ) -> None: ...

indexify/function_executor/proto/function_executor_pb2_grpc.py

@@ -1,307 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import warnings
-
-import grpc
-
-from indexify.function_executor.proto import (
-    function_executor_pb2 as indexify_dot_function__executor_dot_proto_dot_function__executor__pb2,
-)
-
-GRPC_GENERATED_VERSION = "1.68.1"
-GRPC_VERSION = grpc.__version__
-_version_not_supported = False
-
-try:
-    from grpc._utilities import first_version_is_lower
-
-    _version_not_supported = first_version_is_lower(
-        GRPC_VERSION, GRPC_GENERATED_VERSION
-    )
-except ImportError:
-    _version_not_supported = True
-
-if _version_not_supported:
-    raise RuntimeError(
-        f"The grpc package installed is at version {GRPC_VERSION},"
-        + f" but the generated code in indexify/function_executor/proto/function_executor_pb2_grpc.py depends on"
-        + f" grpcio>={GRPC_GENERATED_VERSION}."
-        + f" Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}"
-        + f" or downgrade your generated code using grpcio-tools<={GRPC_VERSION}."
-    )
-
-
-class FunctionExecutorStub(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def __init__(self, channel):
-        """Constructor.
-
-        Args:
-            channel: A grpc.Channel.
-        """
-        self.initialize = channel.unary_unary(
-            "/function_executor_service.FunctionExecutor/initialize",
-            request_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InitializeRequest.SerializeToString,
-            response_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InitializeResponse.FromString,
-            _registered_method=True,
-        )
-        self.initialize_invocation_state_server = channel.stream_stream(
-            "/function_executor_service.FunctionExecutor/initialize_invocation_state_server",
-            request_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InvocationStateResponse.SerializeToString,
-            response_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InvocationStateRequest.FromString,
-            _registered_method=True,
-        )
-        self.run_task = channel.unary_unary(
-            "/function_executor_service.FunctionExecutor/run_task",
-            request_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.RunTaskRequest.SerializeToString,
-            response_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.RunTaskResponse.FromString,
-            _registered_method=True,
-        )
-        self.check_health = channel.unary_unary(
-            "/function_executor_service.FunctionExecutor/check_health",
-            request_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.HealthCheckRequest.SerializeToString,
-            response_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.HealthCheckResponse.FromString,
-            _registered_method=True,
-        )
-        self.get_info = channel.unary_unary(
-            "/function_executor_service.FunctionExecutor/get_info",
-            request_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InfoRequest.SerializeToString,
-            response_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InfoResponse.FromString,
-            _registered_method=True,
-        )
-
-
-class FunctionExecutorServicer(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def initialize(self, request, context):
-        """Initializes the Function Executor to run tasks
-        for a particular function. This method is called only
-        once per Function Executor as it can only run a single function.
-        It should be called before calling RunTask for the function.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
-
-    def initialize_invocation_state_server(self, request_iterator, context):
-        """Initializes a server that sends requests to the client to perform actions on
-        a task's graph invocation state. This method is called only once per Function Executor
-        It should be called before calling RunTask for the function.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
-
-    def run_task(self, request, context):
-        """Executes the task defined in the request.
-        Multiple tasks can be running in parallel.
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
-
-    def check_health(self, request, context):
-        """Health check method to check if the Function Executor is healthy."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
-
-    def get_info(self, request, context):
-        """Information about this Function Executor."""
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details("Method not implemented!")
-        raise NotImplementedError("Method not implemented!")
-
-
-def add_FunctionExecutorServicer_to_server(servicer, server):
-    rpc_method_handlers = {
-        "initialize": grpc.unary_unary_rpc_method_handler(
-            servicer.initialize,
-            request_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InitializeRequest.FromString,
-            response_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InitializeResponse.SerializeToString,
-        ),
-        "initialize_invocation_state_server": grpc.stream_stream_rpc_method_handler(
-            servicer.initialize_invocation_state_server,
-            request_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InvocationStateResponse.FromString,
-            response_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InvocationStateRequest.SerializeToString,
-        ),
-        "run_task": grpc.unary_unary_rpc_method_handler(
-            servicer.run_task,
-            request_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.RunTaskRequest.FromString,
-            response_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.RunTaskResponse.SerializeToString,
-        ),
-        "check_health": grpc.unary_unary_rpc_method_handler(
-            servicer.check_health,
-            request_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.HealthCheckRequest.FromString,
-            response_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.HealthCheckResponse.SerializeToString,
-        ),
-        "get_info": grpc.unary_unary_rpc_method_handler(
-            servicer.get_info,
-            request_deserializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InfoRequest.FromString,
-            response_serializer=indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InfoResponse.SerializeToString,
-        ),
-    }
-    generic_handler = grpc.method_handlers_generic_handler(
-        "function_executor_service.FunctionExecutor", rpc_method_handlers
-    )
-    server.add_generic_rpc_handlers((generic_handler,))
-    server.add_registered_method_handlers(
-        "function_executor_service.FunctionExecutor", rpc_method_handlers
-    )
-
-
-# This class is part of an EXPERIMENTAL API.
-class FunctionExecutor(object):
-    """Missing associated documentation comment in .proto file."""
-
-    @staticmethod
-    def initialize(
-        request,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
-        return grpc.experimental.unary_unary(
-            request,
-            target,
-            "/function_executor_service.FunctionExecutor/initialize",
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InitializeRequest.SerializeToString,
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InitializeResponse.FromString,
-            options,
-            channel_credentials,
-            insecure,
-            call_credentials,
-            compression,
-            wait_for_ready,
-            timeout,
-            metadata,
-            _registered_method=True,
-        )
-
-    @staticmethod
-    def initialize_invocation_state_server(
-        request_iterator,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
-        return grpc.experimental.stream_stream(
-            request_iterator,
-            target,
-            "/function_executor_service.FunctionExecutor/initialize_invocation_state_server",
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InvocationStateResponse.SerializeToString,
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InvocationStateRequest.FromString,
-            options,
-            channel_credentials,
-            insecure,
-            call_credentials,
-            compression,
-            wait_for_ready,
-            timeout,
-            metadata,
-            _registered_method=True,
-        )
-
-    @staticmethod
-    def run_task(
-        request,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
-        return grpc.experimental.unary_unary(
-            request,
-            target,
-            "/function_executor_service.FunctionExecutor/run_task",
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.RunTaskRequest.SerializeToString,
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.RunTaskResponse.FromString,
-            options,
-            channel_credentials,
-            insecure,
-            call_credentials,
-            compression,
-            wait_for_ready,
-            timeout,
-            metadata,
-            _registered_method=True,
-        )
-
-    @staticmethod
-    def check_health(
-        request,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
-        return grpc.experimental.unary_unary(
-            request,
-            target,
-            "/function_executor_service.FunctionExecutor/check_health",
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.HealthCheckRequest.SerializeToString,
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.HealthCheckResponse.FromString,
-            options,
-            channel_credentials,
-            insecure,
-            call_credentials,
-            compression,
-            wait_for_ready,
-            timeout,
-            metadata,
-            _registered_method=True,
-        )
-
-    @staticmethod
-    def get_info(
-        request,
-        target,
-        options=(),
-        channel_credentials=None,
-        call_credentials=None,
-        insecure=False,
-        compression=None,
-        wait_for_ready=None,
-        timeout=None,
-        metadata=None,
-    ):
-        return grpc.experimental.unary_unary(
-            request,
-            target,
-            "/function_executor_service.FunctionExecutor/get_info",
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InfoRequest.SerializeToString,
-            indexify_dot_function__executor_dot_proto_dot_function__executor__pb2.InfoResponse.FromString,
-            options,
-            channel_credentials,
-            insecure,
-            call_credentials,
-            compression,
-            wait_for_ready,
-            timeout,
-            metadata,
-            _registered_method=True,
-        )

indexify/function_executor/proto/message_validator.py

@@ -1,38 +0,0 @@
-from typing import Any
-
-from .function_executor_pb2 import SerializedObject
-
-
-class MessageValidator:
-    def __init__(self, message: Any):
-        self._message = message
-
-    def required_field(self, field_name: str) -> "MessageValidator":
-        if not self._message.HasField(field_name):
-            raise ValueError(
-                f"Field '{field_name}' is required in {type(self._message).__name__}"
-            )
-        return self
-
-    def required_serialized_object(self, field_name: str) -> "MessageValidator":
-        """Validates the SerializedObject.
-
-        Raises: ValueError: If the SerializedObject is invalid or not present."""
-        self.required_field(field_name)
-        return self.optional_serialized_object(field_name)
-
-    def optional_serialized_object(self, field_name: str) -> "MessageValidator":
-        """Validates the SerializedObject.
-
-        Raises: ValueError: If the SerializedObject is invalid."""
-        if not self._message.HasField(field_name):
-            return self
-
-        serializedObject: SerializedObject = getattr(self._message, field_name)
-        if not serializedObject.HasField("string") and not serializedObject.HasField(
-            "bytes"
-        ):
-            raise ValueError("oneof 'data' is required in SerializedObject")
-        if not serializedObject.HasField("content_type"):
-            raise ValueError("Field 'content_type' is required in SerializedObject")
-        return self

indexify/function_executor/proto/server_configuration.py

@@ -1,19 +0,0 @@
-# We send function inputs and outputs over gRPC.
-# -1 means unlimited. We don't want to limit the size of data customers are using.
-# The effective max message size in this case is about 1.9 GB, see the max payload test.
-# This is due to internal hard gRPC limits. When we want to increase the message sizes
-# we'll have to implement chunking for large messages.
-_MAX_GRPC_MESSAGE_LENGTH = -1
-# Disable port reuse: fail if multiple Function Executor Servers attempt to bind to the
-# same port. This happens when Indexify users misconfigure the Servers. Disabling the port
-# reuse results in a clear error message on Server startup instead of obscure errors later
-# while Indexify cluster is serving tasks.
-# If we don't disable port reuse then a random Server gets the requests so wrong tasks get
-# routed to wrong servers.
-_REUSE_SERVER_PORT = 0
-
-GRPC_SERVER_OPTIONS = [
-    ("grpc.max_receive_message_length", _MAX_GRPC_MESSAGE_LENGTH),
-    ("grpc.max_send_message_length", _MAX_GRPC_MESSAGE_LENGTH),
-    ("grpc.so_reuseport", _REUSE_SERVER_PORT),
-]

indexify/function_executor/server.py

@@ -1,29 +0,0 @@
-from concurrent.futures import ThreadPoolExecutor
-
-import grpc
-
-from .proto.function_executor_pb2_grpc import add_FunctionExecutorServicer_to_server
-from .proto.server_configuration import GRPC_SERVER_OPTIONS
-from .service import Service
-
-# Temporary limit until we have a better way to control this.
-# This limits the number of concurrent tasks that Function Executor can run.
-MAX_RPC_CONCURRENCY = 100
-
-
-class Server:
-    def __init__(self, server_address: str, service: Service):
-        self._server_address: str = server_address
-        self._service: Service = service
-
-    def run(self):
-        """Runs Function Executor Service at the configured address."""
-        server = grpc.server(
-            thread_pool=ThreadPoolExecutor(max_workers=MAX_RPC_CONCURRENCY),
-            maximum_concurrent_rpcs=MAX_RPC_CONCURRENCY,
-            options=GRPC_SERVER_OPTIONS,
-        )
-        add_FunctionExecutorServicer_to_server(self._service, server)
-        server.add_insecure_port(self._server_address)
-        server.start()
-        server.wait_for_termination()