Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

163 lines
4.7 KiB

//========= 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