Convert to class

This commit is contained in:
Cohee
2025-03-20 10:21:41 +02:00
parent df4700914c
commit 3355c682ca
2 changed files with 62 additions and 51 deletions

View File

@@ -66,7 +66,7 @@ import { init as statsInit, onExit as statsOnExit } from './src/endpoints/stats.
import { checkForNewContent } from './src/endpoints/content-manager.js';
import { init as settingsInit } from './src/endpoints/settings.js';
import { redirectDeprecatedEndpoints, ServerStartup, setupPrivateEndpoints } from './src/server-startup.js';
import { verifyCharactersDiskCache } from './src/endpoints/characters.js';
import { diskCache } from './src/endpoints/characters.js';
// Unrestrict console logs display limit
util.inspect.defaultOptions.maxArrayLength = null;
@@ -269,7 +269,7 @@ async function preSetupTasks() {
const directories = await getUserDirectoriesList();
await checkForNewContent(directories);
await ensureThumbnailCache();
await verifyCharactersDiskCache(directories);
await diskCache.verify(directories);
cleanUploads();
migrateAccessLog();

View File

@@ -32,73 +32,75 @@ const isAndroid = process.platform === 'android';
const useShallowCharacters = !!getConfigValue('performance.lazyLoadCharacters', false, 'boolean');
const useDiskCache = !!getConfigValue('performance.useDiskCache', true, 'boolean');
const diskCache = {
class DiskCache {
/** @type {string} */
static DIRECTORY = 'characters';
/** @type {number} */
static REMOVAL_INTERVAL = 60000;
/** @type {import('node-persist').LocalStorage} */
#instance;
/** @type {NodeJS.Timeout} */
#removalInterval;
/**
* @type {import('node-persist').LocalStorage?}
* @private
* Queue for removal of cache entries.
* @type {Set<string>}
*/
_instance: null,
/**
* @type {NodeJS.Timeout?}
* @private
*/
_removalInterval: null,
removalQueue = new Set();
/**
* Processes the removal queue.
* @returns {Promise<void>}
* @private
*/
_removeCacheEntries: async function() {
async #removeCacheEntries() {
try {
if (!useDiskCache || this.removalQueue.size === 0) {
return;
}
const keys = await diskCache.instance().then(i => i.keys());
const keys = await this.instance().then(i => i.keys());
for (const item of this.removalQueue) {
const key = keys.find(k => k.startsWith(item));
if (key) {
await diskCache.instance().then(i => i.removeItem(key));
await this.instance().then(i => i.removeItem(key));
}
}
this.removalQueue.clear();
} catch (error) {
console.error('Error while removing cache entries:', error);
}
},
};
/**
* Gets the disk cache instance.
* @returns {Promise<import('node-persist').LocalStorage>}
*/
instance: async function() {
if (this._instance) {
return this._instance;
async instance() {
if (this.#instance) {
return this.#instance;
}
const cacheDir = path.join(globalThis.DATA_ROOT, '_cache', 'characters');
this._instance = storage.create({ dir: cacheDir, ttl: false });
await this._instance.init();
this._removalInterval = setInterval(this._removeCacheEntries.bind(this), 60000);
return this._instance;
},
/**
* Queue for removal of cache entries.
* @type {Set<string>}
*/
removalQueue: new Set(),
};
const cacheDir = path.join(globalThis.DATA_ROOT, '_cache', DiskCache.DIRECTORY);
this.#instance = storage.create({ dir: cacheDir, ttl: false });
await this.#instance.init();
this.#removalInterval = setInterval(this.#removeCacheEntries.bind(this), DiskCache.REMOVAL_INTERVAL);
return this.#instance;
}
/**
* Verifies disk cache size and prunes it if necessary.
* @param {import('../users.js').UserDirectoryList[]} directoriesList List of user directories
* @returns {Promise<void>}
*/
export async function verifyCharactersDiskCache(directoriesList) {
async verify(directoriesList) {
if (!useDiskCache) {
return;
}
const cache = await diskCache.instance();
const cache = await this.instance();
const validKeys = [];
for (const dir of directoriesList) {
const files = fs.readdirSync(dir.characters);
@@ -116,6 +118,15 @@ export async function verifyCharactersDiskCache(directoriesList) {
}
}
dispose() {
if (this.#removalInterval) {
clearInterval(this.#removalInterval);
}
}
}
export const diskCache = new DiskCache();
/**
* Reads the character card from the specified image file.
* @param {string} inputFile - Path to the image file