94 lines
2.6 KiB
JavaScript
94 lines
2.6 KiB
JavaScript
// ═══════════════════════════════════════════════════════════════════════════
|
||
// MiniSearch Index Manager
|
||
// 全文搜索索引管理
|
||
// ═══════════════════════════════════════════════════════════════════════════
|
||
|
||
import MiniSearch from '../../../libs/minisearch.mjs';
|
||
|
||
// 索引缓存：chatId -> { index, updatedAt }
|
||
const indexCache = new Map();
|
||
|
||
/**
|
||
* 获取或创建搜索索引
|
||
* @param {string} chatId
|
||
* @param {Array} events - L2 事件
|
||
* @param {number} storeUpdatedAt - store.updatedAt 时间戳
|
||
* @returns {MiniSearch}
|
||
*/
|
||
export function getSearchIndex(chatId, events, storeUpdatedAt) {
|
||
const cached = indexCache.get(chatId);
|
||
|
||
// 缓存有效
|
||
if (cached && cached.updatedAt >= storeUpdatedAt) {
|
||
return cached.index;
|
||
}
|
||
|
||
// 重建索引
|
||
const index = new MiniSearch({
|
||
fields: ['title', 'summary', 'participants'],
|
||
storeFields: ['id'],
|
||
searchOptions: {
|
||
boost: { title: 2, participants: 1.5, summary: 1 },
|
||
fuzzy: 0.2,
|
||
prefix: true,
|
||
},
|
||
});
|
||
|
||
// 索引事件
|
||
const docs = events.map(e => ({
|
||
id: e.id,
|
||
title: e.title || '',
|
||
summary: e.summary || '',
|
||
participants: (e.participants || []).join(' '),
|
||
}));
|
||
|
||
index.addAll(docs);
|
||
|
||
// 缓存
|
||
indexCache.set(chatId, { index, updatedAt: storeUpdatedAt });
|
||
|
||
// 限制缓存数量
|
||
if (indexCache.size > 5) {
|
||
const firstKey = indexCache.keys().next().value;
|
||
indexCache.delete(firstKey);
|
||
}
|
||
|
||
return index;
|
||
}
|
||
|
||
/**
|
||
* 关键词搜索
|
||
* @returns {Map<string, number>} - eventId -> 归一化分数
|
||
*/
|
||
export function searchByKeywords(index, queryText, limit = 20) {
|
||
if (!queryText?.trim()) return new Map();
|
||
|
||
const results = index.search(queryText, { limit });
|
||
|
||
if (results.length === 0) return new Map();
|
||
|
||
// 归一化分数到 0-1
|
||
const maxScore = results[0].score;
|
||
const scores = new Map();
|
||
|
||
for (const r of results) {
|
||
scores.set(r.id, r.score / maxScore);
|
||
}
|
||
|
||
return scores;
|
||
}
|
||
|
||
/**
|
||
* 清除指定聊天的索引缓存
|
||
*/
|
||
export function invalidateIndex(chatId) {
|
||
indexCache.delete(chatId);
|
||
}
|
||
|
||
/**
|
||
* 清除所有索引缓存
|
||
*/
|
||
export function clearAllIndexes() {
|
||
indexCache.clear();
|
||
}
|