cSQLStatement.h

Go to the documentation of this file.

//
//
 
#ifndef _INC_cSQLStatement_H
#define _INC_cSQLStatement_H
#ifndef NO_PRAGMA_ONCE
#pragma once
#endif
 
#include "cSQLBase.h"
#include "../Variant/cVariant.h"
#include "../Scriptable/IScriptableObj.h"
#include "GrayCore/include/cArray.h"
#include "GrayCore/include/cFilePath.h"
#ifdef USE_ISCRIPTABLE
#include "../ScriptContext/cScriptSourceCurrent.h"
#endif
 
namespace GrayLib
{
        class GRAYLIB_LINK cSQLDatabase;
        typedef cRefPtr<cSQLDatabase> cSQLDatabasePtr;
        class GRAYLIB_LINK cSQLCmd;
        DECLARE_INTERFACE(ISQLStatement);
 
        class GRAYLIB_LINK cSQLStatement
                : public cSQLBase
                , public IScriptableEnumerator
        {
 
                friend class cSQLDatabase;
 
        public:
                cStringA m_sFromTable;          
 
        protected:
                cSQLDatabasePtr m_pDatabase;    
                cArrayStruct<cSQLColumnInfo> m_aColumnInfo;     
 
        private:
                static const TIMESYSD_t k_FetchWait = 500;      
 
        protected:
                ISQLStatement* get_ISQLStatement() const;
 
                ITERATE_t ReadColumnCount();
                bool ReadColumnInfo(SQL_COLUMN_t iColumn, OUT cSQLColumnInfo& info); 
                bool AllocISQLStatement(const cSQLCmd& oCmd);
 
        public:
                cSQLStatement();
                virtual ~cSQLStatement();
 
                bool isConnected() const;
 
                virtual cSQLDriver* get_SQLDriver() const;
                virtual cSQLBase* get_SQLParent() const;
 
                virtual HRESULT ExecuteAndFetch(cSQLDatabase* pDatabase, cSQLCmd& oCmd, cExceptionHolder* pExceptionRet = nullptr);
                virtual HRESULT FetchNext(bool bWaitIfBusy = false);
 
                void ExecCancel();
                bool CloseStatement(bool bDrop = true, bool bThrowOnError = false);
                bool Close();
 
                SQL_COLUMN_t GetColumnCount();
                cSQLColumnInfo* GetColumnInfo(SQL_COLUMN_t iColumn = k_SQL_COL_FIRST);
                SQL_COLUMN_t FindColumn(const char* pszFieldName);
                SQL_COLUMN_t VerifyColumn(SQL_COLUMN_t iColumn, const char* pszFieldName);
 
                bool GetColumnString(SQL_COLUMN_t iColumn, OUT cStringA& sResult);
                cStringA GetColumnStr(SQL_COLUMN_t iColumn = k_SQL_COL_FIRST);
                int GetColumnInt(SQL_COLUMN_t iColumn = k_SQL_COL_FIRST);
                virtual bool GetColumnVar(SQL_COLUMN_t iColumn, OUT cVariant& vValRet);
 
                STDMETHOD(s_GetEnum)(ITERATE_t i, OUT cStringA& rsPropName, OUT cVariant& vValRet) override;
 
                cStringA GetColumnStr(const char* pszFieldName)
                {
                        return GetColumnStr(FindColumn(pszFieldName));
                }
                int GetColumnInt(const char* pszFieldName)
                {
                        return GetColumnInt(FindColumn(pszFieldName));
                }
                bool GetColumnVar(const char* pszFieldName, OUT cVariant& vVal)
                {
                        return GetColumnVar(FindColumn(pszFieldName), vVal);
                }
        };
 
        class GRAYLIB_LINK cSQLStatementQ
                : public cSQLStatement
                , public IScriptableGetter
        {
                typedef cSQLStatement SUPER_t;
 
        public:
                cArrayStruct<cVariant> m_aColumnData;   
 
        protected:
                int ReadRecordset();
        public:
                cSQLStatementQ();
                virtual ~cSQLStatementQ();
 
                SQL_TYPE_t SetColumnType(SQL_COLUMN_t iColumn, SQL_TYPE_t eType);
                SQL_TYPE_t RemoveColumn(SQL_COLUMN_t iColumn);
 
                virtual HRESULT ExecuteAndFetch(cSQLDatabase* pDatabase, cSQLCmd& oCmd, cExceptionHolder* pExceptionRet = nullptr) override;
                virtual HRESULT FetchNext(bool bWaitIfBusy = false) override;
                virtual bool GetColumnVar(SQL_COLUMN_t iColumn, OUT cVariant& vVal) override;
 
                bool GetColumnVar(const char* pszFieldName, OUT cVariant& vVal)
                {
                        return GetColumnVar(FindColumn(pszFieldName), vVal);
                }
 
                STDMETHOD(s_GetEnum)(ITERATE_t i, OUT cStringA& rsPropName, OUT cVariant& rvValRet) override;
                STDMETHOD(s_GetByName)(const ATOMCHAR_t* pszPropName, OUT cVariant& vValRet) override;
        };
 
#ifdef USE_ISCRIPTABLE
        class GRAYLIB_LINK cScriptSourceSQLStatement : public cScriptSourceCurrent, public cSQLStatementQ
        {
        public:
                cScriptSourceSQLStatement();
                ~cScriptSourceSQLStatement();
                virtual cStringA get_SourceDesc() const;
        };
#endif
};
#endif // !_INC_cSQLStatement_H