Skip to content

feat: subscribe to submitted function #198

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
hanchchch wants to merge 2 commits into
base: main
Choose a base branch
from
Open

feat: subscribe to submitted function #198

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
7c0e384
feat: subscribe to submitted function
hanchchch Feb 9, 2026
4fda3ec
feat: only one subscriber at a time
hanchchch Feb 10, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions src/isolate/server/definitions/server.proto
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,9 @@ service Isolate {

// Submit a function to be run without waiting for results.
rpc Submit (SubmitRequest) returns (SubmitResponse) {}

// Subscribe to a submitted function.
rpc Subscribe (SubscribeRequest) returns (stream PartialRunResult) {}

// Set the metadata for a task.
rpc SetMetadata (SetMetadataRequest) returns (SetMetadataResponse) {}
Expand Down Expand Up @@ -44,6 +47,10 @@ message SubmitRequest {
TaskMetadata metadata = 2;
}

message SubscribeRequest {
string task_id = 1;
}

message TaskMetadata {
// Labels to attach to the logs.
map<string, string> logger_labels = 1;
Expand Down
56 changes: 29 additions & 27 deletions src/isolate/server/definitions/server_pb2.py
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

89 changes: 52 additions & 37 deletions src/isolate/server/definitions/server_pb2.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,37 +2,33 @@
@generated by mypy-protobuf. Do not edit manually!
isort:skip_file
"""

import builtins
import collections.abc
from isolate.connections.grpc.definitions import common_pb2
import google.protobuf.descriptor
import google.protobuf.internal.containers
import google.protobuf.message
import google.protobuf.struct_pb2
import sys

if sys.version_info >= (3, 8):
import typing as typing_extensions
else:
import typing_extensions
import typing

DESCRIPTOR: google.protobuf.descriptor.FileDescriptor

@typing_extensions.final
@typing.final
class BoundFunction(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

ENVIRONMENTS_FIELD_NUMBER: builtins.int
FUNCTION_FIELD_NUMBER: builtins.int
SETUP_FUNC_FIELD_NUMBER: builtins.int
STREAM_LOGS_FIELD_NUMBER: builtins.int
stream_logs: builtins.bool
@property
def environments(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___EnvironmentDefinition]: ...
@property
def function(self) -> common_pb2.SerializedObject: ...
@property
def setup_func(self) -> common_pb2.SerializedObject: ...
stream_logs: builtins.bool
def __init__(
self,
*,
Expand All @@ -41,13 +37,13 @@ class BoundFunction(google.protobuf.message.Message):
setup_func: common_pb2.SerializedObject | None = ...,
stream_logs: builtins.bool = ...,
) -> None: ...
def HasField(self, field_name: typing_extensions.Literal["_setup_func", b"_setup_func", "function", b"function", "setup_func", b"setup_func"]) -> builtins.bool: ...
def ClearField(self, field_name: typing_extensions.Literal["_setup_func", b"_setup_func", "environments", b"environments", "function", b"function", "setup_func", b"setup_func", "stream_logs", b"stream_logs"]) -> None: ...
def WhichOneof(self, oneof_group: typing_extensions.Literal["_setup_func", b"_setup_func"]) -> typing_extensions.Literal["setup_func"] | None: ...
def HasField(self, field_name: typing.Literal["_setup_func", b"_setup_func", "function", b"function", "setup_func", b"setup_func"]) -> builtins.bool: ...
def ClearField(self, field_name: typing.Literal["_setup_func", b"_setup_func", "environments", b"environments", "function", b"function", "setup_func", b"setup_func", "stream_logs", b"stream_logs"]) -> None: ...
def WhichOneof(self, oneof_group: typing.Literal["_setup_func", b"_setup_func"]) -> typing.Literal["setup_func"] | None: ...

global___BoundFunction = BoundFunction

@typing_extensions.final
@typing.final
class EnvironmentDefinition(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -56,24 +52,25 @@ class EnvironmentDefinition(google.protobuf.message.Message):
FORCE_FIELD_NUMBER: builtins.int
kind: builtins.str
"""Kind of the isolate environment."""
force: builtins.bool
"""Whether to force-create this environment or not."""
@property
def configuration(self) -> google.protobuf.struct_pb2.Struct:
"""A free-form definition of environment properties."""
force: builtins.bool
"""Whether to force-create this environment or not."""

def __init__(
self,
*,
kind: builtins.str = ...,
configuration: google.protobuf.struct_pb2.Struct | None = ...,
force: builtins.bool = ...,
) -> None: ...
def HasField(self, field_name: typing_extensions.Literal["configuration", b"configuration"]) -> builtins.bool: ...
def ClearField(self, field_name: typing_extensions.Literal["configuration", b"configuration", "force", b"force", "kind", b"kind"]) -> None: ...
def HasField(self, field_name: typing.Literal["configuration", b"configuration"]) -> builtins.bool: ...
def ClearField(self, field_name: typing.Literal["configuration", b"configuration", "force", b"force", "kind", b"kind"]) -> None: ...

global___EnvironmentDefinition = EnvironmentDefinition

@typing_extensions.final
@typing.final
class SubmitRequest(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -82,25 +79,42 @@ class SubmitRequest(google.protobuf.message.Message):
@property
def function(self) -> global___BoundFunction:
"""The function to run."""

@property
def metadata(self) -> global___TaskMetadata:
"""Task metadata."""

def __init__(
self,
*,
function: global___BoundFunction | None = ...,
metadata: global___TaskMetadata | None = ...,
) -> None: ...
def HasField(self, field_name: typing_extensions.Literal["function", b"function", "metadata", b"metadata"]) -> builtins.bool: ...
def ClearField(self, field_name: typing_extensions.Literal["function", b"function", "metadata", b"metadata"]) -> None: ...
def HasField(self, field_name: typing.Literal["function", b"function", "metadata", b"metadata"]) -> builtins.bool: ...
def ClearField(self, field_name: typing.Literal["function", b"function", "metadata", b"metadata"]) -> None: ...

global___SubmitRequest = SubmitRequest

@typing_extensions.final
@typing.final
class SubscribeRequest(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

TASK_ID_FIELD_NUMBER: builtins.int
task_id: builtins.str
def __init__(
self,
*,
task_id: builtins.str = ...,
) -> None: ...
def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...

global___SubscribeRequest = SubscribeRequest

@typing.final
class TaskMetadata(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

@typing_extensions.final
@typing.final
class LoggerLabelsEntry(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -114,22 +128,23 @@ class TaskMetadata(google.protobuf.message.Message):
key: builtins.str = ...,
value: builtins.str = ...,
) -> None: ...
def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...

LOGGER_LABELS_FIELD_NUMBER: builtins.int
@property
def logger_labels(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
"""Labels to attach to the logs."""

def __init__(
self,
*,
logger_labels: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
) -> None: ...
def ClearField(self, field_name: typing_extensions.Literal["logger_labels", b"logger_labels"]) -> None: ...
def ClearField(self, field_name: typing.Literal["logger_labels", b"logger_labels"]) -> None: ...

global___TaskMetadata = TaskMetadata

@typing_extensions.final
@typing.final
class SubmitResponse(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -140,11 +155,11 @@ class SubmitResponse(google.protobuf.message.Message):
*,
task_id: builtins.str = ...,
) -> None: ...
def ClearField(self, field_name: typing_extensions.Literal["task_id", b"task_id"]) -> None: ...
def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...

global___SubmitResponse = SubmitResponse

@typing_extensions.final
@typing.final
class SetMetadataRequest(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -159,12 +174,12 @@ class SetMetadataRequest(google.protobuf.message.Message):
task_id: builtins.str = ...,
metadata: global___TaskMetadata | None = ...,
) -> None: ...
def HasField(self, field_name: typing_extensions.Literal["metadata", b"metadata"]) -> builtins.bool: ...
def ClearField(self, field_name: typing_extensions.Literal["metadata", b"metadata", "task_id", b"task_id"]) -> None: ...
def HasField(self, field_name: typing.Literal["metadata", b"metadata"]) -> builtins.bool: ...
def ClearField(self, field_name: typing.Literal["metadata", b"metadata", "task_id", b"task_id"]) -> None: ...

global___SetMetadataRequest = SetMetadataRequest

@typing_extensions.final
@typing.final
class SetMetadataResponse(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -174,7 +189,7 @@ class SetMetadataResponse(google.protobuf.message.Message):

global___SetMetadataResponse = SetMetadataResponse

@typing_extensions.final
@typing.final
class ListRequest(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -184,7 +199,7 @@ class ListRequest(google.protobuf.message.Message):

global___ListRequest = ListRequest

@typing_extensions.final
@typing.final
class TaskInfo(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -195,11 +210,11 @@ class TaskInfo(google.protobuf.message.Message):
*,
task_id: builtins.str = ...,
) -> None: ...
def ClearField(self, field_name: typing_extensions.Literal["task_id", b"task_id"]) -> None: ...
def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...

global___TaskInfo = TaskInfo

@typing_extensions.final
@typing.final
class ListResponse(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -211,11 +226,11 @@ class ListResponse(google.protobuf.message.Message):
*,
tasks: collections.abc.Iterable[global___TaskInfo] | None = ...,
) -> None: ...
def ClearField(self, field_name: typing_extensions.Literal["tasks", b"tasks"]) -> None: ...
def ClearField(self, field_name: typing.Literal["tasks", b"tasks"]) -> None: ...

global___ListResponse = ListResponse

@typing_extensions.final
@typing.final
class CancelRequest(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand All @@ -226,11 +241,11 @@ class CancelRequest(google.protobuf.message.Message):
*,
task_id: builtins.str = ...,
) -> None: ...
def ClearField(self, field_name: typing_extensions.Literal["task_id", b"task_id"]) -> None: ...
def ClearField(self, field_name: typing.Literal["task_id", b"task_id"]) -> None: ...

global___CancelRequest = CancelRequest

@typing_extensions.final
@typing.final
class CancelResponse(google.protobuf.message.Message):
DESCRIPTOR: google.protobuf.descriptor.Descriptor

Expand Down
Loading