Upload 2 files

database.js

@@ -4,16 +4,11 @@ import fetch from 'node-fetch';
 import { createGunzip } from 'zlib';
 import { createInterface } from 'readline';
 
-// --- Константы ---
 const DB_FILE = './movies_search.db';
 const DATA_URL = 'https://huggingface.co/datasets/opex792/kinopoisk/resolve/main/consolidated/kinopoisk.jsonl.gz?download=true';
 
 let db;
 
-/**
- * Инициализирует соединение с базой данных и создает необходимые таблицы,
- * включая FTS5-таблицу для полнотекстового поиска.
- */
 export async function initializeDatabase() {
     db = await open({ filename: DB_FILE, driver: sqlite3.Database });
     console.log('Подключение к SQLite (Search API) установлено.');
@@ -21,7 +16,6 @@ export async function initializeDatabase() {
     await db.exec('PRAGMA journal_mode = WAL;');
     await db.exec('PRAGMA synchronous = NORMAL;');
     
-    // Таблица для хранения полных данных о фильмах
     await db.exec(`
         CREATE TABLE IF NOT EXISTS movies (
             id INTEGER PRIMARY KEY,
@@ -29,7 +23,6 @@ export async function initializeDatabase() {
         );
     `);
 
-    // Виртуальная FTS5 таблица для эффективного полнотекстового поиска
     await db.exec(`
         CREATE VIRTUAL TABLE IF NOT EXISTS movie_names USING fts5(
             movie_id UNINDEXED,
@@ -39,13 +32,11 @@ export async function initializeDatabase() {
     `);
 
     console.log('Таблицы `movies` и `movie_names` (FTS5) готовы к работе.');
+    
+    // ИСПРАВЛЕНИЕ: Возвращаем объект базы данных для использования в других модулях.
     return db;
 }
 
-/**
- * Полностью обновляет данные: скачивает, распаковывает и индексирует фильмы.
- * Работает в потоковом режиме, чтобы не потреблять много оперативной памяти.
- */
 export async function refreshData() {
     console.log('Начало полной загрузки и индексации данных...');
 
@@ -56,7 +47,6 @@ export async function refreshData() {
     response.body.pipe(gunzip);
     const rl = createInterface({ input: gunzip, crlfDelay: Infinity });
 
-    // Очищаем таблицы перед новой загрузкой
     await db.exec('BEGIN TRANSACTION;');
     await db.exec('DELETE FROM movies;');
     await db.exec('DELETE FROM movie_names;');
@@ -72,10 +62,8 @@ export async function refreshData() {
         const movie = JSON.parse(line);
         if (!movie.id) continue;
 
-        // 1. Сохраняем полный JSON объекта фильма
         await movieStmt.run(movie.id, JSON.stringify(movie));
 
-        // 2. Индексируем все возможные названия для поиска
         const names = new Set();
         if (movie.name) names.add(movie.name);
         if (movie.alternativeName) names.add(movie.alternativeName);
@@ -99,15 +87,7 @@ export async function refreshData() {
     console.log(`✅ Процесс завершен. Всего загружено и проиндексировано ${processedCount} фильмов.`);
 }
 
-/**
- * Выполняет полнотекстовый поиск по проиндексированным названиям с пагинацией.
- * @param {string} queryString - Поисковый запрос.
- * @param {number} page - Номер страницы.
- * @param {number} limit - Количество результатов на странице.
- * @returns {Promise<{docs: Array<object>, total: number}>} Объект с результатами и общим количеством.
- */
 export async function searchByName(queryString, page = 1, limit = 10) {
-    // Этап 1: Получаем общее количество уникальных фильмов, соответствующих запросу
     const countResult = await db.get(
         `SELECT count(DISTINCT movie_id) as total FROM movie_names WHERE movie_names MATCH ?`,
         [queryString]
@@ -118,7 +98,6 @@ export async function searchByName(queryString, page = 1, limit = 10) {
         return { docs: [], total: 0 };
     }
 
-    // Этап 2: Получаем ID фильмов для текущей страницы, отсортированные по релевантности (rank)
     const offset = (page - 1) * limit;
     const searchResults = await db.all(
         `SELECT DISTINCT movie_id FROM movie_names WHERE movie_names MATCH ? ORDER BY rank BM25(1.0, 5.0) LIMIT ? OFFSET ?`,
@@ -129,17 +108,14 @@ export async function searchByName(queryString, page = 1, limit = 10) {
         return { docs: [], total };
     }
 
-    // Этап 3: Извлекаем полные данные для найденных ID из нашей локальной таблицы `movies`
     const movieIds = searchResults.map(r => r.movie_id);
     const placeholders = movieIds.map(() => '?').join(',');
     const moviesData = await db.all(`SELECT data FROM movies WHERE id IN (${placeholders})`, movieIds);
 
-    // Восстанавливаем порядок, заданный поиском
     const movieMap = new Map(moviesData.map(m => {
         const movieJson = JSON.parse(m.data);
         return [movieJson.id, movieJson];
     }));
-
     const docs = movieIds.map(id => movieMap.get(id));
 
     return { docs, total };

server.js

@@ -5,18 +5,17 @@ import { initializeDatabase, refreshData, searchByName } from './database.js';
 
 // --- Константы ---
 const PORT = process.env.PORT || 7860;
-const API_VERSION = "1.4"; // Версия API из документации
+const API_VERSION = "1.4";
 
 // --- Настройка приложения Express ---
 const app = express();
 app.use(cors());
 app.use(express.json());
 
-let isReady = false; // Флаг готовности сервера к приему запросов
+let isReady = false;
 
 // --- Middleware для проверки готовности ---
 app.use((req, res, next) => {
-    // Пропускаем health-check, чтобы всегда можно было проверить состояние
     if (!isReady && req.path !== '/health') {
         return res.status(503).json({ message: 'Сервис временно недоступен, идет инициализация данных.', statusCode: 503 });
     }
@@ -28,7 +27,6 @@ app.get('/health', (req, res) => {
     res.status(200).json({ status: isReady ? 'UP' : 'DOWN', message: isReady ? 'Сервис готов к работе' : 'Идет инициализация' });
 });
 
-// Эндпоинт для поиска, полностью совместимый с документацией
 app.get(`/v${API_VERSION}/movie/search`, async (req, res) => {
     const { query, page, limit } = req.query;
 
@@ -37,15 +35,12 @@ app.get(`/v${API_VERSION}/movie/search`, async (req, res) => {
     }
 
     const pageNum = parseInt(page, 10) || 1;
-    const limitNum = Math.min(parseInt(limit, 10) || 10, 250); // Лимит от 1 до 250
+    const limitNum = Math.min(parseInt(limit, 10) || 10, 250);
 
     try {
         const { docs, total } = await searchByName(query, pageNum, limitNum);
         const pages = Math.ceil(total / limitNum);
-
-        // Формируем ответ в строгом соответствии со спецификацией
         res.json({ docs, total, limit: limitNum, page: pageNum, pages });
-
     } catch (e) {
         console.error(`Ошибка при поиске по запросу "${query}":`, e);
         res.status(500).json({ message: 'Внутренняя ошибка сервера при поиске.', error: e.message });
@@ -55,16 +50,21 @@ app.get(`/v${API_VERSION}/movie/search`, async (req, res) => {
 // --- Логика запуска сервера ---
 (async () => {
     try {
-        await initializeDatabase();
+        // ИСПРАВЛЕНИЕ: Получаем объект 'db' после инициализации.
+        const db = await initializeDatabase();
         
         const needsRebuild = process.argv.includes('--rebuild-index');
-        const count = await db.get('SELECT COUNT(*) as count FROM movies');
+        
+        // ИСПРАВЛЕНИЕ: Используем полученный 'db' для запроса.
+        const result = await db.get('SELECT COUNT(*) as count FROM movies');
+        const moviesCount = result.count;
 
-        // Запускаем полную перестройку индекса, если передан флаг или база пуста
-        if (needsRebuild || count.count === 0) {
+        if (needsRebuild || moviesCount === 0) {
             if (needsRebuild) console.log('Обнаружен флаг --rebuild-index. Запуск принудительной переиндексации.');
-            if (count.count === 0) console.log('База данных пуста. Запуск первоначальной загрузки.');
+            if (moviesCount === 0) console.log('База данных пуста. Запуск первоначальной загрузки.');
             await refreshData();
+        } else {
+            console.log(`В базе уже есть ${moviesCount} фильмов. Запуск в штатном режиме.`);
         }
 
         isReady = true;
@@ -75,7 +75,6 @@ app.get(`/v${API_VERSION}/movie/search`, async (req, res) => {
             console.log(`URL для поиска: http://localhost:${PORT}/v${API_VERSION}/movie/search`);
         });
 
-        // Планируем автоматическое обновление данных каждый день в 3:00 ночи
         cron.schedule('0 3 * * *', async () => {
             console.log('Начало планового ежедневного обновления данных...');
             isReady = false;