watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2025-12-06 01:18:00 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

add cascading deletes to revenue/segment table

Browse source 
Closes #3798
This commit is contained in:
Francis Cao 2025-12-02 09:41:46 -08:00
parent f5896f071b
commit 935517ce3a
1 changed files with 41 additions and 29 deletions
Download patch file Download diff file Expand all files Collapse all files

70
src/queries/prisma/website.ts
View file

@ -137,6 +137,9 @@ export async function resetWebsite(websiteId: string) {
return transaction(
[
client.revenue.deleteMany({
where: { websiteId },
}),
client.eventData.deleteMany({
where: { websiteId },
}),
@ -175,35 +178,44 @@ export async function deleteWebsite(websiteId: string) {
const { client, transaction } = prisma;
const cloudMode = !!process.env.CLOUD_MODE;
return transaction([
client.eventData.deleteMany({
where: { websiteId },
}),
client.sessionData.deleteMany({
where: { websiteId },
}),
client.websiteEvent.deleteMany({
where: { websiteId },
}),
client.session.deleteMany({
where: { websiteId },
}),
client.report.deleteMany({
where: {
websiteId,
},
}),
cloudMode
? client.website.update({
data: {
deletedAt: new Date(),
},
where: { id: websiteId },
})
: client.website.delete({
where: { id: websiteId },
}),
]).then(async data => {
return transaction(
[
client.revenue.deleteMany({
where: { websiteId },
}),
client.eventData.deleteMany({
where: { websiteId },
}),
client.sessionData.deleteMany({
where: { websiteId },
}),
client.websiteEvent.deleteMany({
where: { websiteId },
}),
client.session.deleteMany({
where: { websiteId },
}),
client.report.deleteMany({
where: { websiteId },
}),
client.segment.deleteMany({
where: { websiteId },
}),
cloudMode
? client.website.update({
data: {
deletedAt: new Date(),
},
where: { id: websiteId },
})
: client.website.delete({
where: { id: websiteId },
}),
],
{
timeout: 30000,
},
).then(async data => {
if (cloudMode) {
await redis.client.del(`website:${websiteId}`);
}