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

api/api-keys: load keys once per cluster

Browse source
This commit is contained in:
jj 2024-11-01 13:57:53 +00:00
parent 3f505f6520
commit 48883486fa
No known key found for this signature in database
1 changed files with 25 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -1,6 +1,7 @@
import { env } from "../config.js"; import { env, isCluster } from "../config.js";
import { readFile } from "node:fs/promises"; import { readFile } from "node:fs/promises";
import { Green, Yellow } from "../misc/console-text.js"; import { Green, Yellow } from "../misc/console-text.js";
import cluster from "node:cluster";
import ip from "ipaddr.js"; import ip from "ipaddr.js";
// this function is a modified variation of code // this function is a modified variation of code
@ -131,7 +132,18 @@ const loadKeys = async (source) => {
} }
validateKeys(updated); validateKeys(updated);
keys = formatKeys(updated);
if (isCluster && cluster.workers) {
for (const worker of Object.values(cluster.workers)) {
worker.send({ api_keys: updated });
}
}
updateKeys(updated);
}
const updateKeys = (newKeys) => {
keys = formatKeys(newKeys);
} }
const wrapLoad = (url, initial = false) => { const wrapLoad = (url, initial = false) => {
@ -204,8 +216,16 @@ export const validateAuthorization = (req) => {
} }
export const setup = (url) => { export const setup = (url) => {
if (cluster.isPrimary) {
wrapLoad(url, true); wrapLoad(url, true);
if (env.keyReloadInterval > 0) { if (env.keyReloadInterval > 0) {
setInterval(() => wrapLoad(url), env.keyReloadInterval * 1000); setInterval(() => wrapLoad(url), env.keyReloadInterval * 1000);
} }
} else if (cluster.isWorker) {
process.on('message', (message) => {
if ('api_keys' in message) {
updateKeys(message.api_keys);
}
});
}
} }