Hessenuk/DiscordTickets
Hessenuk/DiscordTickets
2
0
Fork 0
mirror of https://github.com/Hessenuk/DiscordTickets.git synced 2025-02-23 10:51:22 +02:00
Code Issues Packages Projects Releases Wiki Activity

perf: single transaction for archiving messages

Browse Source
This commit is contained in:
Isaac 2025-02-12 21:08:22 +00:00
parent 440a9b745c
commit b5384bca6c
No known key found for this signature in database
GPG Key ID: 0DE40AE37BBA5C33
1 changed files with 112 additions and 87 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

141
src/lib/tickets/archiver.js
View File

@ -31,17 +31,31 @@ module.exports = class TicketArchiver {
}
}
const channels = message.mentions.channels;
const channels = [...message.mentions.channels.values()];
const members = [...message.mentions.members.values()];
const roles = [...message.mentions.roles.values()];
// const ticket = { connect: { id: ticketId } };
const worker = await reusable('crypto');
try {
const queries = [];
if (message.member) {
members.push(message.member);
roles.push(hoistedRole(message.member));
} else {
this.client.log.warn('Message member does not exist');
await this.client.prisma.archivedUser.upsert({
create: {},
queries.push(
this.client.prisma.archivedUser.upsert({
create: {
ticketId,
userId: 'default',
},
select: { ticketId: true }, // default is to return all scalar fields
update: {},
where: {
ticketId_userId: {
@ -49,18 +63,23 @@ module.exports = class TicketArchiver {
userId: 'default',
},
},
});
}),
);
}
for (const role of roles) {
const data = {
colour: role.hexColor.slice(1),
name: role.name,
roleId: role.id,
ticket: { connect: { id: ticketId } },
};
await this.client.prisma.archivedRole.upsert({
create: data,
queries.push(
this.client.prisma.archivedRole.upsert({
create: {
...data,
roleId: role.id,
ticketId,
},
select: { ticketId: true },
update: data,
where: {
ticketId_roleId: {
@ -68,11 +87,10 @@ module.exports = class TicketArchiver {
ticketId,
},
},
});
}),
);
}
const worker = await reusable('crypto');
try {
for (const member of members) {
const data = {
avatar: member.avatar || member.user.avatar, // TODO: save avatar in user/avatars/
@ -80,12 +98,16 @@ module.exports = class TicketArchiver {
discriminator: member.user.discriminator,
displayName: member.displayName ? await worker.encrypt(member.displayName) : null,
roleId: !!member && hoistedRole(member).id,
ticketId,
userId: member.user.id,
username: await worker.encrypt(member.user.username),
};
await this.client.prisma.archivedUser.upsert({
create: data,
queries.push(
this.client.prisma.archivedUser.upsert({
create: {
...data,
ticketId,
userId: member.user.id,
},
select: { ticketId: true },
update: data,
where: {
ticketId_userId: {
@ -93,46 +115,20 @@ module.exports = class TicketArchiver {
userId: member.user.id,
},
},
});
}),
);
}
let reference;
if (message.reference) reference = await message.fetchReference();
const messageD = {
author: {
connect: {
ticketId_userId: {
ticketId,
userId: message.author?.id || 'default',
},
},
},
content: await worker.encrypt(
JSON.stringify({
attachments: [...message.attachments.values()],
components: [...message.components.values()],
content: message.content,
embeds: message.embeds.map(embed => ({ ...embed })),
reference: reference ? reference.id : null,
}),
),
createdAt: message.createdAt,
edited: !!message.editedAt,
external,
id: message.id,
};
return await this.client.prisma.ticket.update({
data: {
archivedChannels: {
upsert: channels.map(channel => {
for (const channel of channels) {
const data = {
channelId: channel.id,
name: channel.name,
ticketId,
};
return {
queries.push(
this.client.prisma.archivedChannel.upsert({
create: data,
select: { ticketId: true },
update: data,
where: {
ticketId_channelId: {
@ -140,19 +136,48 @@ module.exports = class TicketArchiver {
ticketId,
},
},
};
}),
);
}
const data = {
// author: {
// connect: {
// ticketId_userId: {
// ticketId,
// userId: message.author?.id || 'default',
// },
// },
// },
content: await worker.encrypt(
JSON.stringify({
attachments: [...message.attachments.values()],
components: [...message.components.values()],
content: message.content,
embeds: message.embeds.map(embed => ({ ...embed })),
reference: message.reference?.messageId ?? null,
}),
),
createdAt: message.createdAt,
edited: !!message.editedAt,
external,
};
queries.push(
this.client.prisma.archivedMessage.upsert({
create: {
...data,
authorId: message.author?.id || 'default',
id: message.id,
ticketId,
},
archivedMessages: {
upsert: {
create: messageD,
update: messageD,
select: { ticketId: true },
update: data,
where: { id: message.id },
},
},
},
where: { id: ticketId },
});
}),
);
return await this.client.prisma.$transaction(queries);
} finally {
await worker.terminate();
}