Properly handle multiple jobs per report

This commit is contained in:
2024-03-24 03:05:53 -04:00
parent 55469939d5
commit d3f5a1ab8d
3 changed files with 60 additions and 10 deletions

View File

@ -21,18 +21,19 @@ export async function onRequestPost(context: RequestContext) {
if (!value) return jsonError("Report is missing", 500);
const coconutData = (await context.env.DATA.get(`coconutjob_${id}`, {
const coconutData = (await context.env.DATA.get(`coconutdata_${id}`, {
type: "json",
})) as {
attachments: string[];
token: string;
} | null;
if (coconutData) {
const coconutResponsePromises = [];
const responsePromises = [];
for (const attachment of coconutData.attachments)
coconutResponsePromises.push(
for (const attachment of coconutData.attachments) {
const token = crypto.randomUUID();
responsePromises.push(
fetch("https://api.coconut.co/v2/jobs", {
body: JSON.stringify({
input: {
@ -48,8 +49,8 @@ export async function onRequestPost(context: RequestContext) {
},
notification: {
params: {
report_id: id,
token: coconutData.token,
attachment: attachment,
token,
},
type: "http",
url: `https://${context.request.headers.get("host")}/api/coconut`,
@ -76,9 +77,14 @@ export async function onRequestPost(context: RequestContext) {
},
method: "POST",
}),
context.env.DATA.put(`coconutjob_${attachment}`, token, {
expirationTtl: 600,
}),
);
}
await Promise.allSettled(coconutResponsePromises);
await Promise.allSettled(responsePromises);
await context.env.DATA.delete(`coconutdata_${id}`);
}
if (context.env.REPORTS_WEBHOOK) {