mirror of
https://github.com/XRPLF/clio.git
synced 2025-11-20 19:56:00 +00:00
1cfa06c9aa
Support AWS Keyspace queries --------- Co-authored-by: Ayaz Salikhov <mathbunnyru@users.noreply.github.com> Co-authored-by: Alex Kremer <akremer@ripple.com>
141 lines
4.4 KiB
C++
141 lines
4.4 KiB
C++
//------------------------------------------------------------------------------
|
|
/*
|
|
This file is part of clio: https://github.com/XRPLF/clio
|
|
Copyright (c) 2025, the clio developers.
|
|
|
|
Permission to use, copy, modify, and distribute this software for any
|
|
purpose with or without fee is hereby granted, provided that the above
|
|
copyright notice and this permission notice appear in all copies.
|
|
|
|
THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
//==============================================================================
|
|
|
|
#pragma once
|
|
|
|
#include "data/cassandra/Concepts.hpp"
|
|
#include "data/cassandra/Handle.hpp"
|
|
#include "data/cassandra/Schema.hpp"
|
|
#include "data/cassandra/SettingsProvider.hpp"
|
|
#include "data/cassandra/Types.hpp"
|
|
#include "util/log/Logger.hpp"
|
|
|
|
#include <boost/json/string.hpp>
|
|
#include <fmt/compile.h>
|
|
|
|
#include <functional>
|
|
#include <memory>
|
|
|
|
namespace data::cassandra {
|
|
|
|
/**
|
|
* @brief Manages the DB schema and provides access to prepared statements.
|
|
*/
|
|
template <SomeSettingsProvider SettingsProviderType>
|
|
class KeyspaceSchema : public Schema<SettingsProvider> {
|
|
public:
|
|
using Schema::Schema;
|
|
|
|
/**
|
|
* @brief Construct a new Keyspace Schema object
|
|
*
|
|
* @param settingsProvider The settings provider
|
|
*/
|
|
struct KeyspaceStatements : public Schema<SettingsProvider>::Statements {
|
|
using Schema<SettingsProvider>::Statements::Statements;
|
|
|
|
//
|
|
// Insert queries
|
|
//
|
|
PreparedStatement insertLedgerRange = [this]() {
|
|
return handle_.get().prepare(
|
|
fmt::format(
|
|
R"(
|
|
INSERT INTO {} (is_latest, sequence) VALUES (?, ?) IF NOT EXISTS
|
|
)",
|
|
qualifiedTableName(settingsProvider_.get(), "ledger_range")
|
|
)
|
|
);
|
|
}();
|
|
|
|
//
|
|
// Update (and "delete") queries
|
|
//
|
|
PreparedStatement updateLedgerRange = [this]() {
|
|
return handle_.get().prepare(
|
|
fmt::format(
|
|
R"(
|
|
UPDATE {}
|
|
SET sequence = ?
|
|
WHERE is_latest = ?
|
|
IF sequence = ?
|
|
)",
|
|
qualifiedTableName(settingsProvider_.get(), "ledger_range")
|
|
)
|
|
);
|
|
}();
|
|
|
|
PreparedStatement selectLedgerRange = [this]() {
|
|
return handle_.get().prepare(
|
|
fmt::format(
|
|
R"(
|
|
SELECT sequence
|
|
FROM {}
|
|
WHERE is_latest in (True, False)
|
|
)",
|
|
qualifiedTableName(settingsProvider_.get(), "ledger_range")
|
|
)
|
|
);
|
|
}();
|
|
|
|
//
|
|
// Select queries
|
|
//
|
|
PreparedStatement selectNFTsAfterTaxonKeyspaces = [this]() {
|
|
return handle_.get().prepare(
|
|
fmt::format(
|
|
R"(
|
|
SELECT token_id
|
|
FROM {}
|
|
WHERE issuer = ?
|
|
AND taxon > ?
|
|
ORDER BY taxon ASC, token_id ASC
|
|
LIMIT ?
|
|
)",
|
|
qualifiedTableName(settingsProvider_.get(), "issuer_nf_tokens_v2")
|
|
)
|
|
);
|
|
}();
|
|
};
|
|
|
|
void
|
|
prepareStatements(Handle const& handle) override
|
|
{
|
|
LOG(log_.info()) << "Preparing aws keyspace statements";
|
|
statements_ = std::make_unique<KeyspaceStatements>(settingsProvider_, handle);
|
|
LOG(log_.info()) << "Finished preparing statements";
|
|
}
|
|
|
|
/**
|
|
* @brief Provides access to statements.
|
|
*
|
|
* @return The statements
|
|
*/
|
|
std::unique_ptr<KeyspaceStatements> const&
|
|
operator->() const
|
|
{
|
|
return statements_;
|
|
}
|
|
|
|
private:
|
|
std::unique_ptr<KeyspaceStatements> statements_{nullptr};
|
|
};
|
|
|
|
} // namespace data::cassandra
|