mirror of
https://github.com/alliedmodders/amxmodx.git
synced 2024-10-17 15:56:52 +03:00
106 lines
2.6 KiB
C++
106 lines
2.6 KiB
C++
// vim: set ts=4 sw=4 tw=99 noet:
|
|
//
|
|
// AMX Mod X, based on AMX Mod by Aleksander Naszko ("OLO").
|
|
// Copyright (C) The AMX Mod X Development Team.
|
|
//
|
|
// This software is licensed under the GNU General Public License, version 3 or higher.
|
|
// Additional exceptions apply. For full license details, see LICENSE.txt or visit:
|
|
// https://alliedmods.net/amxmodx-license
|
|
|
|
//
|
|
// MySQL Module
|
|
//
|
|
|
|
#ifndef _INCLUDE_MYSQL_THREADING_H
|
|
#define _INCLUDE_MYSQL_THREADING_H
|
|
|
|
#include "IThreader.h"
|
|
#include "ISQLDriver.h"
|
|
#include <am-string.h>
|
|
#include <sh_stack.h>
|
|
|
|
struct QueuedResultInfo
|
|
{
|
|
AmxQueryInfo amxinfo;
|
|
bool connect_success;
|
|
bool query_success;
|
|
float queue_time;
|
|
};
|
|
|
|
class AtomicResult :
|
|
public IResultSet,
|
|
public IResultRow
|
|
{
|
|
friend class MysqlThread;
|
|
public:
|
|
AtomicResult();
|
|
~AtomicResult();
|
|
public:
|
|
//free the handle if necessary (see IQuery).
|
|
virtual void FreeHandle();
|
|
virtual unsigned int RowCount();
|
|
virtual unsigned int FieldCount();
|
|
virtual const char *FieldNumToName(unsigned int num);
|
|
virtual bool FieldNameToNum(const char *name, unsigned int *columnId);
|
|
virtual bool IsDone();
|
|
virtual IResultRow *GetRow();
|
|
virtual void NextRow();
|
|
virtual void Rewind();
|
|
virtual bool NextResultSet();
|
|
public:
|
|
virtual const char *GetString(unsigned int columnId);
|
|
virtual const char *GetStringSafe(unsigned int columnId);
|
|
virtual double GetDouble(unsigned int columnId);
|
|
virtual float GetFloat(unsigned int columnId);
|
|
virtual int GetInt(unsigned int columnId);
|
|
virtual bool IsNull(unsigned int columnId);
|
|
virtual const char *GetRaw(unsigned int columnId, size_t *length);
|
|
public:
|
|
void CopyFrom(IResultSet *rs);
|
|
private:
|
|
void _InternalClear();
|
|
private:
|
|
unsigned int m_RowCount;
|
|
unsigned int m_FieldCount;
|
|
size_t m_AllocSize;
|
|
ke::AString **m_Table;
|
|
unsigned int m_CurRow;
|
|
bool m_IsFree;
|
|
};
|
|
|
|
class MysqlThread : public IThread
|
|
{
|
|
public:
|
|
MysqlThread();
|
|
~MysqlThread();
|
|
public:
|
|
void SetInfo(const char *host, const char *user, const char *pass, const char *db, int port, unsigned int max_timeout);
|
|
void SetCharacterSet(const char *charset);
|
|
void SetQuery(const char *query);
|
|
void SetCellData(cell data[], ucell len);
|
|
void SetForward(int forward);
|
|
void Invalidate();
|
|
void Execute();
|
|
public:
|
|
void RunThread(IThreadHandle *pHandle);
|
|
void OnTerminate(IThreadHandle *pHandle, bool cancel);
|
|
private:
|
|
ke::AString m_query;
|
|
ke::AString m_host;
|
|
ke::AString m_user;
|
|
ke::AString m_pass;
|
|
ke::AString m_db;
|
|
unsigned int m_max_timeout;
|
|
ke::AString m_charset;
|
|
int m_port;
|
|
cell *m_data;
|
|
ucell m_datalen;
|
|
size_t m_maxdatalen;
|
|
int m_fwd;
|
|
QueuedResultInfo m_qrInfo;
|
|
AtomicResult m_atomicResult;
|
|
};
|
|
|
|
#endif //_INCLUDE_MYSQL_THREADING_H
|
|
|