cODBCEnv.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cODBCEnv_H
#define _INC_cODBCEnv_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cODBC.h"
#include "Sql/ISQLDriver.h"
#include "Sql/cSQLCmd.h"
 
#ifdef USE_SQL
 
#include <odbcinst.h>   // SQLGetInstalledDrivers. SQL_SUCCESS
#include <sqltypes.h>   // use SQLRETURN, SQLSMALLINT, SQLHANDLE, SQL_HANDLE_DBC
 
namespace GrayLib
{
        class GRAYLIB_LINK cODBCBase : public cRefBase
        {
                typedef cRefBase SUPER_t;
 
        public:
                const SQLSMALLINT m_nSQLHandleType;     
 
                SQLHANDLE m_hSQL;               
                SQLRETURN m_nSQLRet;    
 
        public:
                cODBCBase(SQLSMALLINT nSQLHandleType, cODBCBase* pParent);
                virtual ~cODBCBase();
 
                inline bool isSQLRetOK2() const noexcept
                {
                        return m_nSQLRet == SQL_SUCCESS || m_nSQLRet == SQL_SUCCESS_WITH_INFO ;
                }
 
                virtual bool SetSQLAttr(long nAttribute, void* pValue) = 0;
 
                virtual HRESULT FillException(cSQLException* pException = nullptr);
 
                IUNKNOWN_DISAMBIG(SUPER_t);
 
#define cODBCBase_DISAMBIG virtual HRESULT FillException(cSQLException* pException=nullptr) override { return SUPER_t::FillException(pException); } 
 
        };
 
        //***********************************************************
 
        class GRAYLIB_LINK cODBCStatement : public cODBCBase, public ISQLStatement
        {
                typedef cODBCBase SUPER_t;
                friend class cODBCDatabase;
 
        private:
                cStringF m_sCommand;    
 
        public:
                cODBCStatement(cODBCBase* pParent, const cSQLCmd& oCmd)
                        : cODBCBase(SQL_HANDLE_STMT, pParent)
                        , m_sCommand(oCmd.get_Command())
                {
                        // use SQLPrepare ??
                }
 
                virtual bool SetSQLAttr(long nAttribute, void* pValue) override;
 
                virtual HRESULT CloseStatement(bool bDrop) override;
 
                virtual HRESULT ExecuteAndFetch(bool bResultsExpected) override;
 
                virtual void ExecCancel() override;
 
                virtual HRESULT FetchNext(bool bMore = false) override;
 
                virtual ITERATE_t ReadColumnCount() override;
 
                virtual bool ReadColumnInfo(SQL_COLUMN_t iColumn, OUT cSQLColumnInfo& info) override;
 
                void GetData_Raw(SQL_COLUMN_t iColumn, SQL_TYPE_t nType, OUT void* pBuffer, size_t nSizeBuffer, OUT INT_PTR* pnSizeRet);
 
                bool GetColumnVarNumber(SQL_COLUMN_t iColumn, cSQLColumnInfo* pColInfo, OUT cVariant& vValRet, SQL_TYPE_t nCType);
 
                bool GetColumnVarTime(SQL_COLUMN_t iColumn, OUT cVariant& vValRet);
 
                bool GetColumnVarText(SQL_COLUMN_t iColumn, OUT cVariant& vValRet);
 
                bool GetColumnVarBlob(SQL_COLUMN_t iColumn, cSQLColumnInfo* pColInfo, OUT cVariant& vValRet);
 
                static SQL_TYPE_t GRAYCALL UpdateColType(cSQLColumnInfo* pColInfo);
 
                virtual bool GetColumnVar(SQL_COLUMN_t iColumn, cSQLColumnInfo* pColInfo, OUT cVariant& vValRet);
 
                IUNKNOWN_DISAMBIG(SUPER_t);
                cODBCBase_DISAMBIG;
        };
 
        //***********************************************************
 
        class GRAYLIB_LINK cODBCDatabase : public cODBCBase, public ISQLDatabase
        {
                typedef cODBCBase SUPER_t;
        public:
                cODBCDatabase(cODBCBase* pParent) : cODBCBase(SQL_HANDLE_DBC, pParent)
                {
                }
 
                virtual bool SetSQLAttr(long nAttribute, void* pValue) override;
 
                virtual HRESULT OpenConnect(const cSQLConnectionString& rConnectionString, INT_PTR hParentWnd) override;
 
                bool GetInfo(SQLUSMALLINT nInfoType, void* pValue, OUT SQLSMALLINT* pnValueLength);
 
                bool GetInfoStr(SQLUSMALLINT nInfoType, OUT cStringA& sVal)
                {
 
                        FILECHAR_t szBuffer[StrT::k_LEN_MAX];
                        SQLSMALLINT nStringLength = STRMAX(szBuffer);
 
                        if (!GetInfo(nInfoType, szBuffer, &nStringLength))
                        {
                                sVal.Empty();
                                return false;
                        }
 
                        szBuffer[STRMAX(szBuffer)] = '\0';              // just in case
                        sVal = (char*)szBuffer;
                        return true;
                }
 
                virtual HRESULT ReadConfigDB(OUT cSQLConfigDB& cfg) override;
 
                virtual bool Disconnect() override;
 
                virtual HRESULT EndTransaction(bool bCommit) override;
 
                virtual ISQLStatement* AllocISQLStatement(const cSQLCmd& oCmd) override
                {
                        return new cODBCStatement(this, oCmd);
                }
 
                IUNKNOWN_DISAMBIG(SUPER_t);
                cODBCBase_DISAMBIG;
        };
 
        //***********************************************************
 
        class GRAYLIB_LINK cODBCEnv : public cODBCBase, public cSingleton<cODBCEnv>, public ISQLDriver
        {
 
                typedef cODBCBase SUPER_t;
                CHEAPOBJECT_IMPL;
 
        public:
                cODBCEnv();
 
                virtual bool SetSQLAttr(long nAttribute, void* pValue) override;
 
                virtual ISQLDatabase* AllocISQLDatabase(const cSQLConnectionString& rConnectionString) override
                {
                        UNREFERENCED_REFERENCE(rConnectionString);      // don't need this now. use it later.
                        return new cODBCDatabase(this);
                }
 
                cArrayString<FILECHAR_t> GetInstalledDrivers();
 
                IUNKNOWN_DISAMBIG(SUPER_t);
                cODBCBase_DISAMBIG;
        };
}
#endif
#endif