shuzhiren-comfyui/任务队列后端/config/Config.js

31 lines
938 B
JavaScript

import fs from 'fs';
import path from 'path';
import { fileURLToPath } from 'url';
import { dirname } from 'path';
const __filename = fileURLToPath(import.meta.url);
const __dirname = dirname(__filename);
const modelPath = path.join(__dirname, 'model.json');
const modelData = JSON.parse(fs.readFileSync(modelPath, 'utf8'));
const platformPath = path.join(__dirname, 'Platform.json');
const platformData = JSON.parse(fs.readFileSync(platformPath, 'utf8'));
const CostPath = path.join(__dirname, 'cost.json');
const CostData = JSON.parse(fs.readFileSync(CostPath, 'utf8'));
function getExternalCapacityMax() {
let total = 0;
for (const modelObj of Object.values(modelData)) {
for (const [platformName, info] of Object.entries(modelObj)) {
if (platformName === 'comfyui') {
total += info.concurrency || 0;
}
}
}
return total;
}
export { modelData, platformData, CostData, getExternalCapacityMax };