diff options
Diffstat (limited to 'src/shared/Database/Database.h')
| -rw-r--r-- | src/shared/Database/Database.h | 113 |
1 files changed, 113 insertions, 0 deletions
diff --git a/src/shared/Database/Database.h b/src/shared/Database/Database.h new file mode 100644 index 00000000000..40097773430 --- /dev/null +++ b/src/shared/Database/Database.h @@ -0,0 +1,113 @@ +/* + * Copyright (C) 2005-2008 MaNGOS <http://www.mangosproject.org/> + * + * This program is free software; you can redistribute it and/or modify + * it under the terms of the GNU General Public License as published by + * the Free Software Foundation; either version 2 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA + */ + +#ifndef DATABASE_H +#define DATABASE_H + +#include "zthread/Thread.h" +#include "../src/zthread/ThreadImpl.h" +#include "Utilities/HashMap.h" +#include "Database/SqlDelayThread.h" + +class SqlTransaction; +class SqlResultQueue; +class SqlQueryHolder; + +typedef HM_NAMESPACE::hash_map<ZThread::ThreadImpl*, SqlTransaction*> TransactionQueues; +typedef HM_NAMESPACE::hash_map<ZThread::ThreadImpl*, SqlResultQueue*> QueryQueues; + +#define MAX_QUERY_LEN 1024 + +class MANGOS_DLL_SPEC Database +{ + protected: + Database() : m_threadBody(NULL), m_delayThread(NULL) {}; + + TransactionQueues m_tranQueues; ///< Transaction queues from diff. threads + QueryQueues m_queryQueues; ///< Query queues from diff threads + SqlDelayThread* m_threadBody; ///< Pointer to delay sql executer + ZThread::Thread* m_delayThread; ///< Pointer to executer thread + + public: + + virtual ~Database(); + + virtual bool Initialize(const char *infoString); + virtual void InitDelayThread() = 0; + virtual void HaltDelayThread() = 0; + + virtual QueryResult* Query(const char *sql) = 0; + QueryResult* PQuery(const char *format,...) ATTR_PRINTF(2,3); + + /// Async queries and query holders, implemented in DatabaseImpl.h + template<class Class> + bool AsyncQuery(Class *object, void (Class::*method)(QueryResult*), const char *sql); + template<class Class, typename ParamType1> + bool AsyncQuery(Class *object, void (Class::*method)(QueryResult*, ParamType1), ParamType1 param1, const char *sql); + template<typename ParamType1> + bool AsyncQuery(void (*method)(QueryResult*, ParamType1), ParamType1 param1, const char *sql); + template<class Class> + bool AsyncPQuery(Class *object, void (Class::*method)(QueryResult*), const char *format,...) ATTR_PRINTF(4,5); + template<class Class, typename ParamType1> + bool AsyncPQuery(Class *object, void (Class::*method)(QueryResult*, ParamType1), ParamType1 param1, const char *format,...) ATTR_PRINTF(5,6); + template<typename ParamType1> + bool AsyncPQuery(void (*method)(QueryResult*, ParamType1), ParamType1 param1, const char *format,...) ATTR_PRINTF(5,6); + template<class Class> + bool DelayQueryHolder(Class *object, void (Class::*method)(QueryResult*, SqlQueryHolder*), SqlQueryHolder *holder); + template<class Class, typename ParamType1> + bool DelayQueryHolder(Class *object, void (Class::*method)(QueryResult*, SqlQueryHolder*, ParamType1), SqlQueryHolder *holder, ParamType1 param1); + + virtual bool Execute(const char *sql) = 0; + bool PExecute(const char *format,...) ATTR_PRINTF(2,3); + virtual bool DirectExecute(const char* sql) = 0; + bool DirectPExecute(const char *format,...) ATTR_PRINTF(2,3); + + // Writes SQL commands to a LOG file (see mangosd.conf "LogSQL") + bool PExecuteLog(const char *format,...) ATTR_PRINTF(2,3); + + virtual bool BeginTransaction() // nothing do if DB not support transactions + { + return true; + } + virtual bool CommitTransaction() // nothing do if DB not support transactions + { + return true; + } + virtual bool RollbackTransaction() // can't rollback without transaction support + { + return false; + } + + virtual operator bool () const = 0; + + virtual unsigned long escape_string(char *to, const char *from, unsigned long length) { strncpy(to,from,length); return length; } + void escape_string(std::string& str); + + // must be called before first query in thread (one time for thread using one from existed Database objects) + virtual void ThreadStart(); + // must be called before finish thread run (one time for thread using one from existed Database objects) + virtual void ThreadEnd(); + + // sets the result queue of the current thread, be careful what thread you call this from + void SetResultQueue(SqlResultQueue * queue); + + private: + bool m_logSQL; + std::string m_logsDir; +}; +#endif |