mirror of
https://github.com/bitburner-official/bitburner-src.git
synced 2024-11-08 08:43:53 +01:00
NETSCRIPT: Rework disableLog for efficiency (#1589)
The current implementation was naive; disableLog("ALL") was storing a key for every function, and iterating over a different object to do it (when iterating over objects is quite slow). The common cases of Bitburner (and especially batching, where efficiency matters most) are either never disabling anything, or disabling "ALL". This optimizes for these two cases, at the expense of slightly more complicated code to deal with the less-common edge cases.
This commit is contained in:
parent
385a9dc11d
commit
34db6e8b26
@ -131,7 +131,7 @@ export class WorkerScript {
|
||||
}
|
||||
|
||||
shouldLog(fn: string): boolean {
|
||||
return this.disableLogs[fn] == null;
|
||||
return !(this.disableLogs.ALL || this.disableLogs[fn]);
|
||||
}
|
||||
|
||||
log(func: string, txt: () => string): void {
|
||||
|
@ -495,37 +495,46 @@ export const ns: InternalAPI<NSFull> = {
|
||||
},
|
||||
disableLog: (ctx) => (_fn) => {
|
||||
const fn = helpers.string(ctx, "fn", _fn);
|
||||
if (fn === "ALL") {
|
||||
for (const fn of Object.keys(possibleLogs)) {
|
||||
ctx.workerScript.disableLogs[fn] = true;
|
||||
}
|
||||
helpers.log(ctx, () => `Disabled logging for all functions`);
|
||||
} else if (possibleLogs[fn] === undefined) {
|
||||
if (possibleLogs[fn] === undefined) {
|
||||
throw helpers.errorMessage(ctx, `Invalid argument: ${fn}.`);
|
||||
}
|
||||
if (fn === "ALL") {
|
||||
ctx.workerScript.disableLogs = allDisabled;
|
||||
// No need to log here, it's been disabled.
|
||||
} else {
|
||||
ctx.workerScript.disableLogs[fn] = true;
|
||||
helpers.log(ctx, () => `Disabled logging for ${fn}`);
|
||||
// We don't track individual log entries when all are disabled.
|
||||
if (!ctx.workerScript.disableLogs["ALL"]) {
|
||||
ctx.workerScript.disableLogs[fn] = true;
|
||||
helpers.log(ctx, () => `Disabled logging for ${fn}`);
|
||||
}
|
||||
}
|
||||
},
|
||||
enableLog: (ctx) => (_fn) => {
|
||||
const fn = helpers.string(ctx, "fn", _fn);
|
||||
if (fn === "ALL") {
|
||||
for (const fn of Object.keys(possibleLogs)) {
|
||||
delete ctx.workerScript.disableLogs[fn];
|
||||
}
|
||||
helpers.log(ctx, () => `Enabled logging for all functions`);
|
||||
} else if (possibleLogs[fn] === undefined) {
|
||||
if (possibleLogs[fn] === undefined) {
|
||||
throw helpers.errorMessage(ctx, `Invalid argument: ${fn}.`);
|
||||
}
|
||||
delete ctx.workerScript.disableLogs[fn];
|
||||
helpers.log(ctx, () => `Enabled logging for ${fn}`);
|
||||
if (fn === "ALL") {
|
||||
ctx.workerScript.disableLogs = {};
|
||||
helpers.log(ctx, () => `Enabled logging for all functions`);
|
||||
} else {
|
||||
if (ctx.workerScript.disableLogs["ALL"]) {
|
||||
// As an optimization, we normally store only that key, but we have to
|
||||
// expand it out to all keys at this point.
|
||||
// Conveniently, possibleLogs serves as a model for "all keys disabled."
|
||||
ctx.workerScript.disableLogs = Object.assign({}, possibleLogs, { ALL: false, [fn]: false });
|
||||
} else {
|
||||
ctx.workerScript.disableLogs[fn] = false;
|
||||
}
|
||||
helpers.log(ctx, () => `Enabled logging for ${fn}`);
|
||||
}
|
||||
},
|
||||
isLogEnabled: (ctx) => (_fn) => {
|
||||
const fn = helpers.string(ctx, "fn", _fn);
|
||||
if (possibleLogs[fn] === undefined) {
|
||||
throw helpers.errorMessage(ctx, `Invalid argument: ${fn}.`);
|
||||
}
|
||||
return !ctx.workerScript.disableLogs[fn];
|
||||
return ctx.workerScript.shouldLog(fn);
|
||||
},
|
||||
getScriptLogs:
|
||||
(ctx) =>
|
||||
@ -1840,7 +1849,13 @@ export function NetscriptFunctions(ws: WorkerScript): NSFull {
|
||||
return NSProxy(ws, ns, [], { args: ws.args.slice(), pid: ws.pid, enums });
|
||||
}
|
||||
|
||||
const possibleLogs = Object.fromEntries([...getFunctionNames(ns, "")].map((a) => [a, true]));
|
||||
const possibleLogs = Object.fromEntries(getFunctionNames(ns, "").map((a) => [a, true]));
|
||||
possibleLogs.ALL = true;
|
||||
|
||||
// We reuse this object for *all* scripts that disable all keys, to prevent memory growth.
|
||||
// Any script that needs a custom set of values will use a fresh object.
|
||||
const allDisabled = { ALL: true } as const;
|
||||
|
||||
/** Provides an array of all function names on a nested object */
|
||||
function getFunctionNames(obj: object, prefix: string): string[] {
|
||||
const functionNames: string[] = [];
|
||||
|
87
test/jest/Netscript/DisableLog.test.ts
Normal file
87
test/jest/Netscript/DisableLog.test.ts
Normal file
@ -0,0 +1,87 @@
|
||||
import type { Script } from "../../../src/Script/Script";
|
||||
import type { ScriptFilePath } from "../../../src/Paths/ScriptFilePath";
|
||||
import { FormatsNeedToChange } from "../../../src/ui/formatNumber";
|
||||
import { Server } from "../../../src/Server/Server";
|
||||
import { RunningScript } from "../../../src/Script/RunningScript";
|
||||
import { CompletedProgramName } from "../../../src/Programs/Enums";
|
||||
import { WorkerScript } from "../../../src/Netscript/WorkerScript";
|
||||
import { NetscriptFunctions } from "../../../src/NetscriptFunctions";
|
||||
import { AddToAllServers, DeleteServer } from "../../../src/Server/AllServers";
|
||||
|
||||
test("Edge cases of disableLog", function () {
|
||||
// Ensure that formatting functions work properly
|
||||
FormatsNeedToChange.emit();
|
||||
let server;
|
||||
try {
|
||||
server = new Server({ hostname: "home", adminRights: true, maxRam: 8 });
|
||||
server.programs.push(CompletedProgramName.bruteSsh, CompletedProgramName.ftpCrack);
|
||||
AddToAllServers(server);
|
||||
// We don't need this script to be runnable, it just needs to exist so that
|
||||
// we can create a RunningScript and WorkerScript object.
|
||||
expect(server.writeToScriptFile("test.js" as ScriptFilePath, "")).toEqual({ overwritten: false });
|
||||
const script = server.scripts.get("test.js" as ScriptFilePath) as Script;
|
||||
|
||||
const runningScript = new RunningScript(script, 2);
|
||||
const ws = new WorkerScript(runningScript, 1, NetscriptFunctions);
|
||||
|
||||
const ns = ws.env.vars;
|
||||
|
||||
// Generate logs in a specific pattern that checks edge cases in
|
||||
// disableLog. We want to check various combinations of things that
|
||||
// are/aren't disabled, as well as previously disabled.
|
||||
ns.brutessh("home");
|
||||
ns.ftpcrack("home");
|
||||
ns.print("before disableLog ALL");
|
||||
|
||||
expect(() => ns.disableLog("all")).toThrow("Invalid argument: all.");
|
||||
ns.disableLog("ALL");
|
||||
|
||||
ns.brutessh("home");
|
||||
ns.ftpcrack("home");
|
||||
ns.print("after disableLog ALL");
|
||||
|
||||
ns.enableLog("brutessh");
|
||||
ns.brutessh("home");
|
||||
ns.ftpcrack("home");
|
||||
ns.print("after enableLog brutessh");
|
||||
|
||||
ns.disableLog("brutessh");
|
||||
ns.enableLog("ftpcrack");
|
||||
ns.brutessh("home");
|
||||
ns.ftpcrack("home");
|
||||
ns.print("after enableLog ftpcrack");
|
||||
|
||||
ns.enableLog("ftpcrack");
|
||||
ns.brutessh("home");
|
||||
ns.ftpcrack("home");
|
||||
|
||||
ns.print("after redundant enable");
|
||||
ns.disableLog("ALL");
|
||||
ns.brutessh("home");
|
||||
ns.ftpcrack("home");
|
||||
ns.enableLog("ALL");
|
||||
ns.brutessh("home");
|
||||
ns.ftpcrack("home");
|
||||
ns.print("end");
|
||||
|
||||
expect(runningScript.logs).toEqual([
|
||||
"brutessh: Executed BruteSSH.exe on 'home' to open SSH port (22).",
|
||||
"ftpcrack: Executed FTPCrack.exe on 'home' to open FTP port (21).",
|
||||
"before disableLog ALL",
|
||||
"disableLog: Invalid argument: all.",
|
||||
"after disableLog ALL",
|
||||
"brutessh: SSH Port (22) already opened on 'home'.",
|
||||
"after enableLog brutessh",
|
||||
"ftpcrack: FTP Port (21) already opened on 'home'.",
|
||||
"after enableLog ftpcrack",
|
||||
"ftpcrack: FTP Port (21) already opened on 'home'.",
|
||||
"after redundant enable",
|
||||
"enableLog: Enabled logging for all functions",
|
||||
"brutessh: SSH Port (22) already opened on 'home'.",
|
||||
"ftpcrack: FTP Port (21) already opened on 'home'.",
|
||||
"end",
|
||||
]);
|
||||
} finally {
|
||||
DeleteServer(server.hostname);
|
||||
}
|
||||
});
|
Loading…
Reference in New Issue
Block a user