toeverything/AFFiNE
1
1
Fork 0
mirror of https://github.com/toeverything/AFFiNE.git synced 2024-12-02 14:33:54 +03:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): better search result (#7787)

Browse Source
This commit is contained in:
EYHN 2024-08-08 08:56:54 +00:00
parent c7aabd3a8d
commit 6ef5675be1
No known key found for this signature in database
GPG Key ID: 46C9E26A75AB276C
1 changed files with 14 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
packages/common/infra/src/sync/indexer/impl/indexeddb/inverted-index.ts
View File

@ -179,8 +179,9 @@ export class FullTextInvertedIndex implements InvertedIndex {
const matched = new Map<
number,
{
score: number[];
positions: Map<number, [number, number][]>;
score: number;
index: number;
ranges: [number, number][];
}
>();
for (const token of queryTokens) {
@ -250,27 +251,20 @@ export class FullTextInvertedIndex implements InvertedIndex {
maxScore === minScore
? score
: (score - minScore) / (maxScore - minScore);
const match = matched.get(nid) || {
score: [] as number[],
positions: new Map(),
};
match.score.push(normalizedScore);
const ranges = match.positions.get(position.index) || [];
ranges.push(...position.ranges);
match.positions.set(position.index, ranges);
matched.set(nid, match);
const match = matched.get(nid);
if (!match || normalizedScore > match.score) {
matched.set(nid, {
score: normalizedScore,
index: position.index,
ranges: position.ranges,
});
}
}
}
const match = new Match();
for (const [nid, { score, positions }] of matched) {
match.addScore(
nid,
score.reduce((acc, s) => acc + s, 0)
);
for (const [index, ranges] of positions) {
match.addHighlighter(nid, this.fieldKey, index, ranges);
}
for (const [nid, { score, index, ranges }] of matched) {
match.addScore(nid, score);
match.addHighlighter(nid, this.fieldKey, index, ranges);
}
return match;
}