Compare commits

..

No commits in common. "68512ec8718d70bb44532c6e86be4c1d06e9993d" and "a37fd81052f296f010e64d1a66f3e045d4a0e0b4" have entirely different histories.

3 changed files with 29 additions and 116 deletions

View File

@ -94,7 +94,12 @@ class Controller extends EventEmitter {
const {
projectName,
shardCount = 1,
shardOptions = {},
serverOptions = {} as ServerOptions,
logger = {},
discord = {},
databases = {},
rabbitConfig = {} as BrokerOptions
} = this.#_options;
this.#_logger.info(`Spawning ${shardCount} shards`);
@ -119,12 +124,22 @@ class Controller extends EventEmitter {
process.stdin.on('data', (data) => {
const raw = data.toString('utf-8');
const words = raw.split(' ').map(word => word.trim());
this.#handleStdin(words);
this._handleStdin(words);
});
const promises = [];
for (let i = 0; i < shardCount; i++) {
const shard = this.createShard();
const shard = new Shard(this, i, {
serverOptions: {
...serverOptions, logger, discord, databases, rabbitConfig
},
...shardOptions,
env: this.#_options.env,
path: this.#_serverFilePath
});
this.#_logger.attach(shard);
this._setListeners(shard);
this.#_shards.set(i, shard);
promises.push(shard.spawn());
// setTimeout(() => {
@ -138,44 +153,13 @@ class Controller extends EventEmitter {
}
createShard () {
const ids = this.shards.map(s => s.id);
const id = ids.length ? Math.max(...ids) + 1 : 0;
const shard = new Shard(this, id, this.shardOptions);
this.shards.set(shard.id, shard);
this.#_logger.attach(shard);
this.#setListeners(shard);
return shard;
}
get shardOptions () {
const {
shardOptions = {},
serverOptions = {} as ServerOptions,
logger = {},
discord = {},
databases = {},
rabbitConfig = {} as BrokerOptions
} = this.#_options;
return {
serverOptions: {
...serverOptions, logger, discord, databases, rabbitConfig
},
...shardOptions,
env: this.#_options.env,
path: this.#_serverFilePath
};
}
#handleMessage (shard: Shard, msg: IPCMessage) {
_handleMessage (shard: Shard, msg: IPCMessage) {
if (msg._logger)
return;
this.#_logger.debug(`Message from ${shard.id}: ${inspect(msg)}`);
}
#setListeners (shard: Shard) {
_setListeners (shard: Shard) {
shard.on('death', () => this.#_logger.info(`Shard ${shard.id} has died`));
shard.on('fatal', ({ message }) => this.#_logger.warn(`Shard ${shard.id} has died fatally: ${message}`));
shard.on('shutdown', () => this.#_logger.info(`Shard ${shard.id} is shutting down gracefully`));
@ -185,10 +169,10 @@ class Controller extends EventEmitter {
shard.on('error', (err) => this.#_logger.error(`Shard ${shard.id} ran into an error:\n${err.stack}`));
shard.on('warn', (msg) => this.#_logger.warn(`Warning from shard ${shard.id}: ${msg}`, { broadcast: true }));
shard.on('message', (msg) => this.#handleMessage(shard, msg));
shard.on('message', (msg) => this._handleMessage(shard, msg));
}
async #handleStdin (words: string[]) {
async _handleStdin (words: string[]) {
words = words.filter(word => word.length);
if (!words.length)
@ -211,8 +195,7 @@ class Controller extends EventEmitter {
if (rest.args.help) {
if (command.help)
// eslint-disable-next-line no-console
return console.log(command.help);
return command.help;
return this.#_logger.info(`Help for ${command.name} unavailable at this time`);
}

View File

@ -1,6 +1,6 @@
import BaseCommand from "../BaseCommand.js";
import Controller from "../Controller.js";
import { ArgsResult, OptionType } from '@navy.gif/commandparser';
import { OptionType } from '@navy.gif/commandparser';
class StartCommand extends BaseCommand {
@ -12,44 +12,12 @@ class StartCommand extends BaseCommand {
aliases: [ 's' ],
flag: true,
type: OptionType.INTEGER
}, {
name: 'amount',
type: OptionType.INTEGER
}]
});
}
async execute ({ args }: { subcommand: string, args: ArgsResult }): Promise<string> {
const { shard, amount } = args;
const { shards } = this.controller;
const n = amount?.value as number || 0;
let id = -1;
if (shard)
id = shard.value as number;
let out = '';
if (id >= 0) {
const s = shards.get(id);
if (!s)
return `Shard ${id} doesn't exist`;
if (s.ready)
return `Shard ${id} is already running`;
await s.spawn();
out += `Started shard ${id}`;
}
if (n > 0) {
for (let i = 0; i < n; i++) {
const s = this.controller.createShard();
await s.spawn(true);
}
out += `\nSpawned ${n} shards`;
}
return out;
execute (_message: unknown, _args?: unknown): Promise<unknown> {
throw new Error("Method not implemented.");
}
}

View File

@ -1,53 +1,15 @@
import { ArgsResult, OptionType } from "@navy.gif/commandparser";
import BaseCommand from "../BaseCommand.js";
import Controller from "../Controller.js";
class StopCommand extends BaseCommand {
constructor (controller: Controller) {
super(controller, {
name: 'stop',
options: [{
name: 'shard',
type: OptionType.INTEGER,
flag: true,
}, {
name: 'all',
flag: true,
type: OptionType.BOOLEAN,
valueOptional: true,
defaultValue: true
}, {
name: 'prune',
flag: true,
type: OptionType.BOOLEAN,
valueOptional: true,
defaultValue: true
}]
name: 'stop'
});
}
async execute ({ args }: { subcommand: string, args: ArgsResult }) {
const id = args.shard?.value;
const all = args.all?.value || false;
const prune = args.prune?.value || false;
if (typeof id !== 'number' && !all)
return `Must supply either --all or --shard option`;
const { shards } = this.controller;
const targets = all ? shards : shards.filter(shard => shard.id === id);
const ids = targets.map(s => s.id);
for (const shard of targets.values()) {
await shard.kill();
if (prune)
shards.delete(shard.id);
}
let out = `Stopped ${all ? `all shards (${ids.join(', ')})` : `shard ${id}`}.`;
if (prune)
out += ' Pruned stopped shards';
return out;
async execute (_message: unknown, _args?: unknown) {
//
}
}