Skip to content

Commit 7574c69

Browse files
mpcgridmpcgrid
authored
feat(webapp): Add support for resetting idempotency keys (#2777)
Add support for resetting idempotency keys both from ui and sdk ## ✅ Checklist - [x] I have followed every step in the [contributing guide](https://github.com/triggerdotdev/trigger.dev/blob/main/CONTRIBUTING.md) - [x] The PR title follows the convention. - [x] I ran and tested the code works --- ## Testing - Created a new run with a idempotency idempotencyKey. - Started a new run with the same task and got redirected to the first run. - Deleted the key from the UI on the run details - Started a new run with the same task and it created a new one - Did the above steps using the SDK --- ## Changelog - Add new action route for resetting idempotency keys via UI - Add reset button in Idempotency section of run detail view - Added API and SDK for resetting imdepotency - Updated docs page for this feature --- ## Screenshots _[Screenshots]_ <img width="438" height="363" alt="Screenshot 2025-12-11 at 11 56 37" src="https://github.com/user-attachments/assets/30b8ef5e-8aac-4d04-b57a-9bf30d085dcb" />
1 parent 06cbe6e commit 7574c69

File tree

11 files changed

+353
-10
lines changed

11 files changed

+353
-10
lines changed

‎.changeset/quick-plums-tan.md‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
---
2+
"@trigger.dev/sdk": patch
3+
"@trigger.dev/core": patch
4+
---
5+
6+
Added support for idempotency reset

‎apps/webapp/app/routes/api.v1.idempotencyKeys.$key.reset.ts‎

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
import { json } from "@remix-run/server-runtime";
2+
import { ServiceValidationError } from "~/v3/services/baseService.server";
3+
import { z } from "zod";
4+
import { createActionApiRoute } from "~/services/routeBuilders/apiBuilder.server";
5+
import { ResetIdempotencyKeyService } from "~/v3/services/resetIdempotencyKey.server";
6+
import { logger } from "~/services/logger.server";
7+
8+
const ParamsSchema = z.object({
9+
key: z.string(),
10+
});
11+
12+
const BodySchema = z.object({
13+
taskIdentifier: z.string().min(1, "Task identifier is required"),
14+
});
15+
16+
export const { action } = createActionApiRoute(
17+
{
18+
params: ParamsSchema,
19+
body: BodySchema,
20+
allowJWT: true,
21+
corsStrategy: "all",
22+
authorization: {
23+
action: "write",
24+
resource: () => ({}),
25+
superScopes: ["write:runs", "admin"],
26+
},
27+
},
28+
async ({ params, body, authentication }) => {
29+
const service = new ResetIdempotencyKeyService();
30+
31+
try {
32+
const result = await service.call(
33+
params.key,
34+
body.taskIdentifier,
35+
authentication.environment
36+
);
37+
return json(result, { status: 200 });
38+
} catch (error) {
39+
if (error instanceof ServiceValidationError) {
40+
return json({ error: error.message }, { status: error.status ?? 400 });
41+
}
42+
43+
logger.error("Failed to reset idempotency key via API", {
44+
error: error instanceof Error ? { name: error.name, message: error.message, stack: error.stack } : String(error),
45+
});
46+
47+
return json({ error: "Internal Server Error" }, { status: 500 });
48+
}
49+
50+
}
51+
);

‎apps/webapp/app/routes/resources.orgs.$organizationSlug.projects.$projectParam.env.$envParam.runs.$runParam.idempotencyKey.reset.tsx‎

Lines changed: 127 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,127 @@
1+
import { parse } from "@conform-to/zod";
2+
import { type ActionFunction, json } from "@remix-run/node";
3+
import { z } from "zod";
4+
import { prisma } from "~/db.server";
5+
import { jsonWithErrorMessage } from "~/models/message.server";
6+
import { logger } from "~/services/logger.server";
7+
import { requireUserId } from "~/services/session.server";
8+
import { ResetIdempotencyKeyService } from "~/v3/services/resetIdempotencyKey.server";
9+
import { v3RunParamsSchema } from "~/utils/pathBuilder";
10+
11+
export const resetIdempotencyKeySchema = z.object({
12+
taskIdentifier: z.string().min(1, "Task identifier is required"),
13+
});
14+
15+
export const action: ActionFunction = async ({ request, params }) => {
16+
const userId = await requireUserId(request);
17+
const { projectParam, organizationSlug, envParam, runParam } =
18+
v3RunParamsSchema.parse(params);
19+
20+
const formData = await request.formData();
21+
const submission = parse(formData, { schema: resetIdempotencyKeySchema });
22+
23+
if (!submission.value) {
24+
return json(submission);
25+
}
26+
27+
try {
28+
const { taskIdentifier } = submission.value;
29+
30+
const taskRun = await prisma.taskRun.findFirst({
31+
where: {
32+
friendlyId: runParam,
33+
project: {
34+
slug: projectParam,
35+
organization: {
36+
slug: organizationSlug,
37+
members: {
38+
some: {
39+
userId,
40+
},
41+
},
42+
},
43+
},
44+
runtimeEnvironment: {
45+
slug: envParam,
46+
},
47+
},
48+
select: {
49+
id: true,
50+
idempotencyKey: true,
51+
taskIdentifier: true,
52+
runtimeEnvironmentId: true,
53+
},
54+
});
55+
56+
if (!taskRun) {
57+
submission.error = { runParam: ["Run not found"] };
58+
return json(submission);
59+
}
60+
61+
if (!taskRun.idempotencyKey) {
62+
return jsonWithErrorMessage(
63+
submission,
64+
request,
65+
"This run does not have an idempotency key"
66+
);
67+
}
68+
69+
if (taskRun.taskIdentifier !== taskIdentifier) {
70+
submission.error = { taskIdentifier: ["Task identifier does not match this run"] };
71+
return json(submission);
72+
}
73+
74+
const environment = await prisma.runtimeEnvironment.findUnique({
75+
where: {
76+
id: taskRun.runtimeEnvironmentId,
77+
},
78+
include: {
79+
project: {
80+
include: {
81+
organization: true,
82+
},
83+
},
84+
},
85+
});
86+
87+
if (!environment) {
88+
return jsonWithErrorMessage(
89+
submission,
90+
request,
91+
"Environment not found"
92+
);
93+
}
94+
95+
const service = new ResetIdempotencyKeyService();
96+
97+
await service.call(taskRun.idempotencyKey, taskIdentifier, {
98+
...environment,
99+
organizationId: environment.project.organizationId,
100+
organization: environment.project.organization,
101+
});
102+
103+
return json({ success: true });
104+
} catch (error) {
105+
if (error instanceof Error) {
106+
logger.error("Failed to reset idempotency key", {
107+
error: {
108+
name: error.name,
109+
message: error.message,
110+
stack: error.stack,
111+
},
112+
});
113+
return jsonWithErrorMessage(
114+
submission,
115+
request,
116+
`Failed to reset idempotency key: ${error.message}`
117+
);
118+
} else {
119+
logger.error("Failed to reset idempotency key", { error });
120+
return jsonWithErrorMessage(
121+
submission,
122+
request,
123+
`Failed to reset idempotency key: ${JSON.stringify(error)}`
124+
);
125+
}
126+
}
127+
};

‎apps/webapp/app/routes/resources.orgs.$organizationSlug.projects.$projectParam.env.$envParam.runs.$runParam.spans.$spanParam/route.tsx‎

Lines changed: 56 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import {
2+
ArrowPathIcon,
23
CheckIcon,
34
CloudArrowDownIcon,
45
EnvelopeIcon,
@@ -29,6 +30,7 @@ import { Header2, Header3 } from "~/components/primitives/Headers";
2930
import { Paragraph } from "~/components/primitives/Paragraph";
3031
import * as Property from "~/components/primitives/PropertyTable";
3132
import { Spinner } from "~/components/primitives/Spinner";
33+
import { toast } from "sonner";
3234
import {
3335
Table,
3436
TableBody,
@@ -40,6 +42,7 @@ import {
4042
import { TabButton, TabContainer } from "~/components/primitives/Tabs";
4143
import { TextLink } from "~/components/primitives/TextLink";
4244
import { InfoIconTooltip, SimpleTooltip } from "~/components/primitives/Tooltip";
45+
import { ToastUI } from "~/components/primitives/Toast";
4346
import { RunTimeline, RunTimelineEvent, SpanTimeline } from "~/components/run/RunTimeline";
4447
import { PacketDisplay } from "~/components/runs/v3/PacketDisplay";
4548
import { RunIcon } from "~/components/runs/v3/RunIcon";
@@ -69,6 +72,7 @@ import {
6972
v3BatchPath,
7073
v3DeploymentVersionPath,
7174
v3RunDownloadLogsPath,
75+
v3RunIdempotencyKeyResetPath,
7276
v3RunPath,
7377
v3RunRedirectPath,
7478
v3RunSpanPath,
@@ -81,6 +85,7 @@ import { CompleteWaitpointForm } from "../resources.orgs.$organizationSlug.proje
8185
import { requireUserId } from "~/services/session.server";
8286
import type { SpanOverride } from "~/v3/eventRepository/eventRepository.types";
8387
import { RealtimeStreamViewer } from "../resources.orgs.$organizationSlug.projects.$projectParam.env.$envParam.runs.$runParam.streams.$streamKey/route";
88+
import { action as resetIdempotencyKeyAction } from "../resources.orgs.$organizationSlug.projects.$projectParam.env.$envParam.runs.$runParam.idempotencyKey.reset";
8489

8590
export const loader = async ({ request, params }: LoaderFunctionArgs) => {
8691
const userId = await requireUserId(request);
@@ -293,6 +298,28 @@ function RunBody({
293298
const isAdmin = useHasAdminAccess();
294299
const { value, replace } = useSearchParams();
295300
const tab = value("tab");
301+
const resetFetcher = useTypedFetcher<typeof resetIdempotencyKeyAction>();
302+
303+
// Handle toast messages from the reset action
304+
useEffect(() => {
305+
if (resetFetcher.data && resetFetcher.state === "idle") {
306+
// Check if the response indicates success
307+
if (resetFetcher.data && typeof resetFetcher.data === "object" && "success" in resetFetcher.data && resetFetcher.data.success === true) {
308+
toast.custom(
309+
(t) => (
310+
<ToastUI
311+
variant="success"
312+
message="Idempotency key reset successfully"
313+
t={t as string}
314+
/>
315+
),
316+
{
317+
duration: 5000,
318+
}
319+
);
320+
}
321+
}
322+
}, [resetFetcher.data, resetFetcher.state]);
296323

297324
return (
298325
<div className="grid h-full max-h-full grid-rows-[2.5rem_2rem_1fr_3.25rem] overflow-hidden bg-background-bright">
@@ -543,17 +570,37 @@ function RunBody({
543570
<Property.Item>
544571
<Property.Label>Idempotency</Property.Label>
545572
<Property.Value>
546-
<div className="break-all">{run.idempotencyKey ? run.idempotencyKey : "–"}</div>
547-
{run.idempotencyKey && (
548-
<div>
549-
Expires:{" "}
550-
{run.idempotencyKeyExpiresAt ? (
551-
<DateTime date={run.idempotencyKeyExpiresAt} />
552-
) : (
553-
"–"
573+
<div className="flex items-start justify-between gap-2">
574+
<div className="flex-1">
575+
<div className="break-all">{run.idempotencyKey ? run.idempotencyKey : "–"}</div>
576+
{run.idempotencyKey && (
577+
<div>
578+
Expires:{" "}
579+
{run.idempotencyKeyExpiresAt ? (
580+
<DateTime date={run.idempotencyKeyExpiresAt} />
581+
) : (
582+
"–"
583+
)}
584+
</div>
554585
)}
555586
</div>
556-
)}
587+
{run.idempotencyKey && (
588+
<resetFetcher.Form
589+
method="post"
590+
action={v3RunIdempotencyKeyResetPath(organization, project, environment, { friendlyId: runParam })}
591+
>
592+
<input type="hidden" name="taskIdentifier" value={run.taskIdentifier} />
593+
<Button
594+
type="submit"
595+
variant="minimal/small"
596+
LeadingIcon={ArrowPathIcon}
597+
disabled={resetFetcher.state === "submitting"}
598+
>
599+
{resetFetcher.state === "submitting" ? "Resetting..." : "Reset"}
600+
</Button>
601+
</resetFetcher.Form>
602+
)}
603+
</div>
557604
</Property.Value>
558605
</Property.Item>
559606
<Property.Item>

‎apps/webapp/app/utils/pathBuilder.ts‎

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -329,6 +329,17 @@ export function v3RunStreamingPath(
329329
return `${v3RunPath(organization, project, environment, run)}/stream`;
330330
}
331331

332+
export function v3RunIdempotencyKeyResetPath(
333+
organization: OrgForPath,
334+
project: ProjectForPath,
335+
environment: EnvironmentForPath,
336+
run: v3RunForPath
337+
) {
338+
return `/resources/orgs/${organizationParam(organization)}/projects/${projectParam(
339+
project
340+
)}/env/${environmentParam(environment)}/runs/${run.friendlyId}/idempotencyKey/reset`;
341+
}
342+
332343
export function v3SchedulesPath(
333344
organization: OrgForPath,
334345
project: ProjectForPath,

‎apps/webapp/app/v3/services/resetIdempotencyKey.server.ts‎

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
import type { AuthenticatedEnvironment } from "~/services/apiAuth.server";
2+
import { BaseService, ServiceValidationError } from "./baseService.server";
3+
import { logger } from "~/services/logger.server";
4+
5+
export class ResetIdempotencyKeyService extends BaseService {
6+
public async call(
7+
idempotencyKey: string,
8+
taskIdentifier: string,
9+
authenticatedEnv: AuthenticatedEnvironment
10+
): Promise<{ id: string }> {
11+
const { count } = await this._prisma.taskRun.updateMany({
12+
where: {
13+
idempotencyKey,
14+
taskIdentifier,
15+
runtimeEnvironmentId: authenticatedEnv.id,
16+
},
17+
data: {
18+
idempotencyKey: null,
19+
idempotencyKeyExpiresAt: null,
20+
},
21+
});
22+
23+
if (count === 0) {
24+
throw new ServiceValidationError(
25+
`No runs found with idempotency key: ${idempotencyKey} and task: ${taskIdentifier}`,
26+
404
27+
);
28+
}
29+
30+
logger.info(
31+
`Reset idempotency key: ${idempotencyKey} for task: ${taskIdentifier} in env: ${authenticatedEnv.id}, affected ${count} run(s)`
32+
);
33+
34+
return { id: idempotencyKey };
35+
}
36+
}

‎docs/idempotency.mdx‎

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,29 @@ function hash(payload: any): string {
153153
}
154154
```
155155

156+
## Resetting idempotency keys
157+
158+
You can reset an idempotency key to clear it from all associated runs. This is useful if you need to allow a task to be triggered again with the same idempotency key.
159+
160+
When you reset an idempotency key, it will be cleared for all runs that match both the task identifier and the idempotency key in the current environment. This allows you to trigger the task again with the same key.
161+
162+
```ts
163+
import { idempotencyKeys } from "@trigger.dev/sdk";
164+
165+
// Reset an idempotency key for a specific task
166+
await idempotencyKeys.reset("my-task", "my-idempotency-key");
167+
```
168+
169+
The `reset` function requires both parameters:
170+
- `taskIdentifier`: The identifier of the task (e.g., `"my-task"`)
171+
- `idempotencyKey`: The idempotency key to reset
172+
173+
After resetting, any subsequent triggers with the same idempotency key will create new task runs instead of returning the existing ones.
174+
175+
<Note>
176+
Resetting an idempotency key only affects runs in the current environment. The reset is scoped to the specific task identifier and idempotency key combination.
177+
</Note>
178+
156179
## Important notes
157180

158181
Idempotency keys, even the ones scoped globally, are actually scoped to the task and the environment. This means that you cannot collide with keys from other environments (e.g. dev will never collide with prod), or to other projects and orgs.

0 commit comments

Comments
 (0)