postgresql_backend.h

Go to the documentation of this file.

// BSD 3-Clause License
// Copyright (c) 2025, 🍀☀🌕🌥 🌊
// See the LICENSE file in the project root for full license information.
 
#pragma once
 
#include "../core/backend_base.h"
#include "../core/backend_registry.h"
 
#include <memory>
#include <string>
#include <vector>
#include <atomic>
 
namespace database
{
namespace backends
{
 
class postgresql_backend
    : public core::backend_base<postgresql_backend, database_types::postgres>
{
public:
    static constexpr const char* backend_name() { return "postgresql_backend"; }
 
    postgresql_backend();
 
    ~postgresql_backend() override = default;
 
    // database_backend interface implementation
 
    kcenon::common::Result<core::database_result> select_query(const std::string& query_string) override;
 
    kcenon::common::VoidResult execute_query(const std::string& query_string) override;
 
    [[nodiscard]] kcenon::common::Result<core::database_result> select_prepared(
        const std::string& query,
        const std::vector<core::database_value>& params) override;
 
    [[nodiscard]] kcenon::common::VoidResult execute_prepared(
        const std::string& query,
        const std::vector<core::database_value>& params) override;
 
    kcenon::common::VoidResult begin_transaction() override;
 
    kcenon::common::VoidResult commit_transaction() override;
 
    kcenon::common::VoidResult rollback_transaction() override;
 
    bool in_transaction() const override;
 
    std::string last_error() const override;
 
    std::map<std::string, std::string> connection_info() const override;
 
    kcenon::common::Result<uint64_t> execute_batch(const std::vector<std::string>& queries);
 
protected:
    friend class core::backend_base<postgresql_backend, database_types::postgres>;
 
    kcenon::common::VoidResult do_initialize(const core::connection_config& config);
 
    kcenon::common::VoidResult do_shutdown();
 
private:
    std::string build_connection_string(const core::connection_config& config) const;
 
    std::string build_safe_connection_string(const core::connection_config& config) const;
 
    std::string sanitize_error(const std::string& error_message) const;
 
    unsigned int execute_modification_query(const std::string& query_string);
 
    void* connection_{nullptr};                 
    std::atomic<bool> in_transaction_{false};   
    mutable std::string last_error_;            
    core::connection_config connection_config_; 
};
 
} // namespace backends
} // namespace database