Mirrors/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2025-12-28 06:34:11 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

change reset/delete website to interactive transaction with timeout
Some checks failed
Node.js CI / build (push) Has been cancelled
Details

Browse Source 
Closes #3905
This commit is contained in:
Francis Cao 2025-12-22 10:59:33 -08:00
parent 4c0c9e6aa0
commit b0ed4bddb6
1 changed files with 53 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
src/queries/prisma/website.ts
View File

@ -132,42 +132,46 @@ export async function updateWebsite(
}
export async function resetWebsite(websiteId: string) {
const { client, transaction } = prisma;
const { transaction } = prisma;
const cloudMode = !!process.env.CLOUD_MODE;
return transaction(
[
client.revenue.deleteMany({
async tx => {
await tx.revenue.deleteMany({
where: { websiteId },
}),
client.eventData.deleteMany({
});
await tx.eventData.deleteMany({
where: { websiteId },
}),
client.sessionData.deleteMany({
});
await tx.sessionData.deleteMany({
where: { websiteId },
}),
client.websiteEvent.deleteMany({
});
await tx.websiteEvent.deleteMany({
where: { websiteId },
}),
client.session.deleteMany({
});
await tx.session.deleteMany({
where: { websiteId },
}),
client.website.update({
});
const website = await tx.website.update({
where: { id: websiteId },
data: {
resetAt: new Date(),
},
}),
],
});
return website;
},
{
timeout: 30000,
},
).then(async data => {
if (cloudMode) {
await redis.client.set(
`website:${websiteId}`,
data.find(website => website.id),
);
await redis.client.set(`website:${websiteId}`, data);
}
return data;
@ -175,43 +179,52 @@ export async function resetWebsite(websiteId: string) {
}
export async function deleteWebsite(websiteId: string) {
const { client, transaction } = prisma;
const { transaction } = prisma;
const cloudMode = !!process.env.CLOUD_MODE;
return transaction(
[
client.revenue.deleteMany({
async tx => {
await tx.revenue.deleteMany({
where: { websiteId },
}),
client.eventData.deleteMany({
});
await tx.eventData.deleteMany({
where: { websiteId },
}),
client.sessionData.deleteMany({
});
await tx.sessionData.deleteMany({
where: { websiteId },
}),
client.websiteEvent.deleteMany({
});
await tx.websiteEvent.deleteMany({
where: { websiteId },
}),
client.session.deleteMany({
});
await tx.session.deleteMany({
where: { websiteId },
}),
client.report.deleteMany({
});
await tx.report.deleteMany({
where: { websiteId },
}),
client.segment.deleteMany({
});
await tx.segment.deleteMany({
where: { websiteId },
}),
cloudMode
? client.website.update({
});
const website = cloudMode
? await tx.website.update({
data: {
deletedAt: new Date(),
},
where: { id: websiteId },
})
: client.website.delete({
: await tx.website.delete({
where: { id: websiteId },
}),
],
});
return website;
},
{
timeout: 30000,
},