65 lines
1.6 KiB
JavaScript
65 lines
1.6 KiB
JavaScript
const fs = require("fs");
|
|
const path = require("path");
|
|
const redis = require("@redis/client");
|
|
let bench_path;
|
|
if (process.env.FRAPPE_BENCH_ROOT) {
|
|
bench_path = process.env.FRAPPE_BENCH_ROOT;
|
|
} else {
|
|
bench_path = path.resolve(__dirname, "..", "..");
|
|
}
|
|
|
|
const dns = require("dns");
|
|
|
|
// Since node17, node resolves to ipv6 unless system is configured otherwise.
|
|
// In Frappe context using ipv4 - 127.0.0.1 is fine.
|
|
dns.setDefaultResultOrder("ipv4first");
|
|
|
|
function get_conf() {
|
|
// defaults
|
|
var conf = {
|
|
socketio_port: 9000,
|
|
};
|
|
|
|
var read_config = function (file_path) {
|
|
const full_path = path.resolve(bench_path, file_path);
|
|
|
|
if (fs.existsSync(full_path)) {
|
|
var bench_config = JSON.parse(fs.readFileSync(full_path));
|
|
for (var key in bench_config) {
|
|
if (bench_config[key]) {
|
|
conf[key] = bench_config[key];
|
|
}
|
|
}
|
|
}
|
|
};
|
|
|
|
// get ports from bench/config.json
|
|
read_config("config.json");
|
|
read_config("sites/common_site_config.json");
|
|
|
|
// set overrides from environment
|
|
if (process.env.FRAPPE_SITE) {
|
|
conf.default_site = process.env.FRAPPE_SITE;
|
|
}
|
|
if (process.env.FRAPPE_REDIS_CACHE) {
|
|
conf.redis_cache = process.env.FRAPPE_REDIS_CACHE;
|
|
}
|
|
if (process.env.FRAPPE_REDIS_QUEUE) {
|
|
conf.redis_queue = process.env.FRAPPE_REDIS_QUEUE;
|
|
}
|
|
if (process.env.FRAPPE_SOCKETIO_PORT) {
|
|
conf.socketio_port = process.env.FRAPPE_SOCKETIO_PORT;
|
|
}
|
|
return conf;
|
|
}
|
|
|
|
function get_redis_subscriber(kind = "redis_queue", options = {}) {
|
|
const conf = get_conf();
|
|
const host = conf[kind];
|
|
return redis.createClient({ url: host, ...options });
|
|
}
|
|
|
|
module.exports = {
|
|
get_conf,
|
|
get_redis_subscriber,
|
|
};
|