database_connection.hpp

// SPDX-License-Identifier: Apache-2.0
// Copyright 2020 - 2024 Pionix GmbH and Contributors to EVerest
 
#pragma once
 
#include <mutex>
#include <sqlite3.h>
 
#include <ocpp/common/support_older_cpp_versions.hpp>
 
#include "sqlite_statement.hpp"
 
namespace ocpp::common {
 
class DatabaseTransactionInterface {
public:
    virtual ~DatabaseTransactionInterface() = default;
 
    virtual void commit() = 0;
 
    virtual void rollback() = 0;
};
 
class DatabaseConnectionInterface {
public:
    virtual ~DatabaseConnectionInterface() = default;
 
    virtual bool open_connection() = 0;
 
    virtual bool close_connection() = 0;
 
    [[nodiscard]] virtual std::unique_ptr<DatabaseTransactionInterface> begin_transaction() = 0;
 
    virtual bool execute_statement(const std::string& statement) = 0;
 
    virtual std::unique_ptr<SQLiteStatementInterface> new_statement(const std::string& sql) = 0;
 
    virtual const char* get_error_message() = 0;
 
    virtual bool clear_table(const std::string& table) = 0;
 
    virtual int64_t get_last_inserted_rowid() = 0;
 
    virtual void set_user_version(uint32_t version) = 0;
 
    virtual uint32_t get_user_version() = 0;
};
 
class DatabaseConnection : public DatabaseConnectionInterface {
private:
    sqlite3* db;
    const fs::path database_file_path;
    std::atomic_uint32_t open_count;
    std::timed_mutex transaction_mutex;
 
    bool close_connection_internal(bool force_close);
 
public:
    explicit DatabaseConnection(const fs::path& database_file_path) noexcept;
 
    virtual ~DatabaseConnection();
 
    bool open_connection() override;
    bool close_connection() override;
 
    [[nodiscard]] std::unique_ptr<DatabaseTransactionInterface> begin_transaction() override;
 
    bool execute_statement(const std::string& statement) override;
    std::unique_ptr<SQLiteStatementInterface> new_statement(const std::string& sql) override;
 
    const char* get_error_message() override;
 
    bool clear_table(const std::string& table) override;
 
    int64_t get_last_inserted_rowid() override;
 
    uint32_t get_user_version() override;
    void set_user_version(uint32_t version) override;
};
 
} // namespace ocpp::common