StyAndCo/zappyzep
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add more detailed automod profiling

Browse source
This commit is contained in:
Dragory 2021-10-05 21:09:03 +03:00
parent ff8d406e8a
commit c0c73751c9
No known key found for this signature in database
GPG key ID: 5F387BA66DF8AAC1
1 changed files with 32 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
backend/src/plugins/Automod/functions/runAutomod.ts
View file

@ -6,6 +6,7 @@ import { AutomodTriggerMatchResult } from "../helpers";
import { availableTriggers } from "../triggers/availableTriggers"; import { availableTriggers } from "../triggers/availableTriggers";
import { AutomodContext, AutomodPluginType } from "../types"; import { AutomodContext, AutomodPluginType } from "../types";
import { checkAndUpdateCooldown } from "./checkAndUpdateCooldown"; import { checkAndUpdateCooldown } from "./checkAndUpdateCooldown";
import { performance } from "perf_hooks";
export async function runAutomod(pluginData: GuildPluginData<AutomodPluginType>, context: AutomodContext) { export async function runAutomod(pluginData: GuildPluginData<AutomodPluginType>, context: AutomodContext) {
const userId = context.user?.id || context.member?.id || context.message?.user_id; const userId = context.user?.id || context.member?.id || context.message?.user_id;
@ -42,11 +43,15 @@ export async function runAutomod(pluginData: GuildPluginData<AutomodPluginType>,
continue; continue;
} }
const ruleStartTime = performance.now();
let matchResult: AutomodTriggerMatchResult<any> | null | undefined; let matchResult: AutomodTriggerMatchResult<any> | null | undefined;
let contexts: AutomodContext[] = []; let contexts: AutomodContext[] = [];
triggerLoop: for (const triggerItem of rule.triggers) { triggerLoop: for (const triggerItem of rule.triggers) {
for (const [triggerName, triggerConfig] of Object.entries(triggerItem)) { for (const [triggerName, triggerConfig] of Object.entries(triggerItem)) {
const triggerStartTime = performance.now();
const trigger = availableTriggers[triggerName]; const trigger = availableTriggers[triggerName];
matchResult = await trigger.match({ matchResult = await trigger.match({
ruleName, ruleName,
@ -83,8 +88,17 @@ export async function runAutomod(pluginData: GuildPluginData<AutomodPluginType>,
})) ?? ""; })) ?? "";
matchResult.fullSummary = `Triggered automod rule **${ruleName}**\n${matchResult.summary}`.trim(); matchResult.fullSummary = `Triggered automod rule **${ruleName}**\n${matchResult.summary}`.trim();
}
if (!rule.allow_further_rules) break triggerLoop; pluginData
.getKnubInstance()
.profiler.addDataPoint(
`automod:${pluginData.guild.id}:${ruleName}:triggers:${triggerName}`,
performance.now() - triggerStartTime,
);
if (matchResult && !rule.allow_further_rules) {
break triggerLoop;
} }
} }
} }
@ -95,6 +109,8 @@ export async function runAutomod(pluginData: GuildPluginData<AutomodPluginType>,
continue; continue;
} }
const actionStartTime = performance.now();
const action = availableActions[actionName]; const action = availableActions[actionName];
action.apply({ action.apply({
@ -104,9 +120,22 @@ export async function runAutomod(pluginData: GuildPluginData<AutomodPluginType>,
actionConfig, actionConfig,
matchResult, matchResult,
}); });
}
if (!rule.allow_further_rules) break; pluginData
.getKnubInstance()
.profiler.addDataPoint(
`automod:${pluginData.guild.id}:${ruleName}:actions:${actionName}`,
performance.now() - actionStartTime,
);
}
}
pluginData
.getKnubInstance()
.profiler.addDataPoint(`automod:${pluginData.guild.id}:${ruleName}`, performance.now() - ruleStartTime);
if (matchResult && !rule.allow_further_rules) {
break;
} }
} }
} }