Skip to content

Fairness Keys & Weights #1036

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

Merged
merged 4 commits into from
Aug 19, 2025
Merged

Fairness Keys & Weights #1036

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
b80f8e2
Update Core submodule
Sushisource Aug 19, 2025
ebc04d8
Update protos
Sushisource Aug 19, 2025
1a439f1
Add new rpcs
Sushisource Aug 19, 2025
d9fe5a4
Add fairness weights/keys
Sushisource Aug 19, 2025
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
4 changes: 4 additions & 0 deletions temporalio/api/batch/v1/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -3,10 +3,12 @@
BatchOperationDeletion,
BatchOperationInfo,
BatchOperationReset,
BatchOperationResetActivities,
BatchOperationSignal,
BatchOperationTermination,
BatchOperationTriggerWorkflowRule,
BatchOperationUnpauseActivities,
BatchOperationUpdateActivityOptions,
BatchOperationUpdateWorkflowExecutionOptions,
)

Expand All @@ -15,9 +17,11 @@
"BatchOperationDeletion",
"BatchOperationInfo",
"BatchOperationReset",
"BatchOperationResetActivities",
"BatchOperationSignal",
"BatchOperationTermination",
"BatchOperationTriggerWorkflowRule",
"BatchOperationUnpauseActivities",
"BatchOperationUpdateActivityOptions",
"BatchOperationUpdateWorkflowExecutionOptions",
]
73 changes: 54 additions & 19 deletions temporalio/api/batch/v1/message_pb2.py
View file
Open in desktop

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

177 changes: 175 additions & 2 deletions temporalio/api/batch/v1/message_pb2.pyi
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@ import google.protobuf.internal.containers
import google.protobuf.message
import google.protobuf.timestamp_pb2

import temporalio.api.activity.v1.message_pb2
import temporalio.api.common.v1.message_pb2
import temporalio.api.enums.v1.batch_operation_pb2
import temporalio.api.enums.v1.reset_pb2
Expand Down Expand Up @@ -336,9 +337,9 @@ class BatchOperationUnpauseActivities(google.protobuf.message.Message):
type: builtins.str
match_all: builtins.bool
reset_attempts: builtins.bool
"""Providing this flag will also reset the number of attempts."""
"""Setting this flag will also reset the number of attempts."""
reset_heartbeat: builtins.bool
"""Providing this flag will also reset the heartbeat details."""
"""Setting this flag will also reset the heartbeat details."""
@property
def jitter(self) -> google.protobuf.duration_pb2.Duration:
"""If set, the activity will start at a random time within the specified jitter
Expand Down Expand Up @@ -431,3 +432,175 @@ class BatchOperationTriggerWorkflowRule(google.protobuf.message.Message):
) -> typing_extensions.Literal["id", "spec"] | None: ...

global___BatchOperationTriggerWorkflowRule = BatchOperationTriggerWorkflowRule

class BatchOperationResetActivities(google.protobuf.message.Message):
"""BatchOperationResetActivities sends activity reset requests in a batch.
NOTE: keep in sync with temporalio.api.workflowservice.v1.ResetActivityRequest
"""

DESCRIPTOR: google.protobuf.descriptor.Descriptor

IDENTITY_FIELD_NUMBER: builtins.int
TYPE_FIELD_NUMBER: builtins.int
MATCH_ALL_FIELD_NUMBER: builtins.int
RESET_ATTEMPTS_FIELD_NUMBER: builtins.int
RESET_HEARTBEAT_FIELD_NUMBER: builtins.int
KEEP_PAUSED_FIELD_NUMBER: builtins.int
JITTER_FIELD_NUMBER: builtins.int
RESTORE_ORIGINAL_OPTIONS_FIELD_NUMBER: builtins.int
identity: builtins.str
"""The identity of the worker/client."""
type: builtins.str
match_all: builtins.bool
reset_attempts: builtins.bool
"""Setting this flag will also reset the number of attempts."""
reset_heartbeat: builtins.bool
"""Setting this flag will also reset the heartbeat details."""
keep_paused: builtins.bool
"""If activity is paused, it will remain paused after reset"""
@property
def jitter(self) -> google.protobuf.duration_pb2.Duration:
"""If set, the activity will start at a random time within the specified jitter
duration, introducing variability to the start time.
"""
restore_original_options: builtins.bool
"""If set, the activity options will be restored to the defaults.
Default options are then options activity was created with.
They are part of the first ActivityTaskScheduled event.
"""
def __init__(
self,
*,
identity: builtins.str = ...,
type: builtins.str = ...,
match_all: builtins.bool = ...,
reset_attempts: builtins.bool = ...,
reset_heartbeat: builtins.bool = ...,
keep_paused: builtins.bool = ...,
jitter: google.protobuf.duration_pb2.Duration | None = ...,
restore_original_options: builtins.bool = ...,
) -> None: ...
def HasField(
self,
field_name: typing_extensions.Literal[
"activity",
b"activity",
"jitter",
b"jitter",
"match_all",
b"match_all",
"type",
b"type",
],
) -> builtins.bool: ...
def ClearField(
self,
field_name: typing_extensions.Literal[
"activity",
b"activity",
"identity",
b"identity",
"jitter",
b"jitter",
"keep_paused",
b"keep_paused",
"match_all",
b"match_all",
"reset_attempts",
b"reset_attempts",
"reset_heartbeat",
b"reset_heartbeat",
"restore_original_options",
b"restore_original_options",
"type",
b"type",
],
) -> None: ...
def WhichOneof(
self, oneof_group: typing_extensions.Literal["activity", b"activity"]
) -> typing_extensions.Literal["type", "match_all"] | None: ...

global___BatchOperationResetActivities = BatchOperationResetActivities

class BatchOperationUpdateActivityOptions(google.protobuf.message.Message):
"""BatchOperationUpdateActivityOptions sends an update-activity-options requests in a batch.
NOTE: keep in sync with temporalio.api.workflowservice.v1.UpdateActivityRequest
"""

DESCRIPTOR: google.protobuf.descriptor.Descriptor

IDENTITY_FIELD_NUMBER: builtins.int
TYPE_FIELD_NUMBER: builtins.int
MATCH_ALL_FIELD_NUMBER: builtins.int
ACTIVITY_OPTIONS_FIELD_NUMBER: builtins.int
UPDATE_MASK_FIELD_NUMBER: builtins.int
RESTORE_ORIGINAL_FIELD_NUMBER: builtins.int
identity: builtins.str
"""The identity of the worker/client."""
type: builtins.str
match_all: builtins.bool
@property
def activity_options(
self,
) -> temporalio.api.activity.v1.message_pb2.ActivityOptions:
"""Update Activity options. Partial updates are accepted and controlled by update_mask."""
@property
def update_mask(self) -> google.protobuf.field_mask_pb2.FieldMask:
"""Controls which fields from `activity_options` will be applied"""
restore_original: builtins.bool
"""If set, the activity options will be restored to the default.
Default options are then options activity was created with.
They are part of the first ActivityTaskScheduled event.
This flag cannot be combined with any other option; if you supply
restore_original together with other options, the request will be rejected.
"""
def __init__(
self,
*,
identity: builtins.str = ...,
type: builtins.str = ...,
match_all: builtins.bool = ...,
activity_options: temporalio.api.activity.v1.message_pb2.ActivityOptions
| None = ...,
update_mask: google.protobuf.field_mask_pb2.FieldMask | None = ...,
restore_original: builtins.bool = ...,
) -> None: ...
def HasField(
self,
field_name: typing_extensions.Literal[
"activity",
b"activity",
"activity_options",
b"activity_options",
"match_all",
b"match_all",
"type",
b"type",
"update_mask",
b"update_mask",
],
) -> builtins.bool: ...
def ClearField(
self,
field_name: typing_extensions.Literal[
"activity",
b"activity",
"activity_options",
b"activity_options",
"identity",
b"identity",
"match_all",
b"match_all",
"restore_original",
b"restore_original",
"type",
b"type",
"update_mask",
b"update_mask",
],
) -> None: ...
def WhichOneof(
self, oneof_group: typing_extensions.Literal["activity", b"activity"]
) -> typing_extensions.Literal["type", "match_all"] | None: ...

global___BatchOperationUpdateActivityOptions = BatchOperationUpdateActivityOptions
Loading
Loading