src/Common/ODBC/Handle.h

Go to the documentation of this file.

#pragma once

namespace AppSecInc
{
        namespace Databases
        {
                namespace ODBC
                {
                        typedef struct
                        {
                                std::wstring state;
                                std::wstring message;
                                int native_error;
                        } ODBCError;

                        typedef struct
                        {
                                std::wstring vendor;
                                std::wstring component;
                                std::wstring datasource;
                                std::wstring text;
                        } ODBCDiagnosticsMessage;

                        class ODBCHandle
                        {
                        protected:
                SQLSMALLINT _type;
                SQLHANDLE _handle;
                SQLHANDLE _hconn;
            private:
                void operator=(const ODBCHandle&);
                        public:
                inline SQLHANDLE Peek() const { return _handle; }
                                inline SQLHANDLE PeekConnection() const { return _hconn; }
                                ODBCHandle();
                                virtual ~ODBCHandle();
                                void Allocate(SQLSMALLINT type, SQLHANDLE hconn);
                void Release();
                static std::wstring GetError(SQLHANDLE handle, int type);
                                static std::vector<ODBCError> GetErrors(SQLHANDLE handle, int type);
                                static std::vector<std::wstring> GetDiagFields(SQLHANDLE handle, int type, int diagidentifier);
                                static std::vector<ODBCDiagnosticsMessage> GetDiagMessages(SQLHANDLE handle, int type);
                std::wstring GetError() const { return GetError(_handle, _type); }
                                std::vector<ODBCError> GetErrors() const { return GetErrors(_handle, _type); }
                                std::vector<std::wstring> GetDiagFields(int diagidentifier) const { return GetDiagFields(_handle, _type, diagidentifier); }
                                std::vector<ODBCDiagnosticsMessage> GetDiagMessages() const { return GetDiagMessages(_handle, _type); }
                        };
                }
        }
}