164 lines
4.7 KiB
C++
164 lines
4.7 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
//
|
|
// Purpose:
|
|
//
|
|
// $NoKeywords: $
|
|
//=============================================================================
|
|
|
|
#ifndef GCSQLQUERY_H
|
|
#define GCSQLQUERY_H
|
|
#ifdef _WIN32
|
|
#pragma once
|
|
#endif
|
|
|
|
#include "gamecoordinator/igcsqlquery.h"
|
|
#include "refcount.h"
|
|
#include "bufferpool.h"
|
|
|
|
namespace GCSDK
|
|
{
|
|
|
|
struct GCSQLBindParam_t
|
|
{
|
|
EGCSQLType m_eType;
|
|
size_t m_nOffset;
|
|
size_t m_cubData;
|
|
};
|
|
|
|
|
|
class CGCSQLQuery
|
|
{
|
|
DECLARE_CLASS_MEMPOOL( CGCSQLQuery );
|
|
public:
|
|
CGCSQLQuery();
|
|
virtual ~CGCSQLQuery();
|
|
|
|
static CBufferPool &GetBufferPool();
|
|
|
|
void SetCommand( const char *pchCommand ) { m_sCommand = pchCommand; }
|
|
|
|
void AddBindParam( const char *pchValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_String, (byte *)pchValue, Q_strlen( pchValue ) );
|
|
}
|
|
|
|
void AddBindParam( const int16 nValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_int16, (byte *)&nValue, sizeof( nValue ) );
|
|
}
|
|
|
|
void AddBindParam( const uint16 uValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_int16, (byte *)&uValue, sizeof( uValue ) );
|
|
}
|
|
|
|
void AddBindParam( const int32 nValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_int32, (byte *)&nValue, sizeof( nValue ) );
|
|
}
|
|
|
|
void AddBindParam( const uint32 uValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_int32, (byte *)&uValue, sizeof( uValue ) );
|
|
}
|
|
|
|
void AddBindParam( const uint64 ulValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_int64, (byte *)&ulValue, sizeof( ulValue ) );
|
|
}
|
|
|
|
void AddBindParam( const uint8 *ubValue, const int cubValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_Blob, (byte *)ubValue, cubValue );
|
|
}
|
|
|
|
void AddBindParam( const float fValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_float, (byte *)&fValue, sizeof ( fValue ) );
|
|
}
|
|
|
|
void AddBindParam( const double dValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_double, (byte *)&dValue, sizeof ( dValue ) );
|
|
}
|
|
|
|
// Image needs a special type since the default var data is blob
|
|
void AddBindParamImage( const uint8 *ubValue, const int cubValue )
|
|
{
|
|
AddBindParamRaw( k_EGCSQLType_Image, (byte *)ubValue, cubValue );
|
|
}
|
|
|
|
void ClearParams();
|
|
|
|
// this is used internally to bind a field with its type. You probably want
|
|
// some version of AddBindParam instead of this.
|
|
void AddBindParamRaw( EGCSQLType eType, const byte *pubData, uint32 cubData );
|
|
|
|
// ------- Interface implementation from IGCSQLQuery -----
|
|
|
|
// get the null-terminated query string itself
|
|
virtual const char *PchCommand() { return m_sCommand.Get(); }
|
|
|
|
// gets the parameter data
|
|
virtual uint32 CnParams() { return m_vecParams.Count(); }
|
|
virtual EGCSQLType EParamType( uint32 uIndex ) { return m_vecParams[uIndex].m_eType; }
|
|
virtual byte *PubParam( uint32 uIndex ) { return (byte *)m_pBufParams->Base() + m_vecParams[uIndex].m_nOffset; }
|
|
virtual uint32 CubParam( uint32 uIndex ) { return ( uint32 )m_vecParams[uIndex].m_cubData; }
|
|
|
|
private:
|
|
CUtlString m_sCommand;
|
|
CUtlBuffer *m_pBufParams;
|
|
CUtlVectorFixedGrowable< GCSQLBindParam_t, 10 > m_vecParams;
|
|
};
|
|
|
|
class CGCSQLQueryGroup : public IGCSQLQuery, public CRefCount
|
|
{
|
|
DECLARE_CLASS_MEMPOOL( CGCSQLQuery );
|
|
|
|
private:
|
|
// create query groups on the heap with alloc
|
|
CGCSQLQueryGroup();
|
|
|
|
// destroy query groups by releasing them
|
|
virtual ~CGCSQLQueryGroup();
|
|
public:
|
|
static CGCSQLQueryGroup *Alloc() { return new CGCSQLQueryGroup(); }
|
|
|
|
void AddQuery( CGCSQLQuery *pQuery );
|
|
void SetName( const char *sName );
|
|
|
|
// returns the number of statements in the transaction
|
|
// represented by this query object
|
|
virtual uint32 GetStatementCount() { return m_vecQueries.Count(); }
|
|
|
|
// returns a string that represents where in the GC this
|
|
// query came from. Usually this is FILE_AND_LINE.
|
|
virtual const char *PchName() { return m_sName; }
|
|
|
|
// get the null-terminated query string itself
|
|
virtual const char *PchCommand( uint32 unStatement ) { return m_vecQueries[unStatement]->PchCommand(); }
|
|
|
|
// gets the parameter data
|
|
virtual uint32 CnParams( uint32 unStatement ) { return m_vecQueries[unStatement]->CnParams(); }
|
|
virtual EGCSQLType EParamType( uint32 unStatement, uint32 uIndex ) { return m_vecQueries[unStatement]->EParamType( uIndex ); }
|
|
virtual byte *PubParam( uint32 unStatement, uint32 uIndex ) { return m_vecQueries[unStatement]->PubParam( uIndex ); }
|
|
virtual uint32 CubParam( uint32 unStatement, uint32 uIndex ) { return m_vecQueries[unStatement]->CubParam( uIndex ); };
|
|
|
|
// reports the result
|
|
virtual void SetResults( IGCSQLResultSetList *pResults );
|
|
IGCSQLResultSetList *GetResults() { return m_pResults; }
|
|
|
|
// clears all the queries in the query group and resets its name
|
|
void Clear();
|
|
private:
|
|
CUtlVector< CGCSQLQuery * > m_vecQueries;
|
|
CUtlString m_sName;
|
|
IGCSQLResultSetList *m_pResults;
|
|
};
|
|
|
|
} // namespace GCSDK
|
|
|
|
#include "tier0/memdbgoff.h"
|
|
|
|
#endif // GCSQLQUERY_H
|