DatabaseBootstrapPlugin.h

Go to the documentation of this file.

#ifndef LINTEL_DATABASEBOOTSTRAPPLUGIN_H
#define LINTEL_DATABASEBOOTSTRAPPLUGIN_H
 
#include <cstdint>
#include <filesystem>
#include <optional>
 
#include "lintel/config.h"
#include "lintel/core/persistence/DatabaseConnectionConfigurations.h"
#include "lintel/core/persistence/Result.h"
#include "lintel/core/plugins/BootstrapPlugin.h"
 
namespace lintel {
class DatabaseBootstrapPlugin : public BootstrapPlugin
{
private:
  struct FileInformation
  {
    std::string m_schemaName;
    int32_t m_version;
    std::filesystem::path m_filePath;
    bool m_isInitFile;
  };
  std::filesystem::path m_configPath = std::filesystem::path(CONFIG_DIRECTORY) / "database";
  std::shared_ptr<DatabaseConnectionConfigurations> m_connectionConfigurations;
 
  static constexpr std::string_view m_createSchemaVersion =
    R"(CREATE TABLE schema_version (name TEXT NOT NULL PRIMARY KEY,version BIGINT NOT NULL);)";
 
  static std::optional<db::Result> hasSchemaVersionTable(
    const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry);
 
  static void createSchemaVersionTable(const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry);
 
  static void initDatabase(const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry,
    const std::map<std::string, std::vector<FileInformation>> &initFiles,
    const std::map<std::string, int32_t> &schemaVersions);
 
  static void updateSchemaVersion(const db::Connection &connection,
    const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry,
    const std::string &schemaName,
    int32_t schemaVersion,
    bool insert = true);
 
  static void executeInitFile(db::Connection &connection,
    const FileInformation &item,
    const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry,
    int32_t &currentSchemaVersion);
 
  static std::string schemaNameOf(const std::string &fileName);
 
  static int32_t versionOf(const std::string &fileName);
 
  static void registerReadinessCheck(const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry);
 
  static void registerConnectionPool(const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry);
 
  void handle(const std::shared_ptr<DatabaseConnectionEntry> &connectionEntry);
 
public:
  explicit DatabaseBootstrapPlugin(std::shared_ptr<DatabaseConnectionConfigurations> connectionConfigurations);
 
  void onStart() override;
 
  BootstrapSequence getPriority() override;
};
 
}// namespace lintel
#endif// LINTEL_DATABASEBOOTSTRAPPLUGIN_H