finished statement cache class

This commit is contained in:
Sascha Kühl 2025-08-05 19:30:40 +02:00
parent 31964e55c1
commit 35fad9f47c
13 changed files with 410 additions and 74 deletions

View File

@ -75,8 +75,8 @@ int main() {
const object::schema schema("Administration"); const object::schema schema("Administration");
sql::connection_pool<sql::connection> pool("postgres://news:news@127.0.0.1:15432/matador", 4); // sql::connection_pool<sql::connection> pool("postgres://news:news@127.0.0.1:15432/matador", 4);
// sql::connection_pool<sql::connection> pool("postgres://test:test123!@127.0.0.1:5432/matador", 4); sql::connection_pool pool("postgres://test:test123!@127.0.0.1:5432/matador", 4);
orm::session ses(pool); orm::session ses(pool);
@ -122,8 +122,3 @@ int main() {
return 0; return 0;
} }
// registering table 'collection_center' (pk field: 'id')
// registering relation table 'collection_center_users' (first fk field: 'collection_center_id', second fk field: 'user_id')
// registering table 'internal_user_directory' (pk field: 'id')
//

View File

@ -26,14 +26,14 @@ public:
virtual utils::result<std::unique_ptr<query_result_impl>, utils::error> fetch(const interface::parameter_binder& bindings) = 0; virtual utils::result<std::unique_ptr<query_result_impl>, utils::error> fetch(const interface::parameter_binder& bindings) = 0;
template < class Type > template < class Type >
void bind_object(Type &obj, const interface::parameter_binder& bindings) { void bind_object(Type &obj, interface::parameter_binder& bindings) {
object_parameter_binder object_binder_; object_parameter_binder object_binder_;
object_binder_.reset(start_index()); object_binder_.reset(start_index());
object_binder_.bind(obj, bindings); object_binder_.bind(obj, bindings);
} }
template < class Type > template < class Type >
void bind(const size_t pos, Type &val, const interface::parameter_binder& bindings) { void bind(const size_t pos, Type &val, interface::parameter_binder& bindings) {
utils::data_type_traits<Type>::bind_value(bindings, adjust_index(pos), val); utils::data_type_traits<Type>::bind_value(bindings, adjust_index(pos), val);
} }

View File

@ -16,11 +16,11 @@ public:
virtual utils::result<std::unique_ptr<query_result_impl>, utils::error> fetch(interface::parameter_binder& bindings) = 0; virtual utils::result<std::unique_ptr<query_result_impl>, utils::error> fetch(interface::parameter_binder& bindings) = 0;
template<class Type> template<class Type>
void bind(const Type &obj, const interface::parameter_binder& bindings) { void bind(const Type &obj, interface::parameter_binder& bindings) {
statement_->bind_object(obj, bindings); statement_->bind_object(obj, bindings);
} }
template<typename Type> template<typename Type>
void bind(size_t pos, Type &value, const interface::parameter_binder& bindings) { void bind(size_t pos, Type &value, interface::parameter_binder& bindings) {
statement_->bind(pos, value, bindings); statement_->bind(pos, value, bindings);
} }
void bind(size_t pos, const char *value, size_t size, interface::parameter_binder& bindings) const; void bind(size_t pos, const char *value, size_t size, interface::parameter_binder& bindings) const;
@ -28,6 +28,8 @@ public:
void reset() const; void reset() const;
[[nodiscard]] std::string sql() const;
[[nodiscard]] std::unique_ptr<utils::attribute_writer> create_binder() const; [[nodiscard]] std::unique_ptr<utils::attribute_writer> create_binder() const;
protected: protected:

View File

@ -31,17 +31,17 @@ struct sql_command_info {
struct query_context { struct query_context {
std::string sql; std::string sql;
sql_command command{}; sql_command command{};
std::string command_name; std::string command_name{};
sql::table table{""}; sql::table table{""};
std::vector<object::attribute_definition> prototype; std::vector<object::attribute_definition> prototype{};
std::vector<std::string> result_vars; std::vector<std::string> result_vars{};
std::vector<std::string> bind_vars; std::vector<std::string> bind_vars{};
std::vector<utils::database_type> bind_types; std::vector<utils::database_type> bind_types{};
std::unordered_map<std::string, std::string> column_aliases; std::unordered_map<std::string, std::string> column_aliases{};
std::unordered_map<std::string, std::string> table_aliases; std::unordered_map<std::string, std::string> table_aliases{};
std::vector<sql_command_info> additional_commands; std::vector<sql_command_info> additional_commands{};
}; };
} }

View File

@ -5,7 +5,6 @@
#include "matador/sql/query_result.hpp" #include "matador/sql/query_result.hpp"
#include "matador/sql/interface/statement_proxy.hpp" #include "matador/sql/interface/statement_proxy.hpp"
#include "matador/sql/interface/parameter_binder.hpp"
#include "matador/utils/error.hpp" #include "matador/utils/error.hpp"
#include "matador/utils/result.hpp" #include "matador/utils/result.hpp"
@ -122,6 +121,8 @@ public:
*/ */
void reset() const; void reset() const;
[[nodiscard]] std::string sql() const;
private: private:
template<class Type> template<class Type>
friend class detail::identifier_binder; friend class detail::identifier_binder;
@ -140,7 +141,7 @@ statement &statement::bind(size_t pos, Type &value) {
template<class Type> template<class Type>
statement &statement::bind(const Type &obj) { statement &statement::bind(const Type &obj) {
statement_proxy_->bind(obj, bindings_); statement_proxy_->bind(obj, *bindings_);
return *this; return *this;
} }

View File

@ -2,8 +2,7 @@
#define STATEMENT_CACHE_HPP #define STATEMENT_CACHE_HPP
#include "matador/sql/executor.hpp" #include "matador/sql/executor.hpp"
#include "matador/sql/statement.hpp"
#include "matador/sql/interface/statement_proxy.hpp"
#include <list> #include <list>
#include <mutex> #include <mutex>
@ -13,9 +12,36 @@ namespace matador::sql {
class connection_pool; class connection_pool;
struct statement_cache_event {
enum class Type { Accessed, Added, Evicted };
Type type;
std::string sql;
std::chrono::steady_clock::time_point timestamp;
};
class statement_cache_observer_interface {
public:
virtual void on_event(const statement_cache_event&) = 0;
virtual ~statement_cache_observer_interface() = default;
};
class statement_cache final { class statement_cache final {
private:
using list_iterator = std::list<size_t>::iterator;
struct cache_entry {
statement stmt;
std::chrono::steady_clock::time_point last_access;
list_iterator position;
};
public: public:
explicit statement_cache(connection_pool &pool, size_t max_size = 50); explicit statement_cache(connection_pool &pool, size_t max_size = 50);
statement_cache(const statement_cache &) = delete;
statement_cache &operator=(const statement_cache &) = delete;
statement_cache(statement_cache &&) = delete;
statement_cache &operator=(statement_cache &&) = delete;
~statement_cache() = default;
[[nodiscard]] utils::result<statement, utils::error> acquire(const query_context &ctx); [[nodiscard]] utils::result<statement, utils::error> acquire(const query_context &ctx);
@ -23,16 +49,22 @@ public:
[[nodiscard]] size_t capacity() const; [[nodiscard]] size_t capacity() const;
[[nodiscard]] bool empty() const; [[nodiscard]] bool empty() const;
void subscribe(statement_cache_observer_interface &observer);
private:
void push(statement_cache_event::Type type, const std::string& sql) const;
private: private:
using list_iterator = std::list<size_t>::iterator;
size_t max_size_{}; size_t max_size_{};
std::list<size_t> usage_list_; // LRU: front = most recent, back = least recent std::list<size_t> usage_list_; // LRU: front = most recent, back = least recent
std::unordered_map<size_t, std::pair<statement, list_iterator>> cache_map_; std::unordered_map<size_t, cache_entry> cache_map_;
std::mutex mutex_; std::mutex mutex_;
connection_pool &pool_; connection_pool &pool_;
const sql::dialect &dialect_; const sql::dialect &dialect_;
std::vector<statement_cache_observer_interface*> observers_;
}; };
} }
#endif //STATEMENT_CACHE_HPP #endif //STATEMENT_CACHE_HPP

View File

@ -71,9 +71,14 @@ connection_pool::connection_pool(const std::string& dns, size_t count)
connection_ptr connection_pool::acquire() { connection_ptr connection_pool::acquire() {
std::unique_lock lock(mutex_); std::unique_lock lock(mutex_);
while (idle_connections_.empty()) { if (!cv.wait_for(lock,
cv.wait(lock); std::chrono::seconds(30),
[this] { return !idle_connections_.empty(); })) {
return {nullptr, this};
} }
// while (idle_connections_.empty()) {
// cv.wait(lock);
// }
return get_next_connection(); return get_next_connection();
} }
@ -89,23 +94,40 @@ connection_ptr connection_pool::try_acquire() {
connection_ptr connection_pool::acquire(const size_t id) { connection_ptr connection_pool::acquire(const size_t id) {
using namespace std::chrono_literals; using namespace std::chrono_literals;
pointer next_connection{nullptr};
auto try_count{0};
std::unique_lock lock(mutex_); std::unique_lock lock(mutex_);
do { if (!cv.wait_for(lock,
if (auto it = idle_connections_.find(id); it != idle_connections_.end()) { 5s,
next_connection = it->second; [this, id] {
auto node = idle_connections_.extract(it); return idle_connections_.find(id) != idle_connections_.end();
inuse_connections_.insert(std::move(node)); })) {
} else { return {nullptr, this};
lock.unlock(); }
std::this_thread::sleep_for(100ms);
lock.lock();
}
} while(try_count++ < 5);
auto it = idle_connections_.find(id);
auto next_connection = it->second;
auto node = idle_connections_.extract(it);
inuse_connections_.insert(std::move(node));
return {next_connection, this}; return {next_connection, this};
// using namespace std::chrono_literals;
// pointer next_connection{nullptr};
// auto try_count{0};
// std::unique_lock lock(mutex_);
//
// do {
// if (auto it = idle_connections_.find(id); it != idle_connections_.end()) {
// next_connection = it->second;
// auto node = idle_connections_.extract(it);
// inuse_connections_.insert(std::move(node));
// } else {
// lock.unlock();
// std::this_thread::sleep_for(100ms);
// lock.lock();
// }
// } while(try_count++ < 5);
//
// return {next_connection, this};
} }
void connection_pool::release(identifiable_connection* c) { void connection_pool::release(identifiable_connection* c) {
@ -116,6 +138,7 @@ void connection_pool::release(identifiable_connection* c) {
if (const auto it = inuse_connections_.find(c->id); it != inuse_connections_.end()) { if (const auto it = inuse_connections_.find(c->id); it != inuse_connections_.end()) {
auto node = inuse_connections_.extract(it); auto node = inuse_connections_.extract(it);
idle_connections_.insert(std::move(node)); idle_connections_.insert(std::move(node));
cv.notify_one();
} }
} }

View File

@ -15,6 +15,10 @@ void statement_proxy::reset() const {
statement_->reset(); statement_->reset();
} }
std::string statement_proxy::sql() const {
return statement_->query_.sql;
}
std::unique_ptr<utils::attribute_writer> statement_proxy::create_binder() const { std::unique_ptr<utils::attribute_writer> statement_proxy::create_binder() const {
return statement_->create_binder(); return statement_->create_binder();
} }

View File

@ -95,4 +95,7 @@ void statement::reset() const
statement_proxy_->reset(); statement_proxy_->reset();
} }
std::string statement::sql() const {
return statement_proxy_->sql();
}
} }

View File

@ -4,35 +4,65 @@
#include "matador/sql/connection_pool.hpp" #include "matador/sql/connection_pool.hpp"
#include <atomic> #include <atomic>
#include <thread>
namespace matador::sql { namespace matador::sql {
namespace internal { namespace internal {
class statement_cache_proxy final : public statement_proxy { class statement_cache_proxy final : public statement_proxy {
public: public:
explicit statement_cache_proxy(std::unique_ptr<statement_impl>&& stmt) struct retry_config {
: statement_proxy(std::move(stmt)) {} size_t max_attempts{10};
std::chrono::milliseconds initial_wait{10};
std::chrono::milliseconds max_wait{250};
};
explicit statement_cache_proxy(std::unique_ptr<statement_impl>&& stmt, connection_pool &pool, const size_t connection_id)
: statement_proxy(std::move(stmt))
, pool_(pool)
, connection_id_(connection_id) {}
utils::result<size_t, utils::error> execute(interface::parameter_binder& bindings) override { utils::result<size_t, utils::error> execute(interface::parameter_binder& bindings) override {
if (!try_lock()) { auto result = try_with_retry([this, &bindings]() -> utils::result<size_t, utils::error> {
return utils::failure(utils::error{ if (!try_lock()) {
error_code::STATEMENT_LOCKED, return utils::failure(utils::error{
"Failed to execute statement because it is already in use" error_code::STATEMENT_LOCKED,
}); "Failed to execute statement because it is already in use"
} });
}
auto guard = statement_guard(*this); auto guard = statement_guard(*this);
return statement_->execute(bindings); if (auto conn = pool_.acquire(connection_id_); !conn.valid()) {
return utils::failure(utils::error{
error_code::EXECUTE_FAILED,
"Failed to execute statement because couldn't lock connection"
});
}
return statement_->execute(bindings);
});
return result;
} }
utils::result<std::unique_ptr<query_result_impl>, utils::error> fetch(interface::parameter_binder& bindings) override {
if (!try_lock()) {
return utils::failure(utils::error{
error_code::STATEMENT_LOCKED,
"Failed to execute statement because it is already in use"
});
}
auto guard = statement_guard(*this); utils::result<std::unique_ptr<query_result_impl>, utils::error> fetch(interface::parameter_binder& bindings) override {
return statement_->fetch(bindings); auto result = try_with_retry([this, &bindings]() -> utils::result<std::unique_ptr<query_result_impl>, utils::error> {
if (!try_lock()) {
return utils::failure(utils::error{
error_code::STATEMENT_LOCKED,
"Failed to execute statement because it is already in use"
});
}
auto guard = statement_guard(*this);
if (const auto conn = pool_.acquire(connection_id_); !conn.valid()) {
return utils::failure(utils::error{
error_code::EXECUTE_FAILED,
"Failed to execute statement because couldn't lock connection"
});
}
return statement_->fetch(bindings);
});
return result;
} }
protected: protected:
@ -41,6 +71,28 @@ protected:
return locked_.compare_exchange_strong(expected, true); return locked_.compare_exchange_strong(expected, true);
} }
template<typename Func>
[[nodiscard]] auto try_with_retry(Func &&func) -> decltype(func()) {
auto current_wait = config_.initial_wait;
for (size_t attempt = 0; attempt < config_.max_attempts; ++attempt) {
if (auto result = func(); result.is_ok() ||
result.err().ec() != error_code::STATEMENT_LOCKED) {
return result;
}
if (attempt + 1 < config_.max_attempts) {
std::this_thread::sleep_for(current_wait);
current_wait = std::min(current_wait * 2, config_.max_wait);
}
}
return utils::failure(utils::error{
error_code::STATEMENT_LOCKED,
"Failed to execute statement because it is already in use"
});
}
void unlock() { void unlock() {
locked_.store(false); locked_.store(false);
} }
@ -56,6 +108,9 @@ private:
private: private:
std::atomic_bool locked_{false}; std::atomic_bool locked_{false};
connection_pool &pool_;
size_t connection_id_{};
retry_config config_{};
}; };
} }
@ -68,35 +123,46 @@ utils::result<statement, utils::error> statement_cache::acquire(const query_cont
std::unique_lock lock(mutex_); std::unique_lock lock(mutex_);
// hash statement // hash statement
const auto key = std::hash<std::string>{}(ctx.sql); const auto key = std::hash<std::string>{}(ctx.sql);
const auto now = std::chrono::steady_clock::now();
// Found in cache. Move it to of the LRU list // Found in cache. Move it to of the LRU list
if (const auto it = cache_map_.find(key); it != cache_map_.end()) { if (const auto it = cache_map_.find(key); it != cache_map_.end()) {
usage_list_.splice(usage_list_.begin(), usage_list_, it->second.second); usage_list_.splice(usage_list_.begin(), usage_list_, it->second.position);
return utils::ok(it->second.first); it->second.last_access = now;
push(statement_cache_event::Type::Accessed, ctx.sql);
return utils::ok(it->second.stmt);
} }
// Prepare a new statement // Prepare a new statement
// acquire pool connection // acquire pool connection
const auto conn = pool_.acquire(); size_t id{};
auto result = conn->perform_prepare(ctx); std::unique_ptr<statement_impl> stmt;
if (!result) { {
return utils::failure(utils::error{error_code::PREPARE_FAILED, std::string("Failed to prepare")}); const auto conn = pool_.acquire();
auto result = conn->perform_prepare(ctx);
if (!result) {
return utils::failure(utils::error{error_code::PREPARE_FAILED, std::string("Failed to prepare")});
}
id = conn.id().value();
stmt = result.release();
} }
// If cache max size reached ensure space // If cache max size reached ensure space
if (cache_map_.size() >= max_size_) { if (cache_map_.size() >= max_size_) {
const auto& key_to_remove = usage_list_.back(); const auto& key_to_remove = usage_list_.back();
cache_map_.erase(key_to_remove); cache_map_.erase(key_to_remove);
usage_list_.pop_back(); usage_list_.pop_back();
push(statement_cache_event::Type::Evicted, ctx.sql);
} }
usage_list_.push_front(key); usage_list_.push_front(key);
const auto it = cache_map_.insert({ const auto it = cache_map_.insert({
key, key,
std::make_pair(statement{ {statement{
std::make_shared<internal::statement_cache_proxy>(result.release())}, std::make_shared<internal::statement_cache_proxy>(std::move(stmt), pool_, id)},
usage_list_.begin()) std::chrono::steady_clock::now(),
usage_list_.begin()}
}).first; }).first;
push(statement_cache_event::Type::Added, ctx.sql);
return utils::ok(it->second.first); return utils::ok(it->second.stmt);
} }
size_t statement_cache::size() const { size_t statement_cache::size() const {
@ -109,4 +175,19 @@ size_t statement_cache::capacity() const {
bool statement_cache::empty() const { bool statement_cache::empty() const {
return cache_map_.empty(); return cache_map_.empty();
} }
void statement_cache::subscribe(statement_cache_observer_interface &observer) {
observers_.push_back(&observer);
}
void statement_cache::push(statement_cache_event::Type type, const std::string &sql) const {
using Clock = std::chrono::steady_clock;
const auto ts = Clock::now();
const statement_cache_event evt{type, sql, ts};
for (auto& obs : observers_) {
if (obs) {
obs->on_event(evt);
}
}
}
} }

View File

@ -66,7 +66,7 @@ TEST_CASE("Test log file sink", "[logger][log][file_sink]") {
strerror_s(buf, 1024, errno); strerror_s(buf, 1024, errno);
FAIL(buf); FAIL(buf);
#else #else
UNIT_FAIL(strerror(errno)); FAIL(strerror(errno));
#endif #endif
} }
@ -89,7 +89,7 @@ TEST_CASE("Test log file sink", "[logger][log][file_sink]") {
strerror_s(buf, 1024, errno); strerror_s(buf, 1024, errno);
FAIL(buf); FAIL(buf);
#else #else
UNIT_FAIL(strerror(errno)); FAIL(strerror(errno));
#endif #endif
} }

View File

@ -2,11 +2,18 @@
#include "test_result_reader.hpp" #include "test_result_reader.hpp"
#include "test_parameter_binder.hpp" #include "test_parameter_binder.hpp"
#include <random>
#include <thread>
namespace matador::test::orm { namespace matador::test::orm {
test_statement::test_statement(const sql::query_context &query) test_statement::test_statement(const sql::query_context &query)
: statement_impl(query) {} : statement_impl(query) {}
utils::result<size_t, utils::error> test_statement::execute(const sql::interface::parameter_binder &/*bindings*/) { utils::result<size_t, utils::error> test_statement::execute(const sql::interface::parameter_binder &/*bindings*/) {
using namespace std::chrono_literals;
std::mt19937 rng(query_.sql.size());
std::uniform_int_distribution dist(10, 40);
std::this_thread::sleep_for(std::chrono::milliseconds(dist(rng)));
return utils::ok(static_cast<size_t>(8)); return utils::ok(static_cast<size_t>(8));
} }

View File

@ -1,18 +1,44 @@
#include <atomic>
#include <catch2/catch_test_macros.hpp> #include <catch2/catch_test_macros.hpp>
#include <matador/query/query.hpp> #include <matador/query/query.hpp>
#include "matador/sql/connection_pool.hpp" #include "matador/sql/connection_pool.hpp"
#include "matador/sql/error_code.hpp"
#include "matador/sql/statement_cache.hpp" #include "matador/sql/statement_cache.hpp"
#include "../backend/test_backend_service.hpp" #include "../backend/test_backend_service.hpp"
#include "ConnectionPoolFixture.hpp" #include "ConnectionPoolFixture.hpp"
#include <queue>
#include <random>
#include <thread>
using namespace matador::test; using namespace matador::test;
using namespace matador::sql; using namespace matador::sql;
using namespace matador::query; using namespace matador::query;
class RecordingObserver final : public statement_cache_observer_interface {
public:
void on_event(const statement_cache_event& evt) override {
std::lock_guard lock(mutex);
events.push(evt);
}
std::optional<statement_cache_event> poll() {
std::lock_guard lock(mutex);
if (events.empty()) return std::nullopt;
auto evt = events.front();
events.pop();
return evt;
}
private:
std::mutex mutex;
std::queue<statement_cache_event> events;
};
TEST_CASE("Test statement cache", "[statement][cache]") { TEST_CASE("Test statement cache", "[statement][cache]") {
backend_provider::instance().register_backend("noop", std::make_unique<orm::test_backend_service>()); backend_provider::instance().register_backend("noop", std::make_unique<orm::test_backend_service>());
@ -52,3 +78,165 @@ TEST_CASE("Test statement cache", "[statement][cache]") {
REQUIRE(!cache.empty()); REQUIRE(!cache.empty());
REQUIRE(cache.capacity() == 2); REQUIRE(cache.capacity() == 2);
} }
TEST_CASE("Test LRU cache evicts oldest entries", "[statement][cache][evict]") {
backend_provider::instance().register_backend("noop", std::make_unique<orm::test_backend_service>());
connection_pool pool("noop://noop.db", 4);
statement_cache cache(pool, 2);
RecordingObserver observer;
cache.subscribe(observer);
REQUIRE(cache.capacity() == 2);
REQUIRE(cache.empty());
auto result = cache.acquire({"SELECT * FROM person"});
REQUIRE(result);
auto stmt1 = result.value();
result = cache.acquire({"SELECT title FROM book"});
REQUIRE(result);
auto stmt2 = result.value();
result = cache.acquire({"SELECT name FROM author"}); // Should evict first statement
REQUIRE(result);
auto stmt3 = result.value();
// Trigger re-prepare of evicted statement
result = cache.acquire({"SELECT 1"});
REQUIRE(result);
auto stmt4 = result.value();
REQUIRE(stmt1.sql() == "SELECT * FROM person");
REQUIRE(stmt2.sql() == "SELECT title FROM book");
REQUIRE(stmt3.sql() == "SELECT name FROM author");
REQUIRE(stmt4.sql() == "SELECT 1");
REQUIRE(cache.size() == 2);
REQUIRE(!cache.empty());
REQUIRE(cache.capacity() == 2);
int added = 0, evicted = 0;
while (auto e = observer.poll()) {
if (e->type == statement_cache_event::Type::Added) added++;
if (e->type == statement_cache_event::Type::Evicted) evicted++;
}
REQUIRE(added >= 3);
REQUIRE(evicted >= 1);
}
TEST_CASE("Test statement reuse avoids reprepare", "[statement][cache][prepare]") {
backend_provider::instance().register_backend("noop", std::make_unique<orm::test_backend_service>());
connection_pool pool("noop://noop.db", 4);
statement_cache cache(pool, 2);
RecordingObserver observer;
cache.subscribe(observer);
REQUIRE(cache.capacity() == 2);
REQUIRE(cache.empty());
auto result = cache.acquire({"SELECT * FROM person"});
REQUIRE(result);
auto stmt1 = result.value();
result = cache.acquire({"SELECT * FROM person"});
REQUIRE(result);
auto stmt2 = result.value();
}
TEST_CASE("Multithreaded stress test", "[statement][cache][stress]") {
backend_provider::instance().register_backend("noop", std::make_unique<orm::test_backend_service>());
constexpr int thread_count = 16;
constexpr int iterations = 1000;
constexpr int sql_pool_size = 10;
std::vector<std::string> sqls;
for (int i = 0; i < sql_pool_size; ++i) {
sqls.push_back("SELECT " + std::to_string(i));
}
connection_pool pool("noop://noop.db", 4);
statement_cache cache(pool, 5);
RecordingObserver observer;
cache.subscribe(observer);
auto start_time = std::chrono::steady_clock::now();
std::atomic_int lock_failed_count{0};
std::atomic_int exec_failed_count{0};
auto worker = [&](const int tid) {
std::mt19937 rng(tid);
std::uniform_int_distribution dist(0, sql_pool_size - 1);
for (int i = 0; i < iterations; ++i) {
const auto& sql = sqls[dist(rng)];
if (const auto result = cache.acquire({sql}); !result) {
FAIL("Failed to acquire statement");
} else {
if (const auto exec_result = result->execute(); !exec_result) {
if (exec_result.err().ec() == error_code::STATEMENT_LOCKED) {
++lock_failed_count;
} else {
++exec_failed_count;
}
}
}
}
};
std::vector<std::thread> threads;
for (int i = 0; i < thread_count; ++i) {
threads.emplace_back(worker, i);
}
for (auto& t : threads) {
t.join();
}
auto end_time = std::chrono::steady_clock::now();
auto duration = std::chrono::duration_cast<std::chrono::milliseconds>(end_time - start_time);
std::cout << "[Performance] Executed " << (thread_count * iterations) << " statements in " << duration.count() << " ms (lock failed: " << lock_failed_count << ", execute failed: " << exec_failed_count << ")\n";
// Some events should be generated
int accessed = 0;
while (auto e = observer.poll()) {
if (e->type == statement_cache_event::Type::Accessed) accessed++;
}
REQUIRE(accessed > 0);
}
TEST_CASE("Race condition simulation with mixed access", "[statement_cache][race]") {
backend_provider::instance().register_backend("noop", std::make_unique<orm::test_backend_service>());
connection_pool pool("noop://noop.db", 4);
statement_cache cache(pool, 5);
constexpr int threads = 8;
constexpr int operations = 500;
auto task = [&](int id) {
for (int i = 0; i < operations; ++i) {
auto sql = "SELECT " + std::to_string(i % 10);
auto result = cache.acquire({sql});
REQUIRE(result);
// if (i % 50 == 0) {
// cache.cleanup_expired_connections();
// }
}
};
std::vector<std::thread> jobs;
for (int i = 0; i < threads; ++i) {
jobs.emplace_back(task, i);
}
for (auto& t : jobs) {
t.join();
}
SUCCEED("Race simulation completed successfully without crash");
}