142 lines
4.1 KiB
JavaScript
142 lines
4.1 KiB
JavaScript
//const _ = require('lodash');
|
|
|
|
const utils = require('./utils');
|
|
|
|
class DbSearcher {
|
|
constructor(db) {
|
|
this.db = db;
|
|
}
|
|
|
|
async selectAuthorIds(query) {
|
|
const db = this.db;
|
|
|
|
let authorRows;
|
|
let authorIds = new Set();
|
|
//сначала выберем все id авторов по фильтру
|
|
//порядок id соответсвует ASC-сортировке по author
|
|
if (query.author) {
|
|
authorRows = await db.select({
|
|
table: 'author',
|
|
map: `(r) => ({id: r.id})`,
|
|
});
|
|
|
|
for (const row of authorRows)
|
|
authorIds.add(row.id);
|
|
} else {//все авторы
|
|
if (!db.searchCache.authorIdsAll) {
|
|
authorRows = await db.select({
|
|
table: 'author',
|
|
map: `(r) => ({id: r.id})`,
|
|
});
|
|
|
|
db.searchCache.authorIdsAll = [];
|
|
for (const row of authorRows) {
|
|
authorIds.add(row.id);
|
|
db.searchCache.authorIdsAll.push(row.id);
|
|
}
|
|
} else {//оптимизация
|
|
for (const id of db.searchCache.authorIdsAll) {
|
|
authorIds.add(id);
|
|
}
|
|
}
|
|
}
|
|
|
|
const idsArr = [];
|
|
idsArr.push(authorIds);
|
|
|
|
//серии
|
|
//названия
|
|
//жанры
|
|
//языки
|
|
|
|
if (idsArr.length > 1)
|
|
authorIds = utils.intersectSet(idsArr);
|
|
|
|
//сортировка
|
|
authorIds = Array.from(authorIds);
|
|
|
|
authorIds.sort((a, b) => a > b);
|
|
|
|
return authorIds;
|
|
}
|
|
|
|
async getAuthorIds(query) {
|
|
const db = this.db;
|
|
|
|
if (!db.searchCache)
|
|
db.searchCache = {};
|
|
|
|
let result;
|
|
|
|
//сначала попробуем найти в кеше
|
|
const q = query;
|
|
const keyArr = [q.author, q.series, q.title, q.genre, q.lang];
|
|
const keyStr = keyArr.join('');
|
|
|
|
if (!keyStr) {//пустой запрос
|
|
if (db.searchCache.authorIdsAll)
|
|
result = db.searchCache.authorIdsAll;
|
|
else
|
|
result = await this.selectAuthorIds(query);
|
|
|
|
} else {//непустой запрос
|
|
const key = JSON.stringify(keyArr);
|
|
|
|
const rows = await db.select({table: 'query_cache', where: `@@id(${db.esc(key)})`});
|
|
if (rows.length) {//нашли в кеше
|
|
await db.insert({
|
|
table: 'query_time',
|
|
replace: true,
|
|
rows: [{id: key, time: Date.now()}],
|
|
});
|
|
|
|
result = rows[0].value;
|
|
} else {//не нашли в кеше, ищем в поисковых таблицах
|
|
result = await this.selectAuthorIds(query);
|
|
|
|
await db.insert({
|
|
table: 'query_cache',
|
|
replace: true,
|
|
rows: [{id: key, value: result}],
|
|
});
|
|
await db.insert({
|
|
table: 'query_time',
|
|
replace: true,
|
|
rows: [{id: key, time: Date.now()}],
|
|
});
|
|
}
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
async search(query) {
|
|
const db = this.db;
|
|
|
|
const authorIds = await this.getAuthorIds(query);
|
|
|
|
const totalFound = authorIds.length;
|
|
const limit = (query.limit ? query.limit : 1000);
|
|
|
|
//выборка найденных авторов
|
|
let result = await db.select({
|
|
table: 'author',
|
|
map: `(r) => ({id: r.id, author: r.author})`,
|
|
where: `
|
|
const all = @all();
|
|
const ids = new Set();
|
|
let n = 0;
|
|
for (const id of all) {
|
|
if (++n > ${db.esc(limit)})
|
|
break;
|
|
ids.add(id);
|
|
}
|
|
return ids;
|
|
`
|
|
});
|
|
|
|
return {result, totalFound};
|
|
}
|
|
}
|
|
|
|
module.exports = DbSearcher; |