OracleConnection.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 <mutex>
#include <sptk5/db/PoolDatabaseConnection.h>
 
#ifdef HAVE_ORACLE
 
#include "DatabaseField.h"
#include <sptk5/db/OracleEnvironment.h>
#include <sptk5/db/OracleStatement.h>
#include <src/drivers/Oracle/OracleBulkInsertQuery.h>
 
namespace sptk {
 
class OracleStatement;
 
class OracleBulkInsertQuery;
 
class SP_EXPORT OracleConnection
    : public PoolDatabaseConnection
{
    friend class Query;
 
    friend class OracleStatement;
 
public:
    using Environment = oracle::occi::Environment;
    using Connection = oracle::occi::Connection;
    using Statement = oracle::occi::Statement;
    using Type = oracle::occi::Type;
 
    Connection* connection() const
    {
        return m_connection.get();
    }
 
    Environment* environment() const
    {
        return m_environment.handle();
    }
 
    Statement* createStatement(const std::string& sql)
    {
        return m_connection->createStatement(sql);
    }
 
    void _openDatabase(const String& connectionString) override;
 
    void _bulkInsert(const String& fullTableName, const Strings& columnNames,
                     const std::vector<VariantVector>& data) override;
 
    void _executeBatchSQL(const Strings& batchSQL, Strings* errors) override;
 
    explicit OracleConnection(const String& connectionString = "", std::chrono::seconds connectTimeout = std::chrono::seconds(60));
 
    ~OracleConnection() override;
 
    void closeDatabase() override;
 
    bool active() const override;
 
    DBHandle handle() const override;
 
    String nativeConnectionString() const override;
 
    String driverDescription() const override;
 
    void objectList(DatabaseObjectType objectType, Strings& objects) override;
 
    static std::map<OracleConnection*, std::shared_ptr<OracleConnection>> s_oracleConnections;
 
protected:
    void driverBeginTransaction() override;
 
    void driverEndTransaction(bool commit) override;
 
    // These methods implement the actions requested by CQuery
    String queryError(const Query* query) const override;
 
    void queryAllocStmt(Query* query) override;
 
    void queryFreeStmt(Query* query) override;
 
    void queryCloseStmt(Query* query) override;
 
    void queryPrepare(Query* query) override;
 
    void queryExecute(Query* query) override;
 
    int queryColCount(Query* query) override;
 
    void queryBindParameters(Query* query) override;
 
    void queryOpen(Query* query) override;
 
    void queryColAttributes(Query* query, int16_t column, int16_t descType, int32_t& value) override;
    void queryColAttributes(Query* query, int16_t column, int16_t descType, char* buff, int len) override;
 
    void queryFetch(Query* query) override;
 
    String paramMark(unsigned paramIndex) override;
 
private:
    using SConnection = std::shared_ptr<Connection>;
 
    mutable std::mutex m_mutex;      
    OracleEnvironment m_environment; 
    SConnection m_connection;        
    std::string m_lastError;         
 
    void executeMultipleStatements(const Strings& statements, Strings* errors);
 
    static void setMaxParamSizes(const CParamVector& enumeratedParams, Statement* stmt,
                                 const QueryColumnTypeSizeMap& columnTypeSizes);
 
    static void createQueryFieldsFromMetadata(Query* query, oracle::occi::ResultSet* resultSet);
 
    static void bulkInsertSingleRow(const Strings& columnNames, const QueryColumnTypeSizeVector& columnTypeSizeVector,
                                    sptk::OracleBulkInsertQuery& insertQuery, const VariantVector& row);
};
 
VariantDataType OracleTypeToVariantType(OracleConnection::Type oracleType, int scale);
 
OracleConnection::Type VariantTypeToOracleType(VariantDataType dataType);
 
#define throwOracleException(description)    \
    {                                        \
        m_lastError = description;           \
        throwDatabaseException(m_lastError); \
    }
 
} // namespace sptk
 
#endif
 
extern "C" {
[[maybe_unused]] SP_DRIVER_EXPORT void* oracle_create_connection(const char* connectionString, size_t connectionTimeoutSeconds);
[[maybe_unused]] SP_DRIVER_EXPORT void oracle_destroy_connection(void* connection);
}