wool 0.1rc20__py3-none-any.whl

wool/core/loadbalancer/base.py

@@ -0,0 +1,125 @@
+from __future__ import annotations
+
+from types import MappingProxyType
+from typing import TYPE_CHECKING
+from typing import AsyncIterator
+from typing import Callable
+from typing import Final
+from typing import Protocol
+from typing import TypeAlias
+from typing import runtime_checkable
+
+from wool._resource_pool import Resource
+from wool.core.discovery.base import WorkerInfo
+from wool.core.worker.connection import WorkerConnection
+
+if TYPE_CHECKING:
+    from wool._work import WoolTask
+
+
+# public
+ConnectionResourceFactory: TypeAlias = Callable[[], Resource[WorkerConnection]]
+
+
+# public
+class NoWorkersAvailable(Exception):
+    """Raised when no workers are available for task dispatch.
+
+    This exception indicates that either no workers exist in the worker pool
+    or all available workers have been tried and failed.
+    """
+
+
+# public
+@runtime_checkable
+class LoadBalancerLike(Protocol):
+    """Protocol for load balancers that dispatch tasks to workers.
+
+    Load balancers implementing this protocol operate on a
+    :class:`LoadBalancerContext` to access workers and their connection
+    factories. The context provides isolation, allowing a single load balancer
+    instance to service multiple worker pools with independent state.
+
+    The dispatch method accepts a :class:`WoolTask` and returns an async
+    iterator that yields task results from the worker.
+    """
+
+    async def dispatch(
+        self,
+        task: WoolTask,
+        *,
+        context: LoadBalancerContext,
+        timeout: float | None = None,
+    ) -> AsyncIterator: ...
+
+
+# public
+class LoadBalancerContext:
+    """Isolated load balancing context for a single worker pool.
+
+    Manages workers and their connection resource factories for a specific
+    worker pool, enabling load balancer instances to service multiple pools
+    with independent state and worker lists.
+    """
+
+    _workers: Final[dict[WorkerInfo, ConnectionResourceFactory]]
+
+    def __init__(self):
+        self._workers = {}
+
+    @property
+    def workers(self) -> MappingProxyType[WorkerInfo, ConnectionResourceFactory]:
+        """Read-only view of workers in this context.
+
+        :returns:
+            Immutable mapping of worker information to connection resource
+            factories. Changes to the underlying context are reflected in
+            the returned proxy.
+        """
+        return MappingProxyType(self._workers)
+
+    def add_worker(
+        self,
+        worker_info: WorkerInfo,
+        connection_resource_factory: ConnectionResourceFactory,
+    ):
+        """Add a worker to this context.
+
+        :param worker_info:
+            Information about the worker to add.
+        :param connection_resource_factory:
+            Factory function that creates connection resources for this worker.
+        """
+        self._workers[worker_info] = connection_resource_factory
+
+    def update_worker(
+        self,
+        worker_info: WorkerInfo,
+        connection_resource_factory: ConnectionResourceFactory,
+        *,
+        upsert: bool = False,
+    ):
+        """Update an existing worker's connection resource factory.
+
+        :param worker_info:
+            Information about the worker to update. If the worker is not
+            present in the context, this method does nothing.
+        :param connection_resource_factory:
+            New factory function that creates connection resources for this
+            worker.
+        :param upsert:
+            Flag indicating whether or not to add the worker if it's not
+            already in the context.
+        """
+        if upsert or worker_info in self._workers:
+            self._workers[worker_info] = connection_resource_factory
+
+    def remove_worker(self, worker_info: WorkerInfo):
+        """Remove a worker from this context.
+
+        :param worker_info:
+            Information about the worker to remove. If the worker is not
+            present in the context, this method does nothing.
+        """
+        if worker_info in self._workers:
+            del self._workers[worker_info]

wool/core/loadbalancer/roundrobin.py

@@ -0,0 +1,101 @@
+from __future__ import annotations
+
+import itertools
+from typing import TYPE_CHECKING
+from typing import AsyncIterator
+from typing import Final
+
+from wool.core.worker.connection import TransientRpcError
+
+from .base import LoadBalancerContext
+from .base import LoadBalancerLike
+from .base import NoWorkersAvailable
+
+if TYPE_CHECKING:
+    from wool._work import WoolTask
+
+
+# public
+class RoundRobinLoadBalancer(LoadBalancerLike):
+    """Round-robin load balancer for distributing tasks across workers.
+
+    Distributes tasks evenly across available workers using a simple round-robin
+    algorithm. Workers are managed through :class:`LoadBalancerContext` instances
+    passed to the dispatch method, enabling a single load balancer to service
+    multiple worker pools with independent state.
+
+    Automatically handles worker failures by trying the next worker in the
+    round-robin cycle. Workers that encounter transient errors remain in the
+    context, while workers that fail with non-transient errors are removed from
+    the context's worker list.
+    """
+
+    _index: Final[dict[LoadBalancerContext, int]]
+
+    def __init__(self):
+        self._index = {}
+
+    async def dispatch(
+        self,
+        task: WoolTask,
+        *,
+        context: LoadBalancerContext,
+        timeout: float | None = None,
+    ) -> AsyncIterator:
+        """Dispatch a task to the next available worker using round-robin.
+
+        Tries workers in one round-robin cycle until dispatch succeeds.
+        Workers that fail to schedule the task with a non-transient error are
+        removed from the context's worker list.
+
+        :param task:
+            The :class:`WoolTask` instance to dispatch to the worker.
+        :param context:
+            The :class:`LoadBalancerContext` containing workers to dispatch to.
+        :param timeout:
+            Timeout in seconds for each dispatch attempt. If ``None``, no
+            timeout is applied.
+        :returns:
+            A streaming dispatch result that yields worker responses.
+        :raises NoWorkersAvailable:
+            If no healthy workers are available to schedule the task.
+        """
+        checkpoint = None
+
+        # Initialize index for this context if not present
+        if context not in self._index:
+            self._index[context] = 0
+
+        while context.workers:
+            if self._index[context] >= len(context.workers):
+                self._index[context] = 0
+
+            worker_info, connection_resource_factory = next(
+                itertools.islice(
+                    context.workers.items(),
+                    self._index[context],
+                    self._index[context] + 1,
+                )
+            )
+
+            if checkpoint is None:
+                checkpoint = worker_info.uid
+            elif worker_info.uid == checkpoint:
+                break
+
+            async with connection_resource_factory() as connection:
+                try:
+                    result = await connection.dispatch(task, timeout=timeout)
+                except TransientRpcError:
+                    self._index[context] = self._index[context] + 1
+                    continue
+                except Exception:
+                    context.remove_worker(worker_info)
+                    if worker_info.uid == checkpoint:
+                        checkpoint = None
+                    continue
+                else:
+                    self._index[context] = self._index[context] + 1
+                    return result
+
+        raise NoWorkersAvailable("No healthy workers available for dispatch")

wool/core/protobuf/__init__.py

@@ -0,0 +1,18 @@
+import os
+import sys
+from typing import Protocol
+
+sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
+
+from . import task as task
+from . import worker as worker
+
+
+class AddServicerToServerProtocol(Protocol):
+    @staticmethod
+    def __call__(servicer, server) -> None: ...
+
+
+add_to_server: dict[type[worker.WorkerServicer], AddServicerToServerProtocol] = {
+    worker.WorkerServicer: worker.add_WorkerServicer_to_server,
+}

wool/core/protobuf/exception.py

@@ -0,0 +1,3 @@
+class ProtobufImportError(ImportError):
+    def __init__(self, exception: ImportError):
+        super().__init__(f"{str(exception)} - ensure protocol buffers are compiled.")

wool/core/protobuf/task.py

@@ -0,0 +1,11 @@
+try:
+    from wool.core.protobuf.task_pb2 import Exception
+    from wool.core.protobuf.task_pb2 import Result
+    from wool.core.protobuf.task_pb2 import Task
+    from wool.core.protobuf.task_pb2 import Worker as Worker
+except ImportError as e:
+    from wool.core.protobuf.exception import ProtobufImportError
+
+    raise ProtobufImportError(e) from e
+
+__all__ = ["Exception", "Result", "Task", "Worker"]

wool/core/protobuf/task_pb2.py

@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: task.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'task.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\ntask.proto\x12\x17wool.core.protobuf.task\"s\n\x04Task\x12\n\n\x02id\x18\x01 \x01(\t\x12\x10\n\x08\x63\x61llable\x18\x02 \x01(\x0c\x12\x0c\n\x04\x61rgs\x18\x03 \x01(\x0c\x12\x0e\n\x06kwargs\x18\x04 \x01(\x0c\x12\x0e\n\x06\x63\x61ller\x18\x05 \x01(\t\x12\r\n\x05proxy\x18\x07 \x01(\x0c\x12\x10\n\x08proxy_id\x18\x08 \x01(\t\"\x16\n\x06Result\x12\x0c\n\x04\x64ump\x18\x01 \x01(\x0c\"\x19\n\tException\x12\x0c\n\x04\x64ump\x18\x01 \x01(\x0c\"%\n\x06Worker\x12\n\n\x02id\x18\x01 \x01(\t\x12\x0f\n\x07\x61\x64\x64ress\x18\x02 \x01(\tb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'task_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_TASK']._serialized_start=39
+  _globals['_TASK']._serialized_end=154
+  _globals['_RESULT']._serialized_start=156
+  _globals['_RESULT']._serialized_end=178
+  _globals['_EXCEPTION']._serialized_start=180
+  _globals['_EXCEPTION']._serialized_end=205
+  _globals['_WORKER']._serialized_start=207
+  _globals['_WORKER']._serialized_end=244
+# @@protoc_insertion_point(module_scope)

wool/core/protobuf/task_pb2.pyi

@@ -0,0 +1,43 @@
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import ClassVar as _ClassVar, Optional as _Optional
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Task(_message.Message):
+    __slots__ = ("id", "callable", "args", "kwargs", "caller", "proxy", "proxy_id")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    CALLABLE_FIELD_NUMBER: _ClassVar[int]
+    ARGS_FIELD_NUMBER: _ClassVar[int]
+    KWARGS_FIELD_NUMBER: _ClassVar[int]
+    CALLER_FIELD_NUMBER: _ClassVar[int]
+    PROXY_FIELD_NUMBER: _ClassVar[int]
+    PROXY_ID_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    callable: bytes
+    args: bytes
+    kwargs: bytes
+    caller: str
+    proxy: bytes
+    proxy_id: str
+    def __init__(self, id: _Optional[str] = ..., callable: _Optional[bytes] = ..., args: _Optional[bytes] = ..., kwargs: _Optional[bytes] = ..., caller: _Optional[str] = ..., proxy: _Optional[bytes] = ..., proxy_id: _Optional[str] = ...) -> None: ...
+
+class Result(_message.Message):
+    __slots__ = ("dump",)
+    DUMP_FIELD_NUMBER: _ClassVar[int]
+    dump: bytes
+    def __init__(self, dump: _Optional[bytes] = ...) -> None: ...
+
+class Exception(_message.Message):
+    __slots__ = ("dump",)
+    DUMP_FIELD_NUMBER: _ClassVar[int]
+    dump: bytes
+    def __init__(self, dump: _Optional[bytes] = ...) -> None: ...
+
+class Worker(_message.Message):
+    __slots__ = ("id", "address")
+    ID_FIELD_NUMBER: _ClassVar[int]
+    ADDRESS_FIELD_NUMBER: _ClassVar[int]
+    id: str
+    address: str
+    def __init__(self, id: _Optional[str] = ..., address: _Optional[str] = ...) -> None: ...

wool/core/protobuf/task_pb2_grpc.py

@@ -0,0 +1,24 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+
+GRPC_GENERATED_VERSION = '1.76.0'
+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},'
+        + ' but the generated code in task_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}.'
+    )

wool/core/protobuf/worker.py

@@ -0,0 +1,26 @@
+try:
+    from wool.core.protobuf.worker_pb2 import Ack
+    from wool.core.protobuf.worker_pb2 import Nack
+    from wool.core.protobuf.worker_pb2 import Response
+    from wool.core.protobuf.worker_pb2 import StopRequest
+    from wool.core.protobuf.worker_pb2 import Void
+    from wool.core.protobuf.worker_pb2 import WorkerInfo
+    from wool.core.protobuf.worker_pb2_grpc import WorkerServicer
+    from wool.core.protobuf.worker_pb2_grpc import WorkerStub
+    from wool.core.protobuf.worker_pb2_grpc import add_WorkerServicer_to_server
+except ImportError as e:
+    from wool.core.protobuf.exception import ProtobufImportError
+
+    raise ProtobufImportError(e) from e
+
+__all__ = [
+    "Ack",
+    "Nack",
+    "Response",
+    "StopRequest",
+    "Void",
+    "WorkerInfo",
+    "WorkerServicer",
+    "WorkerStub",
+    "add_WorkerServicer_to_server",
+]

wool/core/protobuf/worker_pb2.py

@@ -0,0 +1,53 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
+# source: worker.proto
+# Protobuf Python Version: 6.31.1
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import runtime_version as _runtime_version
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    6,
+    31,
+    1,
+    '',
+    'worker.proto'
+)
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+import task_pb2 as task__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x0cworker.proto\x12\x19wool.core.protobuf.worker\x1a\ntask.proto\"\xe1\x01\n\x08Response\x12-\n\x03\x61\x63k\x18\x01 \x01(\x0b\x32\x1e.wool.core.protobuf.worker.AckH\x00\x12/\n\x04nack\x18\x02 \x01(\x0b\x32\x1f.wool.core.protobuf.worker.NackH\x00\x12\x31\n\x06result\x18\x03 \x01(\x0b\x32\x1f.wool.core.protobuf.task.ResultH\x00\x12\x37\n\texception\x18\x04 \x01(\x0b\x32\".wool.core.protobuf.task.ExceptionH\x00\x42\t\n\x07payload\"\x05\n\x03\x41\x63k\"\x16\n\x04Nack\x12\x0e\n\x06reason\x18\x01 \x01(\t\"\x1e\n\x0bStopRequest\x12\x0f\n\x07timeout\x18\x01 \x01(\x02\"\xd0\x01\n\nWorkerInfo\x12\x0b\n\x03uid\x18\x01 \x01(\t\x12\x0c\n\x04host\x18\x02 \x01(\t\x12\x0c\n\x04port\x18\x03 \x01(\x05\x12\x0b\n\x03pid\x18\x04 \x01(\x05\x12\x0f\n\x07version\x18\x05 \x01(\t\x12\x0c\n\x04tags\x18\x06 \x03(\t\x12?\n\x05\x65xtra\x18\x07 \x03(\x0b\x32\x30.wool.core.protobuf.worker.WorkerInfo.ExtraEntry\x1a,\n\nExtraEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\"\x06\n\x04Void2\xab\x01\n\x06Worker\x12P\n\x08\x64ispatch\x12\x1d.wool.core.protobuf.task.Task\x1a#.wool.core.protobuf.worker.Response0\x01\x12O\n\x04stop\x12&.wool.core.protobuf.worker.StopRequest\x1a\x1f.wool.core.protobuf.worker.Voidb\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'worker_pb2', _globals)
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_WORKERINFO_EXTRAENTRY']._loaded_options = None
+  _globals['_WORKERINFO_EXTRAENTRY']._serialized_options = b'8\001'
+  _globals['_RESPONSE']._serialized_start=56
+  _globals['_RESPONSE']._serialized_end=281
+  _globals['_ACK']._serialized_start=283
+  _globals['_ACK']._serialized_end=288
+  _globals['_NACK']._serialized_start=290
+  _globals['_NACK']._serialized_end=312
+  _globals['_STOPREQUEST']._serialized_start=314
+  _globals['_STOPREQUEST']._serialized_end=344
+  _globals['_WORKERINFO']._serialized_start=347
+  _globals['_WORKERINFO']._serialized_end=555
+  _globals['_WORKERINFO_EXTRAENTRY']._serialized_start=511
+  _globals['_WORKERINFO_EXTRAENTRY']._serialized_end=555
+  _globals['_VOID']._serialized_start=557
+  _globals['_VOID']._serialized_end=563
+  _globals['_WORKER']._serialized_start=566
+  _globals['_WORKER']._serialized_end=737
+# @@protoc_insertion_point(module_scope)

wool/core/protobuf/worker_pb2.pyi

@@ -0,0 +1,65 @@
+import task_pb2 as _task_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from collections.abc import Iterable as _Iterable, Mapping as _Mapping
+from typing import ClassVar as _ClassVar, Optional as _Optional, Union as _Union
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class Response(_message.Message):
+    __slots__ = ("ack", "nack", "result", "exception")
+    ACK_FIELD_NUMBER: _ClassVar[int]
+    NACK_FIELD_NUMBER: _ClassVar[int]
+    RESULT_FIELD_NUMBER: _ClassVar[int]
+    EXCEPTION_FIELD_NUMBER: _ClassVar[int]
+    ack: Ack
+    nack: Nack
+    result: _task_pb2.Result
+    exception: _task_pb2.Exception
+    def __init__(self, ack: _Optional[_Union[Ack, _Mapping]] = ..., nack: _Optional[_Union[Nack, _Mapping]] = ..., result: _Optional[_Union[_task_pb2.Result, _Mapping]] = ..., exception: _Optional[_Union[_task_pb2.Exception, _Mapping]] = ...) -> None: ...
+
+class Ack(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class Nack(_message.Message):
+    __slots__ = ("reason",)
+    REASON_FIELD_NUMBER: _ClassVar[int]
+    reason: str
+    def __init__(self, reason: _Optional[str] = ...) -> None: ...
+
+class StopRequest(_message.Message):
+    __slots__ = ("timeout",)
+    TIMEOUT_FIELD_NUMBER: _ClassVar[int]
+    timeout: float
+    def __init__(self, timeout: _Optional[float] = ...) -> None: ...
+
+class WorkerInfo(_message.Message):
+    __slots__ = ("uid", "host", "port", "pid", "version", "tags", "extra")
+    class ExtraEntry(_message.Message):
+        __slots__ = ("key", "value")
+        KEY_FIELD_NUMBER: _ClassVar[int]
+        VALUE_FIELD_NUMBER: _ClassVar[int]
+        key: str
+        value: str
+        def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
+    UID_FIELD_NUMBER: _ClassVar[int]
+    HOST_FIELD_NUMBER: _ClassVar[int]
+    PORT_FIELD_NUMBER: _ClassVar[int]
+    PID_FIELD_NUMBER: _ClassVar[int]
+    VERSION_FIELD_NUMBER: _ClassVar[int]
+    TAGS_FIELD_NUMBER: _ClassVar[int]
+    EXTRA_FIELD_NUMBER: _ClassVar[int]
+    uid: str
+    host: str
+    port: int
+    pid: int
+    version: str
+    tags: _containers.RepeatedScalarFieldContainer[str]
+    extra: _containers.ScalarMap[str, str]
+    def __init__(self, uid: _Optional[str] = ..., host: _Optional[str] = ..., port: _Optional[int] = ..., pid: _Optional[int] = ..., version: _Optional[str] = ..., tags: _Optional[_Iterable[str]] = ..., extra: _Optional[_Mapping[str, str]] = ...) -> None: ...
+
+class Void(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...

wool/core/protobuf/worker_pb2_grpc.py

@@ -0,0 +1,141 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+import warnings
+
+from . import task_pb2 as task__pb2
+from . import worker_pb2 as worker__pb2
+
+GRPC_GENERATED_VERSION = '1.76.0'
+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},'
+        + ' but the generated code in worker_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 WorkerStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.dispatch = channel.unary_stream(
+                '/wool.core.protobuf.worker.Worker/dispatch',
+                request_serializer=task__pb2.Task.SerializeToString,
+                response_deserializer=worker__pb2.Response.FromString,
+                _registered_method=True)
+        self.stop = channel.unary_unary(
+                '/wool.core.protobuf.worker.Worker/stop',
+                request_serializer=worker__pb2.StopRequest.SerializeToString,
+                response_deserializer=worker__pb2.Void.FromString,
+                _registered_method=True)
+
+
+class WorkerServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def dispatch(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def stop(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_WorkerServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'dispatch': grpc.unary_stream_rpc_method_handler(
+                    servicer.dispatch,
+                    request_deserializer=task__pb2.Task.FromString,
+                    response_serializer=worker__pb2.Response.SerializeToString,
+            ),
+            'stop': grpc.unary_unary_rpc_method_handler(
+                    servicer.stop,
+                    request_deserializer=worker__pb2.StopRequest.FromString,
+                    response_serializer=worker__pb2.Void.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'wool.core.protobuf.worker.Worker', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('wool.core.protobuf.worker.Worker', rpc_method_handlers)
+
+
+ # This class is part of an EXPERIMENTAL API.
+class Worker(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def dispatch(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_stream(
+            request,
+            target,
+            '/wool.core.protobuf.worker.Worker/dispatch',
+            task__pb2.Task.SerializeToString,
+            worker__pb2.Response.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
+
+    @staticmethod
+    def stop(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,
+            '/wool.core.protobuf.worker.Worker/stop',
+            worker__pb2.StopRequest.SerializeToString,
+            worker__pb2.Void.FromString,
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)