ODBCEnvironment.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 <sptk5/sptk.h>
 
#ifdef HAVE_ODBC
 
#include <sptk5/Strings.h>
 
#ifdef WIN32
#include <sys/types.h>
#include <time.h>
#include <winsock2.h>
 
#include <windows.h>
#else
#define LPCVOID const void*
#endif
 
#include <cassert>
#include <sqlext.h>
 
#include <mutex>
#include <sptk5/db/QueryParameterList.h>
 
namespace sptk {
 
class ODBCEnvironment;
 
class ODBCConnectionBase;
 
class SP_DRIVER_EXPORT ODBCBase
    : public std::mutex
{
    friend class ODBCConnection;
 
public:
    static constexpr const char* cantSetConnectOption = "Can't set connect option";
    static constexpr const char* cantEndTranscation = "Can't end transaction";
    static constexpr const char* cantGetInformation = "Can't get connect information";
 
    ~ODBCBase() = default;
 
    [[noreturn]] static void exception(const String& text, int line);
 
protected:
    ODBCBase() = default;
 
private:
    ODBCBase& operator=(const ODBCBase& d);
 
    ODBCBase(const ODBCBase&);
};
 
class SP_DRIVER_EXPORT ODBCEnvironment
    : public ODBCBase
{
    friend class ODBCConnectionBase;
 
public:
    SQLHENV handle() const
    {
        return m_hEnvironment.get();
    }
 
protected:
    ODBCEnvironment() = default;
 
    void allocEnv();
 
    bool valid() const
    {
        return m_hEnvironment != nullptr;
    }
 
private:
    std::shared_ptr<uint8_t> m_hEnvironment;
};
 
class SP_DRIVER_EXPORT ODBCConnectionBase
    : public ODBCBase
{
public:
    void allocConnect();
 
    void freeConnect();
 
    void connect(const String& ConnectionString, String& FinalConnectionString, bool EnableDriverPrompt = false);
 
    void disconnect();
 
    SQLHDBC handle() const
    {
        return (SQLHDBC) m_hConnection.get();
    }
 
    bool isConnected()
    {
        std::scoped_lock lock(*this);
        return m_connected;
    }
 
    void setConnectOption(UWORD fOption, UDWORD vParam);
 
    String connectString()
    {
        std::scoped_lock lock(*this);
        return m_connectString;
    }
 
    String driverDescription()
    {
        std::scoped_lock lock(*this);
        return m_driverDescription;
    }
 
    void beginTransaction();
 
    void transact(uint16_t fType);
 
    void commit()
    {
        transact(SQL_COMMIT);
    }
 
    void rollback()
    {
        transact(SQL_ROLLBACK);
    }
 
    static ODBCEnvironment& getEnvironment()
    {
        return m_env;
    }
 
    String errorInformation(const char* action) const;
 
protected:
    bool valid() const
    {
        return m_hConnection != SQL_NULL_HDBC;
    }
 
    void execQuery(const char* query);
 
private:
    ODBCEnvironment& m_cEnvironment {getEnvironment()}; 
    std::shared_ptr<uint8_t> m_hConnection;             
    bool m_connected {false};                           
    String m_connectString;                             
    String m_driverDescription;                         
    static ODBCEnvironment m_env;
};
 
String removeDriverIdentification(const char* error);
 
} // namespace sptk
#endif