107 lines
3.3 KiB
C++
107 lines
3.3 KiB
C++
//========= Copyright Valve Corporation, All rights reserved. ============//
|
|
//
|
|
// Purpose:
|
|
//
|
|
// $NoKeywords: $
|
|
//=============================================================================//
|
|
|
|
#ifndef MYSQL_WRAPPER_H
|
|
#define MYSQL_WRAPPER_H
|
|
#ifdef _WIN32
|
|
#pragma once
|
|
#endif
|
|
|
|
#include "interface.h"
|
|
#include "utlvector.h"
|
|
|
|
class IMySQLRowSet;
|
|
|
|
class CColumnValue {
|
|
public:
|
|
CColumnValue(IMySQLRowSet *pSQL, int iColumn);
|
|
|
|
const char *String();
|
|
long Int32();
|
|
|
|
private:
|
|
IMySQLRowSet *m_pSQL;
|
|
int m_iColumn;
|
|
};
|
|
|
|
class IMySQLRowSet {
|
|
public:
|
|
virtual void Release() = 0;
|
|
|
|
// 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;
|
|
|
|
virtual const char *GetColumnValue_String(int iColumn) = 0;
|
|
virtual long GetColumnValue_Int(int iColumn) = 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;
|
|
};
|
|
|
|
class IMySQL : public IMySQLRowSet {
|
|
public:
|
|
virtual bool InitMySQL(const char *pDBName, const char *pHostName = "",
|
|
const char *pUserName = "",
|
|
const char *pPassword = "") = 0;
|
|
virtual void Release() = 0;
|
|
|
|
// These execute SQL commands. They return 0 if the query was successful.
|
|
virtual int Execute(const char *pString) = 0;
|
|
|
|
// This reads in all of the data in the last row set you queried with
|
|
// Execute and builds a separate copy. This is useful in some of the VMPI
|
|
// tools to have a thread repeatedly execute a slow query, then store off
|
|
// the results for the main thread to parse.
|
|
virtual IMySQLRowSet *DuplicateRowSet() = 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;
|
|
|
|
// Returns the last error message, if an error took place
|
|
virtual const char *GetLastError() = 0;
|
|
};
|
|
|
|
#define MYSQL_WRAPPER_VERSION_NAME "MySQLWrapper001"
|
|
|
|
// ------------------------------------------------------------------------------------------------
|
|
// // Inlines.
|
|
// ------------------------------------------------------------------------------------------------
|
|
// //
|
|
|
|
inline CColumnValue::CColumnValue(IMySQLRowSet *pSQL, int iColumn) {
|
|
m_pSQL = pSQL;
|
|
m_iColumn = iColumn;
|
|
}
|
|
|
|
inline const char *CColumnValue::String() {
|
|
return m_pSQL->GetColumnValue_String(m_iColumn);
|
|
}
|
|
|
|
inline long CColumnValue::Int32() {
|
|
return m_pSQL->GetColumnValue_Int(m_iColumn);
|
|
}
|
|
|
|
#endif // MYSQL_WRAPPER_H
|