CassandraMigrationSchema.hpp

#pragma once
 
#include "data/cassandra/Handle.hpp"
#include "data/cassandra/Schema.hpp"
#include "data/cassandra/SettingsProvider.hpp"
#include "data/cassandra/Types.hpp"
 
#include <fmt/format.h>
 
#include <functional>
#include <string>
 
namespace migration::cassandra::impl {

class CassandraMigrationSchema {
    using SettingsProviderType = data::cassandra::SettingsProvider;
    std::reference_wrapper<SettingsProviderType const> settingsProvider_;
 
public:
    explicit CassandraMigrationSchema(SettingsProviderType const& settings)
        : settingsProvider_{settings}
    {
    }

    data::cassandra::PreparedStatement
    getPreparedFullScanStatement(
        data::cassandra::Handle const& handler,
        std::string const& tableName,
        std::string const& key
    )
    {
        return handler.prepare(
            fmt::format(
                R"(
            SELECT *
              FROM {}
             WHERE TOKEN({}) >= ? AND TOKEN({}) <= ?
            )",
                data::cassandra::qualifiedTableName<SettingsProviderType>(
                    settingsProvider_.get(), tableName
                ),
                key,
                key
            )
        );
    }

     * @param handler The database handler
     * @return The prepared statement to insert into migrator_status table
     */
    data::cassandra::PreparedStatement const&
    getPreparedInsertMigratedMigrator(data::cassandra::Handle const& handler)
    {
        static auto const kPrepared = handler.prepare(
            fmt::format(
                R"(
            INSERT INTO {}
                   (migrator_name, status)
            VALUES (?, ?)
            )",
                data::cassandra::qualifiedTableName<SettingsProviderType>(
                    settingsProvider_.get(), "migrator_status"
                )
            )
        );
        return kPrepared;
    }
};
}  // namespace migration::cassandra::impl