engineering
/
cloudservices
3
0
Fork 1
Code Issues 5 Pull Requests Packages Projects Releases Wiki Activity

changes to memory

merge-requests/4/head
Matthew 2020-05-03 16:35:47 -04:00
parent 0a7ebc9f34
commit ee2ea41d8b
No known key found for this signature in database
GPG Key ID: 766BE43AE75F7559
1 changed files with 64 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

97
src/intervals/memory.ts
View File

@ -16,6 +16,10 @@ const memoryLimits = {
}; };
export default function memory(client: Client) { export default function memory(client: Client) {
const set = new Set<string>();
setInterval(() => {
set.clear();
}, 3600000);
setInterval(async () => { setInterval(async () => {
try { try {
const accounts = await client.db.Account.find(); const accounts = await client.db.Account.find();
@ -27,49 +31,76 @@ export default function memory(client: Client) {
const memoryConversion = mem / 1024 / 1024; const memoryConversion = mem / 1024 / 1024;
const userLimits: { soft?: number, hard?: number } = {}; const userLimits: { soft?: number, hard?: number } = {};
const tier: TierInterface = await client.db.Tier.findOne({ id: acc.tier }).lean().exec(); const tier: TierInterface = await client.db.Tier.findOne({ id: acc.tier }).lean().exec();
userLimits.soft = tier.resourceLimits.ram; userLimits.soft = tier.resourceLimits.ram - 50;
userLimits.hard = tier.resourceLimits.ram + 5; userLimits.hard = tier.resourceLimits.ram + 10;
/* if the user has exceeded their soft memory limit, which is the one described in the /* if the user has exceeded their soft memory limit, which is the one described in the
resource limit guidelines, we'll inform staff. resource limit guidelines, we'll inform staff.
*/ */
if (memoryConversion >= userLimits.soft) { if (memoryConversion >= userLimits.hard && set.has(acc.username)) {
client.signale.info(`RAM Soft Limit Reached | ${acc.username} | ${memoryConversion}/${userLimits.soft} MB`); client.signale.info(`RAM Hard Limit Reached | ${acc.username} | ${memoryConversion}/${userLimits.hard} MB`);
client.util.exec(`killall -9 -u ${acc.username}`);
const embed = new RichEmbed(); const embed = new RichEmbed();
if (client.users.get(acc.userID)) embed.setThumbnail(client.users.get(acc.userID).avatarURL); embed.setTitle('Resource Enforcement Notification');
embed.setDescription('Someone has reached the (hard) resource limit for their tier on RAM. The system has automatically killed all of their processes.');
embed.addField('User', `${acc.username} | <@${acc.userID}> | ${acc.userID}`, true); embed.addField('User', `${acc.username} | <@${acc.userID}> | ${acc.userID}`, true);
embed.addField('Tier', String(acc.tier), true); embed.addField('Tier', String(acc.tier), true);
embed.addField('Memory Usage', `${String(memoryConversion)} MB`, true); embed.addField('Memory Usage', `${String(memoryConversion)} MB`, true);
embed.addField('Memory Limit', `${String(userLimits.soft)} MB`, true); embed.addField('Memory Limit', `${String(userLimits.hard)} MB`, true);
client.createMessage(channelID, { embed });
client.util.createModerationLog(acc.userID, client.guilds.get('446067825673633794').members.get(client.user.id), 1, '[AUTO] Exceeded resource limit for RAM.');
client.util.transport.sendMail({
to: acc.emailAddress,
from: 'Library of Code sp-us | Cloud Services <help@libraryofcode.org>',
subject: 'Your account has been warned',
html: `
<h1>Library of Code sp-us | Cloud Services</h1>
<p>Your account has received an official warning from a Technician. Please get the underlying issue resolved to avoid <i>possible</i> moderative action.</p>
<p><strong>Reason:</strong> [AUTO] Exceeded resource limit for RAM.</p>
<p><strong>Moderator:</strong> ${client.user.username}</p>
<b><i>Library of Code sp-us | Support Team</i></b>
`,
});
client.createMessage(channelID, { embed });
set.delete(acc.username);
} else if (memoryConversion >= userLimits.soft && !set.has(acc.username)) {
client.signale.info(`RAM Soft Limit Reached | ${acc.username} | ${memoryConversion}/${userLimits.soft} MB`);
const embed = new RichEmbed();
if (client.users.get(acc.userID)) embed.setThumbnail(client.users.get(acc.userID).avatarURL);
embed.setTitle('Resource Limit Notification');
embed.setDescription('Someone has reached the (soft) resource limit for their tier on RAM.');
embed.addField('User', `${acc.username} | <@${acc.userID}> | ${acc.userID}`, true);
embed.addField('Tier', String(acc.tier), true);
embed.addField('Memory Usage', `${String(memoryConversion)} MB`, true);
embed.addField('Memory Limit', `${String(userLimits.hard)} MB`, true);
embed.setFooter(client.user.username, client.user.avatarURL); embed.setFooter(client.user.username, client.user.avatarURL);
embed.setTimestamp(); embed.setTimestamp();
await client.createMessage(channelID, { embed });
// if they exceed the hard limit, we'll kill all of their processes. const notifyEmbed = new RichEmbed()
if (memoryConversion >= userLimits.hard) { .setTitle('Cloud Account | Notification')
client.signale.info(`RAM Hard Limit Reached | ${acc.username} | ${memoryConversion}/${userLimits.hard} MB`); .setDescription(`You are about to reach your RAM resource limits, you are currently using '${String(memoryConversion)} MB' and your limit is '${String(userLimits.hard)} MB'. Please correct your usage to avoid further action.`)
client.util.exec(`killall -9 -u ${acc.username}`); .addField('Technician', 'SYSTEM', true)
embed.setTitle('Resource Enforcement Notification'); .setFooter(this.client.user.username, this.client.user.avatarURL)
embed.setDescription('Someone has reached the (hard) resource limit for their tier on RAM. The system has automatically killed all of their processes.'); .setTimestamp();
client.util.createModerationLog(acc.userID, client.guilds.get('446067825673633794').members.get(client.user.id), 1, '[AUTO] Exceeded resource limit for RAM.'); this.client.getDMChannel(acc.userID).then((channel) => {
client.util.transport.sendMail({ channel.createMessage({ notifyEmbed });
to: acc.emailAddress, });
from: 'Library of Code sp-us | Cloud Services <help@libraryofcode.org>', notifyEmbed.addField('User', `${acc.username} | <@${acc.userID}>`, true);
subject: 'Your account has been warned', this.client.createMessage('580950455581147146', { embed });
html: ` this.client.util.transport.sendMail({
<h1>Library of Code sp-us | Cloud Services</h1> to: acc.emailAddress,
<p>Your account has received an official warning from a Moderator. Please get the underlying issue resolved to avoid <i>possible</i> moderative action.</p> from: 'Library of Code sp-us | Cloud Services <help@libraryofcode.org>',
<p><strong>Reason:</strong> [AUTO] Exceeded resource limit for RAM.</p> subject: 'Notification',
<p><strong>Moderator:</strong> ${client.user.username}</p> html: `
<h1>Library of Code sp-us | Cloud Services</h1>
<b><i>Library of Code sp-us | Support Team</i></b> <p>You are about to reach your RAM resource limits, you are currently using '${String(memoryConversion)} MB' and your limit is '${String(userLimits.hard)} MB'. Please correct your usage to avoid further action.</p>
`, <p><strong>Technician:</strong> SYSTEM</p>
});
} else { <b><i>Library of Code sp-us | Support Team</i></b>
embed.setTitle('Resource Limit Notification'); `,
embed.setDescription('Someone has reached the (soft) resource limit for their tier on RAM.'); });
} set.add(acc.username);
// @ts-ignore
client.createMessage(channelID, { embed });
} }
} }
} catch (err) { } catch (err) {