prepared_statement_manager.h

Go to the documentation of this file.

#pragma once
 
#include <memory>
#include <mutex>
#include <unordered_map>
 
namespace kuzu {
namespace main {
 
struct CachedPreparedStatement;
 
class CachedPreparedStatementManager {
public:
    CachedPreparedStatementManager();
    ~CachedPreparedStatementManager();
 
    std::string addStatement(std::unique_ptr<CachedPreparedStatement> statement);
 
    bool containsStatement(const std::string& name) const { return statementMap.contains(name); }
 
    CachedPreparedStatement* getCachedStatement(const std::string& name) const;
 
private:
    std::mutex mtx;
    uint32_t currentIdx = 0;
    std::unordered_map<std::string, std::unique_ptr<CachedPreparedStatement>> statementMap;
};
 
} // namespace main
} // namespace kuzu