diff options
Diffstat (limited to 'src/shared/Database/SqlOperations.cpp')
| -rw-r--r-- | src/shared/Database/SqlOperations.cpp | 26 |
1 files changed, 26 insertions, 0 deletions
diff --git a/src/shared/Database/SqlOperations.cpp b/src/shared/Database/SqlOperations.cpp index 9ca698d733e..396f2e36bc2 100644 --- a/src/shared/Database/SqlOperations.cpp +++ b/src/shared/Database/SqlOperations.cpp @@ -17,16 +17,20 @@ * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ + #include "SqlOperations.h" #include "SqlDelayThread.h" #include "DatabaseEnv.h" #include "DatabaseImpl.h" + /// ---- ASYNC STATEMENTS / TRANSACTIONS ---- + void SqlStatement::Execute(Database *db) { /// just do it db->DirectExecute(m_sql); } + void SqlTransaction::Execute(Database *db) { if(m_queue.empty()) @@ -36,6 +40,7 @@ void SqlTransaction::Execute(Database *db) { char const *sql = m_queue.front(); m_queue.pop(); + if(!db->DirectExecute(sql)) { free((void*)const_cast<char*>(sql)); @@ -47,11 +52,14 @@ void SqlTransaction::Execute(Database *db) } return; } + free((void*)const_cast<char*>(sql)); } db->DirectExecute("COMMIT"); } + /// ---- ASYNC QUERIES ---- + void SqlQuery::Execute(Database *db) { if(!m_callback || !m_queue) @@ -61,6 +69,7 @@ void SqlQuery::Execute(Database *db) /// add the callback to the sql result queue of the thread it originated from m_queue->add(m_callback); } + void SqlResultQueue::Update() { /// execute the callbacks waiting in the synchronization queue @@ -71,16 +80,19 @@ void SqlResultQueue::Update() delete callback; } } + bool SqlQueryHolder::Execute(Trinity::IQueryCallback * callback, SqlDelayThread *thread, SqlResultQueue *queue) { if(!callback || !thread || !queue) return false; + /// delay the execution of the queries, sync them with the delay thread /// which will in turn resync on execution (via the queue) and call back SqlQueryHolderEx *holderEx = new SqlQueryHolderEx(this, callback, queue); thread->Delay(holderEx); return true; } + bool SqlQueryHolder::SetQuery(size_t index, const char *sql) { if(m_queries.size() <= index) @@ -88,16 +100,19 @@ bool SqlQueryHolder::SetQuery(size_t index, const char *sql) sLog.outError("Query index (%u) out of range (size: %u) for query: %s",index,(uint32)m_queries.size(),sql); return false; } + if(m_queries[index].first != NULL) { sLog.outError("Attempt assign query to holder index (%u) where other query stored (Old: [%s] New: [%s])", index,m_queries[index].first,sql); return false; } + /// not executed yet, just stored (it's not called a holder for nothing) m_queries[index] = SqlResultPair(strdup(sql), NULL); return true; } + bool SqlQueryHolder::SetPQuery(size_t index, const char *format, ...) { if(!format) @@ -105,18 +120,22 @@ bool SqlQueryHolder::SetPQuery(size_t index, const char *format, ...) sLog.outError("Query (index: %u) is empty.",index); return false; } + va_list ap; char szQuery [MAX_QUERY_LEN]; va_start(ap, format); int res = vsnprintf( szQuery, MAX_QUERY_LEN, format, ap ); va_end(ap); + if(res==-1) { sLog.outError("SQL Query truncated (and not execute) for format: %s",format); return false; } + return SetQuery(index,szQuery); } + QueryResult* SqlQueryHolder::GetResult(size_t index) { if(index < m_queries.size()) @@ -133,12 +152,14 @@ QueryResult* SqlQueryHolder::GetResult(size_t index) else return NULL; } + void SqlQueryHolder::SetResult(size_t index, QueryResult *result) { /// store the result in the holder if(index < m_queries.size()) m_queries[index].second = result; } + SqlQueryHolder::~SqlQueryHolder() { for(size_t i = 0; i < m_queries.size(); i++) @@ -153,23 +174,28 @@ SqlQueryHolder::~SqlQueryHolder() } } } + void SqlQueryHolder::SetSize(size_t size) { /// to optimize push_back, reserve the number of queries about to be executed m_queries.resize(size); } + void SqlQueryHolderEx::Execute(Database *db) { if(!m_holder || !m_callback || !m_queue) return; + /// we can do this, we are friends std::vector<SqlQueryHolder::SqlResultPair> &queries = m_holder->m_queries; + for(size_t i = 0; i < queries.size(); i++) { /// execute all queries in the holder and pass the results char const *sql = queries[i].first; if(sql) m_holder->SetResult(i, db->Query(sql)); } + /// sync with the caller thread m_queue->add(m_callback); } |