Skip to content

Commit 0f69945

Browse files
SARomanchukSARomanchuk
authored andcommitted
issue:493
add: task_id for retry scheduled task
1 parent 34e280a commit 0f69945

File tree

3 files changed

+4
-0
lines changed

3 files changed

+4
-0
lines changed

taskiq/kicker.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -208,6 +208,7 @@ async def schedule_by_cron(
208208
labels=message.labels,
209209
args=message.args,
210210
kwargs=message.kwargs,
211+
task_id=self.custom_task_id,
211212
cron=cron_str,
212213
cron_offset=cron_offset,
213214
)
@@ -239,6 +240,7 @@ async def schedule_by_time(
239240
labels=message.labels,
240241
args=message.args,
241242
kwargs=message.kwargs,
243+
task_id=self.custom_task_id,
242244
time=time,
243245
)
244246
await source.add_schedule(scheduled)

taskiq/scheduler/scheduled_task/v1.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ class ScheduledTask(BaseModel):
1212
labels: Dict[str, Any]
1313
args: List[Any]
1414
kwargs: Dict[str, Any]
15+
task_id: Optional[str] = None
1516
schedule_id: str = Field(default_factory=lambda: uuid.uuid4().hex)
1617
cron: Optional[str] = None
1718
cron_offset: Optional[Union[str, timedelta]] = None

taskiq/scheduler/scheduled_task/v2.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,7 @@ class ScheduledTask(BaseModel):
1313
labels: Dict[str, Any]
1414
args: List[Any]
1515
kwargs: Dict[str, Any]
16+
task_id: Optional[str] = None
1617
schedule_id: str = Field(default_factory=lambda: uuid.uuid4().hex)
1718
cron: Optional[str] = None
1819
cron_offset: Optional[Union[str, timedelta]] = None

0 commit comments

Comments
 (0)