AutoDatabaseConnection.h

/*
╔══════════════════════════════════════════════════════════════════════════════╗
║                        SIMPLY POWERFUL TOOLKIT (SPTK)                        ║
╟──────────────────────────────────────────────────────────────────────────────╢
║  copyright            © 1999-2022 Alexey Parshin. All rights reserved.       ║
║  email                alexeyp@gmail.com                                      ║
╚══════════════════════════════════════════════════════════════════════════════╝
┌──────────────────────────────────────────────────────────────────────────────┐
│   This library is free software; you can redistribute it and/or modify it    │
│   under the terms of the GNU Library General Public License as published by  │
│   the Free Software Foundation; either version 2 of the License, or (at your │
│   option) any later version.                                                 │
│                                                                              │
│   This library is distributed in the hope that it will be useful, but        │
│   WITHOUT ANY WARRANTY; without even the implied warranty of                 │
│   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Library   │
│   General Public License for more details.                                   │
│                                                                              │
│   You should have received a copy of the GNU Library General Public License  │
│   along with this library; if not, write to the Free Software Foundation,    │
│   Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA.               │
│                                                                              │
│   Please report all bugs and problems to alexeyp@gmail.com.                  │
└──────────────────────────────────────────────────────────────────────────────┘
*/
 
#pragma once
 
#include <memory>
#include <sptk5/db/PoolDatabaseConnection.h>
 
namespace sptk {
 
class DatabaseConnectionPool;
 
class AutoDatabaseConnection
{
    DatabaseConnectionPool& m_connectionPool;
 
    SPoolDatabaseConnection m_connection {nullptr};
 
public:
    explicit AutoDatabaseConnection(DatabaseConnectionPool& connectionPool);
 
    AutoDatabaseConnection(const AutoDatabaseConnection&) = delete;
    AutoDatabaseConnection(AutoDatabaseConnection&&) = default;
 
    ~AutoDatabaseConnection();
 
    AutoDatabaseConnection& operator=(const AutoDatabaseConnection&) = delete;
    AutoDatabaseConnection& operator=(AutoDatabaseConnection&&) = delete;
 
    PoolDatabaseConnection* connection();
 
    void open(const String& connectionString = "") const
    {
        if (!m_connection)
            throw Exception("DB driver not loaded");
 
        m_connection->open(connectionString);
    }
 
    void close() const
    {
        m_connection->close();
    }
 
    bool active() const
    {
        return m_connection != nullptr && m_connection->active();
    }
 
    const DatabaseConnectionString& connectionString() const
    {
        return m_connection->connectionString();
    }
 
    DatabaseConnectionType connectionType() const
    {
        return m_connection->connectionType();
    }
 
    String driverDescription() const
    {
        return m_connection->driverDescription();
    }
 
    void beginTransaction() const
    {
        m_connection->beginTransaction();
    }
 
    void commitTransaction() const
    {
        m_connection->commitTransaction();
    }
 
    void rollbackTransaction() const
    {
        m_connection->rollbackTransaction();
    }
 
    int inTransaction() const
    {
        return m_connection->inTransaction();
    }
 
    void objectList(DatabaseObjectType objectType, Strings& objects) const
    {
        m_connection->objectList(objectType, objects);
    }
 
    void bulkInsert(const String& tableName, const Strings& columnNames, const std::vector<VariantVector>& data) const
    {
        m_connection->bulkInsert(tableName, columnNames, data);
    }
 
    void executeBatchFile(const String& batchFileName, Strings* errors = nullptr) const
    {
        m_connection->executeBatchFile(batchFileName, errors);
    }
 
    void executeBatchSQL(const sptk::Strings& batchSQL, Strings* errors = nullptr) const
    {
        m_connection->executeBatchSQL(batchSQL, errors);
    }
};
 
using DatabaseConnection = std::shared_ptr<AutoDatabaseConnection>;
 
} // namespace sptk