1HEADmaster

1 files changed, 163 insertions, 0 deletions

diff --git a/public/gcsdk/gcsqlquery.h b/public/gcsdk/gcsqlquery.h
new file mode 100644
index 0000000..4a6fb1b
--- /dev/null
+++ b/public/gcsdk/gcsqlquery.h
@@ -0,0 +1,163 @@
+//========= 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