FullTableScannerAdapterBase.hpp

#pragma once
 
#include "migration/cassandra/CassandraMigrationBackend.hpp"
#include "migration/cassandra/impl/FullTableScanner.hpp"
 
#include <boost/asio/spawn.hpp>
 
#include <memory>
#include <utility>
 
namespace migration::cassandra::impl {

template <TableSpec TableDesc>
struct FullTableScannerAdapterBase {
    static_assert(TableSpec<TableDesc>);
 
protected:
    std::shared_ptr<CassandraMigrationBackend> backend_;
 
public:
    virtual ~FullTableScannerAdapterBase() = default;

    FullTableScannerAdapterBase(std::shared_ptr<CassandraMigrationBackend> backend)
        : backend_(std::move(backend))
    {
    }

    void
    readByTokenRange(TokenRange const& range, boost::asio::yield_context yield)
    {
        backend_->migrateInTokenRange<TableDesc>(
            range.start, range.end, [this](auto const& row) { onRowRead(row); }, yield
        );
    }

    virtual void
    onRowRead(TableDesc::Row const& row) = 0;
};
}  // namespace migration::cassandra::impl