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.

105 lines
2.7 KiB

5 years ago
//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose:
//
// $NoKeywords: $
//=============================================================================//
#ifndef MYSQL_WRAPPER_H
#define MYSQL_WRAPPER_H
#ifdef _WIN32
#pragma once
#endif
#include "utlvector.h"
class IMySQL;
// This is a helper class to build queries like the stream IO.
class CMySQLQuery
{
friend class CMySQL;
public:
// This is like a sprintf, but it will grow the string as necessary.
void Format( PRINTF_FORMAT_STRING const char *pFormat, ... );
private:
CUtlVector<char> m_QueryText;
};
class CColumnValue
{
public:
CColumnValue( CMySQL *pSQL, int iColumn );
const char* String();
long Int32();
unsigned long UInt32();
private:
CMySQL *m_pSQL;
int m_iColumn;
};
class IMySQL
{
public:
virtual void Release() = 0;
// These execute SQL commands. They return 0 if the query was successful.
virtual int Execute( const char *pString ) = 0;
virtual int Execute( CMySQLQuery &query ) = 0;
// If you just inserted rows into a table with an AUTO_INCREMENT column,
// then this returns the (unique) value of that column.
virtual unsigned long InsertID() = 0;
// If you just executed a select statement, then you can use these functions to
// iterate over the result set.
// Get the number of columns in the data returned from the last query (if it was a select statement).
virtual int NumFields() = 0;
// Get the name of each column returned by the last query.
virtual const char* GetFieldName( int iColumn ) = 0;
// Call this in a loop until it returns false to iterate over all rows the query returned.
virtual bool NextRow() = 0;
// You can call this to start iterating over the result set from the start again.
// Note: after calling this, you have to call NextRow() to actually get the first row's value ready.
virtual bool SeekToFirstRow() = 0;
virtual CColumnValue GetColumnValue( int iColumn ) = 0;
virtual CColumnValue GetColumnValue( const char *pColumnName ) = 0;
// You can call this to get the index of a column for faster lookups with GetColumnValue( int ).
// Returns -1 if the column can't be found.
virtual int GetColumnIndex( const char *pColumnName ) = 0;
};
IMySQL* InitMySQL( const char *pDBName, const char *pHostName="", const char *pUserName="", const char *pPassword="" );
// ------------------------------------------------------------------------------------------------ //
// Inlines.
// ------------------------------------------------------------------------------------------------ //
inline CColumnValue::CColumnValue( CMySQL *pSQL, int iColumn )
{
m_pSQL = pSQL;
m_iColumn = iColumn;
}
#endif // MYSQL_WRAPPER_H