yume/paricafe
1
0
Fork 0
mirror of https://github.com/paricafe/misskey.git synced 2025-02-17 14:47:29 -06:00
Code Issues Projects Releases Packages Wiki Activity

Reapply "update SearchService"

Browse source 
This reverts commit 6c9dd99083.
This commit is contained in:
fly_mc 2024-11-18 02:43:08 +08:00
parent 1471039199
commit cb9385e229
1 changed files with 51 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

27
packages/backend/src/core/SearchService.ts
View file

@ -4,7 +4,7 @@
*/
import { Inject, Injectable } from '@nestjs/common';
import { In } from 'typeorm';
import { In, SelectQueryBuilder } from 'typeorm';
import { DI } from '@/di-symbols.js';
import type { Config } from '@/config.js';
import { bindThis } from '@/decorators.js';
@ -206,6 +206,7 @@ export class SearchService {
});
return notes.sort((a, b) => a.id > b.id ? -1 : 1);
} else {
const searchByPgroonga = async (makeQuery: (query: SelectQueryBuilder<MiNote>) => void) => {
const query = this.queryService.makePaginationQuery(this.notesRepository.createQueryBuilder('note'), pagination.sinceId, pagination.untilId);
if (opts.userId) {
@ -221,6 +222,7 @@ export class SearchService {
.leftJoinAndSelect('note.renote', 'renote')
.leftJoinAndSelect('reply.user', 'replyUser')
.leftJoinAndSelect('renote.user', 'renoteUser');
makeQuery(query);
if (opts.host) {
if (opts.host === '.') {
@ -235,6 +237,29 @@ export class SearchService {
if (me) this.queryService.generateBlockedUserQuery(query, me);
return await query.limit(pagination.limit).getMany();
};
const searchWord = sqlLikeEscape(q);
const notes = [
...new Map(
(
await Promise.all([
searchByPgroonga((query) => {
query.andWhere('note.text &@~ :q', { q: searchWord });
}),
searchByPgroonga((query) => {
query.andWhere('note.cw &@~ :q', { q: searchWord });
}),
])
)
.flatMap((e) => e)
.map((note) => [note.id, note]),
).values(),
]
.sort((lhs, rhs) => lhs.id < rhs.id ? 1 : -1)
.slice(0, pagination.limit);
return notes;
}
}
}