1

utils/sqlwrapper/sqlhelpers.h

@@ -0,0 +1,162 @@
+//========= Copyright Valve Corporation, All rights reserved. ============//
+//
+// Purpose:
+//
+// $NoKeywords: $
+//=============================================================================
+#include "utlvector.h"
+
+extern "C"
+{
+	#include <WinSock.H> 
+	#include "mysql.h"
+};
+#include "utlvector.h"
+
+typedef enum_field_types ESQLFieldType;
+
+//-----------------------------------------------------------------------------
+// Purpose: helper interface for running queries internal to this dll
+//-----------------------------------------------------------------------------
+class ISQLHelper
+{
+public:
+	// run a sql query on the db
+	virtual bool BInternalQuery( const char *pchQueryString, MYSQL_RES **ppMySQLRes, bool bRecurse = true ) = 0;
+};
+
+
+//-----------------------------------------------------------------------------
+// Purpose: represents the data about a single SQL column
+//-----------------------------------------------------------------------------
+class CSQLColumn 
+{
+public:
+	CSQLColumn() { m_rgchName[0] = 0; }
+	CSQLColumn( const char *pchName, ESQLFieldType eSQLFieldType );
+	virtual const char *PchColumnName() const { return m_rgchName; };
+	virtual EColumnType GetEColumnType() const { return m_EColumnType; };
+	virtual ESQLFieldType GetESQLFieldType() const { return m_ESQLFieldType; };
+
+private:
+	char m_rgchName[64];
+	EColumnType m_EColumnType;
+	ESQLFieldType m_ESQLFieldType;
+};
+
+
+//-----------------------------------------------------------------------------
+// Purpose: encapsulates a table's description
+//-----------------------------------------------------------------------------
+class CSQLTable : public ISQLTable
+{
+public:
+	CSQLTable( const char *pchName );
+	CSQLTable( CSQLTable const &CSQLTable );  // copy constructor
+	void AddColumn( const char *pchName, ESQLFieldType ESQLFieldType );
+	const char *PchName() const { return m_rgchName; };
+	virtual int GetCSQLColumn() const { return m_VecSQLColumn.Count(); };
+	virtual const char *PchColumnName( int iSQLColumn ) const { return m_VecSQLColumn[iSQLColumn].PchColumnName(); };
+	virtual EColumnType GetEColumnType( int iSQLColumn ) const { return m_VecSQLColumn[iSQLColumn].GetEColumnType(); };
+	virtual ESQLFieldType GetESQLFieldType( int iSQLColumn ) const { return m_VecSQLColumn[iSQLColumn].GetESQLFieldType(); };
+
+	void Reset() { m_VecSQLColumn.RemoveAll(); }
+
+#ifdef DBGFLAG_VALIDATE
+	void Validate( CValidator &validator, char *pchName );
+#endif
+
+private:
+	char m_rgchName[64];
+	CUtlVector<CSQLColumn> m_VecSQLColumn;
+};
+
+
+//-----------------------------------------------------------------------------
+// Purpose: encapsulates a db's worth of tables
+//-----------------------------------------------------------------------------
+class CSQLTableSet : public ISQLTableSet
+{
+public:
+	CSQLTableSet() { m_bInit = false; }
+	bool Init( ISQLHelper *pISQLHelper  );
+	bool BInit() { return m_bInit; }
+	virtual int GetCSQLTable() const { return m_VecSQLTable.Count(); };
+	virtual const ISQLTable *PSQLTable( int iSQLTable ) const;
+	
+#ifdef DBGFLAG_VALIDATE
+	void Validate( CValidator &validator, char *pchName );
+#endif
+
+private:
+	static EColumnType EColumnTypeFromPchName( const char *pchName );
+	CUtlVector<CSQLTable> m_VecSQLTable;
+	bool m_bInit;
+};
+
+
+//-----------------------------------------------------------------------------
+// Purpose: describes a single row in a result set
+//-----------------------------------------------------------------------------
+class CSQLRow : public ISQLRow
+{
+public:
+	CSQLRow( MYSQL_ROW *pMySQLRow, const ISQLTable *pSQLTableDescription, int cSQLRowData );
+	~CSQLRow();
+	virtual int GetCSQLRowData() const { return m_VecSQLRowData.Count(); };
+	virtual const char *PchData( int iSQLRowData ) const;
+	virtual int NData( int iSQLRowData ) const;
+	virtual uint64 UlData( int iSQLRowData ) const;
+	virtual float FlData( int iSQLRowData ) const;
+	virtual uint64 UlTime( int iSQLRowData ) const;
+	virtual bool BData( int iSQLRowData ) const;
+	virtual EColumnType GetEColumnType( int iSQLRowData ) const;
+
+#ifdef DBGFLAG_VALIDATE
+	void Validate( CValidator &validator, char *pchName );
+#endif
+
+private:
+	struct SQLRowData_s
+	{
+		union
+		{
+			int nData;
+			float flData;
+			const char *pchData;
+			uint64 ulTime;
+			uint64 ulData;
+		} data;
+		EColumnType eColumnType;
+	};
+
+	CUtlVector<struct SQLRowData_s> m_VecSQLRowData;
+	CUtlVector<char *> m_VecPchStoredStrings;
+};
+
+
+//-----------------------------------------------------------------------------
+// Purpose: encapsulates a result set from a SQL query
+//-----------------------------------------------------------------------------
+class CResultSet : public IResultSet
+{
+public:
+	CResultSet();
+	~CResultSet();
+	bool Query( const char *pchQuery, ISQLHelper *pISQLHelper );
+	virtual int GetCSQLRow() const;
+	virtual const ISQLRow *PSQLRowNextResult();
+	void FreeResult();
+
+#ifdef DBGFLAG_VALIDATE
+	void Validate( CValidator &validator, char *pchName );
+#endif
+
+private:
+	MYSQL_RES *m_MySQLRes;
+	CSQLRow *m_pSQLRow;
+	CSQLTable m_SQLTableDescription;
+	int m_cSQLField;
+	int m_cSQLRow;
+};
+