mirror of
https://github.com/Wx-2025/ST-Amily2-Chat-Optimisation.git
synced 2026-06-06 13:55:51 +00:00
104 lines
3.9 KiB
JavaScript
104 lines
3.9 KiB
JavaScript
/**
|
||
* LoreService — 世界书操作统一服务层
|
||
*
|
||
* 职责：
|
||
* 1. 写锁（Promise chain 串行化，防止多模块并发覆盖同一世界书）
|
||
* 2. ST world-info.js API 的统一门面（减少各模块直接依赖 ST 内部函数）
|
||
* 3. Phase 2.3 将注册为 Amily2Bus 服务，届时外部模块改为 query('LoreService')
|
||
*
|
||
* 当前消费方：
|
||
* - core/super-memory/lorebook-bridge.js → ensureBook()
|
||
* - core/historiographer.js → saveBook()
|
||
* - core/lore.js → （Phase 2.3 后迁入）
|
||
*/
|
||
|
||
import {
|
||
loadWorldInfo,
|
||
createNewWorldInfo,
|
||
saveWorldInfo,
|
||
} from '/scripts/world-info.js';
|
||
|
||
// ── 写锁实现 ─────────────────────────────────────────────────────────────────
|
||
//
|
||
// 所有写操作排入同一个 Promise chain，保证串行执行。
|
||
// 读操作无锁，并发安全。
|
||
|
||
let _writeLock = Promise.resolve();
|
||
|
||
/**
|
||
* 在写锁保护下执行 fn，所有世界书写操作应通过此函数。
|
||
* @template T
|
||
* @param {string} label - 操作标识，用于日志定位
|
||
* @param {() => Promise<T>} fn
|
||
* @returns {Promise<T>}
|
||
*/
|
||
export function withLoreLock(label, fn) {
|
||
const result = _writeLock.then(() => {
|
||
console.log(`[LoreService] 写锁获取: ${label}`);
|
||
return fn();
|
||
});
|
||
// 出错时不阻断后续排队操作，但让错误传播给调用方
|
||
_writeLock = result.then(
|
||
() => { console.log(`[LoreService] 写锁释放: ${label}`); },
|
||
() => { console.warn(`[LoreService] 写锁释放（含错误）: ${label}`); },
|
||
);
|
||
return result;
|
||
}
|
||
|
||
// ── 读操作（无锁）────────────────────────────────────────────────────────────
|
||
|
||
/**
|
||
* 加载世界书数据（只读，不加锁）。
|
||
* @param {string} bookName
|
||
* @returns {Promise<object|null>}
|
||
*/
|
||
export async function loadBook(bookName) {
|
||
return loadWorldInfo(bookName);
|
||
}
|
||
|
||
// ── 写操作（全部走写锁）──────────────────────────────────────────────────────
|
||
|
||
/**
|
||
* 确保世界书存在，不存在则创建。防止并发双重创建。
|
||
* @param {string} bookName
|
||
* @returns {Promise<object>} 世界书数据
|
||
*/
|
||
export async function ensureBook(bookName) {
|
||
return withLoreLock(`ensureBook(${bookName})`, async () => {
|
||
const existing = await loadWorldInfo(bookName);
|
||
if (existing) return existing;
|
||
console.log(`[LoreService] 世界书不存在，正在创建: ${bookName}`);
|
||
return createNewWorldInfo(bookName);
|
||
});
|
||
}
|
||
|
||
/**
|
||
* 保存世界书数据。
|
||
* @param {string} bookName
|
||
* @param {object} bookData
|
||
* @param {boolean} [silent=true]
|
||
* @returns {Promise<void>}
|
||
*/
|
||
export async function saveBook(bookName, bookData, silent = true) {
|
||
return withLoreLock(`saveBook(${bookName})`, () =>
|
||
saveWorldInfo(bookName, bookData, silent)
|
||
);
|
||
}
|
||
|
||
// ── Bus 注册 ──────────────────────────────────────────────────────────────────
|
||
// Bus 注册名：'LoreService'
|
||
// 公开接口：withLoreLock, loadBook, ensureBook, saveBook
|
||
setTimeout(() => {
|
||
try {
|
||
const _ctx = window.Amily2Bus?.register('LoreService');
|
||
if (!_ctx) {
|
||
console.warn('[LoreService] Amily2Bus 尚未就绪，服务注册跳过。');
|
||
return;
|
||
}
|
||
_ctx.expose({ withLoreLock, loadBook, ensureBook, saveBook });
|
||
_ctx.log('LoreService', 'info', 'LoreService 已注册到 Bus。');
|
||
} catch (e) {
|
||
console.error('[LoreService] Bus 注册失败:', e);
|
||
}
|
||
}, 0);
|