Files
pixiv/backend/config/cache-config.js
T

232 lines
6.4 KiB
JavaScript

const fs = require('fs').promises;
const path = require('path');
const { defaultLogger } = require('../utils/logger');
// 创建logger实例
const logger = defaultLogger.child('CacheConfigManager');
/**
* 缓存配置管理器
* 负责管理图片缓存的配置选项
*/
class CacheConfigManager {
constructor() {
// 检测是否在pkg打包环境中运行
const isPkg = process.pkg !== undefined;
if (isPkg) {
// 在打包环境中,使用可执行文件所在目录
this.configPath = path.join(process.cwd(), 'data', 'cache-config.json');
} else {
// 在开发环境中,使用项目根目录的data文件夹
this.configPath = path.join(__dirname, '..', '..', 'data', 'cache-config.json');
}
// 确保路径是绝对路径
this.configPath = path.resolve(this.configPath);
// 默认缓存配置
this.config = {
maxAge: 24 * 60 * 60 * 1000, // 24小时缓存
maxSize: 100 * 1024 * 1024, // 100MB最大缓存大小
cleanupInterval: 2 * 60 * 60 * 1000, // 2小时清理一次(原来是1小时)
enabled: true,
proxy: {
enabled: true,
timeout: 30000,
retryCount: 3,
retryDelay: 1000,
},
allowedExtensions: ['.jpg', '.jpeg', '.png', '.gif', '.webp', '.bmp'],
// 下载检测配置
download: {
concurrentDownloads: 3, // 同时下载任务数
maxConcurrentFiles: 5, // 单个任务内最大并发文件数
threadPoolSize: 16, // Node.js 线程池大小
downloadTimeout: 300000, // 5分钟下载超时
chunkSize: 1024 * 1024, // 1MB块大小
retryAttempts: 3, // 重试次数
retryDelay: 2000, // 重试延迟
// 下载检测配置
useRegistryCheck: true, // 是否使用注册表检测(默认启用)
fallbackToScan: false, // 检测失败时是否回退到扫盘检测
maxFileSize: 50 * 1024 * 1024, // 最大文件大小 50MB
// 存储模式配置
storageMode: 'json', // 存储模式:'json' 或 'database'
},
// 待看名单配置
watchlist: {
// 存储模式:'json' 或 'database'
storageMode: 'json'
},
// 新增Windows特定配置
windows: {
skipInUseFiles: true, // 跳过被占用的文件
maxRetries: 3, // 最大重试次数
retryDelay: 2000, // 重试延迟(毫秒)
waitForRelease: true, // 等待文件释放
maxWaitTime: 10000, // 最大等待时间(毫秒)
}
};
// 确保配置目录存在
this.ensureConfigDir();
}
/**
* 确保配置目录存在
*/
ensureConfigDir() {
try {
const configDir = path.dirname(this.configPath);
if (!require('fs').existsSync(configDir)) {
require('fs').mkdirSync(configDir, { recursive: true });
// logger.info('缓存配置目录创建成功:', configDir);
}
} catch (error) {
logger.error('创建缓存配置目录失败:', error);
}
}
/**
* 初始化配置文件
*/
async initialize() {
try {
// 检查配置文件是否存在
await fs.access(this.configPath);
// logger.info('缓存配置文件已存在');
} catch (error) {
// 配置文件不存在,创建默认配置
logger.info('创建默认缓存配置文件...');
await this.createDefaultConfig();
}
}
/**
* 创建默认配置文件
*/
async createDefaultConfig() {
try {
const configContent = JSON.stringify(this.config, null, 2);
await fs.writeFile(this.configPath, configContent, 'utf8');
// logger.info('默认缓存配置文件创建成功:', this.configPath);
} catch (error) {
logger.error('创建默认缓存配置文件失败:', error);
throw error;
}
}
/**
* 加载配置
*/
async loadConfig() {
try {
const configContent = await fs.readFile(this.configPath, 'utf8');
const config = JSON.parse(configContent);
// 合并默认配置,确保所有字段都存在
return { ...this.config, ...config };
} catch (error) {
// 检查错误类型,如果是文件不存在错误,则创建默认配置文件
if (error.code === 'ENOENT') {
// 文件不存在,创建默认配置
await this.createDefaultConfig();
return this.config;
} else {
// 其他错误,记录日志
logger.error('加载缓存配置失败:', error);
return this.config;
}
}
}
/**
* 保存配置
*/
async saveConfig(config) {
try {
// 更新最后修改时间
config.lastUpdated = new Date().toISOString();
const configContent = JSON.stringify(config, null, 2);
await fs.writeFile(this.configPath, configContent, 'utf8');
logger.info('缓存配置保存成功');
} catch (error) {
logger.error('保存缓存配置失败:', error);
throw error;
}
}
/**
* 更新配置
*/
async updateConfig(updates) {
try {
const currentConfig = await this.loadConfig();
const newConfig = { ...currentConfig, ...updates };
await this.saveConfig(newConfig);
return newConfig;
} catch (error) {
logger.error('更新缓存配置失败:', error);
throw error;
}
}
/**
* 重置为默认配置
*/
async resetToDefault() {
try {
await this.saveConfig(this.config);
logger.info('缓存配置已重置为默认值');
return this.config;
} catch (error) {
logger.error('重置缓存配置失败:', error);
throw error;
}
}
/**
* 验证配置
*/
validateConfig(config) {
const errors = [];
if (config.maxAge < 0) {
errors.push('maxAge 必须大于等于 0');
}
if (config.maxSize < 0) {
errors.push('maxSize 必须大于等于 0');
}
if (config.cleanupInterval < 0) {
errors.push('cleanupInterval 必须大于等于 0');
}
if (config.proxy.timeout < 0) {
errors.push('proxy.timeout 必须大于等于 0');
}
if (config.proxy.retryCount < 0) {
errors.push('proxy.retryCount 必须大于等于 0');
}
if (config.proxy.retryDelay < 0) {
errors.push('proxy.retryDelay 必须大于等于 0');
}
return errors;
}
/**
* 获取配置路径
*/
getConfigPath() {
return this.configPath;
}
}
module.exports = CacheConfigManager;