50 lines
1.3 KiB
C++
50 lines
1.3 KiB
C++
#ifndef _INCLUDE_DATABASE_MANAGER_H_
|
|
#define _INCLUDE_DATABASE_MANAGER_H_
|
|
|
|
#include <IDBDriver.h>
|
|
#include <IHandleSys.h>
|
|
#include "sm_globals.h"
|
|
#include <sh_vector.h>
|
|
|
|
using namespace SourceHook;
|
|
|
|
struct db_driver
|
|
{
|
|
IDBDriver *driver;
|
|
IdentityToken_t *ident;
|
|
HandleType_t htDatabase;
|
|
HandleType_t htQuery;
|
|
Handle_t hDriver;
|
|
};
|
|
|
|
class DBManager :
|
|
public IDBManager,
|
|
public SMGlobalClass,
|
|
public IHandleTypeDispatch
|
|
{
|
|
const char *GetInterfaceName();
|
|
unsigned int GetInterfaceVersion();
|
|
public: //SMGlobalClass
|
|
void OnSourceModAllInitialized();
|
|
public: //IHandleTypeDispatch
|
|
void OnHandleDestroy(HandleType_t type, void *object);
|
|
public: //IDBManager
|
|
void AddDriver(IDBDriver *pDriver, IdentityToken_t *ident);
|
|
void RemoveDriver(IDBDriver *pDriver);
|
|
const DatabaseInfo *FindDatabaseConf(const char *name);
|
|
bool Connect(const char *name, IDBDriver **pdr, IDatabase **pdb, bool persistent);
|
|
unsigned int GetDriverCount();
|
|
IDBDriver *GetDriver(unsigned int index, IdentityToken_t **pToken=NULL);
|
|
public:
|
|
db_driver *GetDriverInfo(unsigned int index);
|
|
private:
|
|
CVector<db_driver *> m_drivers;
|
|
HandleType_t m_DriverType;
|
|
HandleType_t m_DatabaseType;
|
|
HandleType_t m_QueryType;
|
|
};
|
|
|
|
extern DBManager g_DBMan;
|
|
|
|
#endif //_INCLUDE_DATABASE_MANAGER_H_
|