refactor: Improvment rag storage

context/RAGStorage.hpp

@@ -17,50 +17,169 @@
  * along with QodeAssist. If not, see <https://www.gnu.org/licenses/>.
  */
 
+// RAGStorage.hpp
 #pragma once
 
+#include <optional>
 #include <QDateTime>
+#include <QMutex>
 #include <QObject>
 #include <QSqlDatabase>
 #include <QString>
+#include <qsqlquery.h>
 
 #include <RAGData.hpp>
 
 namespace QodeAssist::Context {
 
+/**
+ * @brief Структура для хранения информации о чанке файла
+ */
+struct FileChunkData
+{
+    QString filePath;    ///< Путь к файлу
+    int startLine;       ///< Начальная строка чанка
+    int endLine;         ///< Конечная строка чанка
+    QString content;     ///< Содержимое чанка
+    QDateTime createdAt; ///< Время создания
+    QDateTime updatedAt; ///< Время последнего обновления
+
+    bool isValid() const
+    {
+        return !filePath.isEmpty() && startLine >= 0 && endLine >= startLine && !content.isEmpty();
+    }
+};
+
+/**
+ * @brief Структура для настройки хранилища
+ */
+struct StorageOptions
+{
+    int maxChunkSize = 1024 * 1024; ///< Максимальный размер чанка в байтах
+    int maxVectorSize = 1024;       ///< Максимальный размер вектора
+    bool useCompression = false;    ///< Использовать сжатие данных
+    bool enableLogging = false;     ///< Включить журналирование
+};
+
+/**
+ * @brief Структура для хранения статистики
+ */
+struct StorageStatistics
+{
+    int totalChunks;      ///< Общее количество чанков
+    int totalVectors;     ///< Общее количество векторов
+    int totalFiles;       ///< Общее количество файлов
+    qint64 totalSize;     ///< Общий размер данных
+    QDateTime lastUpdate; ///< Время последнего обновления
+};
+
+/**
+ * @brief Класс для работы с хранилищем RAG данных
+ */
 class RAGStorage : public QObject
 {
     Q_OBJECT
+
 public:
     static constexpr int CURRENT_VERSION = 1;
 
-    explicit RAGStorage(const QString &dbPath, QObject *parent = nullptr);
+    enum class Status { Ok, DatabaseError, ValidationError, VersionError, ConnectionError };
+
+    struct Error
+    {
+        QString message;
+        QString sqlError;
+        QString query;
+        Status status;
+    };
+
+    explicit RAGStorage(
+        const QString &dbPath,
+        const StorageOptions &options = StorageOptions(),
+        QObject *parent = nullptr);
     ~RAGStorage();
 
+    // Инициализация и проверка состояния
     bool init();
+    Status status() const;
+    Error lastError() const;
+    bool isReady() const;
+    QString dbPath() const;
+
+    // Управление транзакциями
+    bool beginTransaction();
+    bool commitTransaction();
+    bool rollbackTransaction();
+
+    // Операции с векторами
     bool storeVector(const QString &filePath, const RAGVector &vector);
     bool updateVector(const QString &filePath, const RAGVector &vector);
     std::optional<RAGVector> getVector(const QString &filePath);
     bool needsUpdate(const QString &filePath);
     QStringList getAllFiles();
-    QString dbPath() const;
 
-    // Новые методы для работы с версией
+    // Операции с чанками
+    bool storeChunk(const FileChunkData &chunk);
+    bool storeChunks(const QList<FileChunkData> &chunks);
+    bool updateChunk(const FileChunkData &chunk);
+    bool updateChunks(const QList<FileChunkData> &chunks);
+    bool deleteChunksForFile(const QString &filePath);
+    std::optional<FileChunkData> getChunk(const QString &filePath, int startLine, int endLine);
+    QList<FileChunkData> getChunksForFile(const QString &filePath);
+    bool chunkExists(const QString &filePath, int startLine, int endLine);
+
+    // Обслуживание
+    int getChunkCount(const QString &filePath);
+    bool deleteOldChunks(const QString &filePath, const QDateTime &olderThan);
+    bool deleteAllChunks();
+    QStringList getFilesWithChunks();
+    bool vacuum();
+    bool backup(const QString &backupPath);
+    bool restore(const QString &backupPath);
+    StorageStatistics getStatistics() const;
+
+    // Версионирование
     int getStorageVersion() const;
-    bool isVersionCompatible() const { return getStorageVersion() == CURRENT_VERSION; }
+    bool isVersionCompatible() const;
+
+signals:
+    void errorOccurred(const Error &error);
+    void operationCompleted(const QString &operation);
 
 private:
     bool createTables();
+    bool createIndices();
     bool createVersionTable();
+    bool createChunksTable();
+    bool createVectorsTable();
     bool openDatabase();
     bool initializeNewStorage();
     bool upgradeStorage(int fromVersion);
+    bool validateSchema() const;
+
     QDateTime getFileLastModified(const QString &filePath);
     RAGVector blobToVector(const QByteArray &blob);
     QByteArray vectorToBlob(const RAGVector &vector);
 
+    void setError(const QString &message, Status status = Status::DatabaseError);
+    void clearError();
+    bool prepareStatements();
+    bool validateChunk(const FileChunkData &chunk) const;
+    bool validateVector(const RAGVector &vector) const;
+
+private:
     QSqlDatabase m_db;
     QString m_dbPath;
+    StorageOptions m_options;
+    mutable QMutex m_mutex;
+    Error m_lastError;
+    Status m_status;
+
+    // Подготовленные запросы
+    QSqlQuery m_insertChunkQuery;
+    QSqlQuery m_updateChunkQuery;
+    QSqlQuery m_insertVectorQuery;
+    QSqlQuery m_updateVectorQuery;
 };
 
 } // namespace QodeAssist::Context