blitzwing/cobalt-7
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

api/cluster: implement broadcast helper

Browse source
This commit is contained in:
jj 2024-11-01 14:49:52 +00:00
parent 2351cf74f4
commit 1c78dac7ed
No known key found for this signature in database
2 changed files with 16 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
api/src/misc/cluster.js
View file

@ -1,7 +1,9 @@
import cluster from "node:cluster";
import net from "node:net";
import { syncSecrets } from "../security/secrets.js";
import { env } from "../config.js";
import { env, isCluster } from "../config.js";
export { isPrimary, isWorker } from "node:cluster";
export const supportsReusePort = async () => {
try {
@ -26,3 +28,13 @@ export const initCluster = async () => {
await syncSecrets();
}
export const broadcast = (message) => {
if (!isCluster || !cluster.isPrimary || !cluster.workers) {
return;
}
for (const worker of Object.values(cluster.workers)) {
worker.send(message);
}
}

10
api/src/security/api-keys.js
View file

@ -1,8 +1,8 @@
import { env, isCluster } from "../config.js";
import { env } from "../config.js";
import { readFile } from "node:fs/promises";
import { Green, Yellow } from "../misc/console-text.js";
import cluster from "node:cluster";
import ip from "ipaddr.js";
import * as cluster from "../misc/cluster.js";
// this function is a modified variation of code
// from https://stackoverflow.com/a/32402438/14855621
@ -133,11 +133,7 @@ const loadKeys = async (source) => {
validateKeys(updated);
if (isCluster && cluster.workers) {
for (const worker of Object.values(cluster.workers)) {
worker.send({ api_keys: updated });
}
}
cluster.broadcast({ api_keys: updated });
updateKeys(updated);
}