From 1d2aafd39bcb79a67357d198ce9b2345642fdd39 Mon Sep 17 00:00:00 2001 From: StormBytePP Date: Wed, 19 Aug 2015 19:02:10 +0200 Subject: Core/Build: Merge common library and move database out of shared --- src/server/database/Database/QueryHolder.h | 66 ++++++++++++++++++++++++++++++ 1 file changed, 66 insertions(+) create mode 100644 src/server/database/Database/QueryHolder.h (limited to 'src/server/database/Database/QueryHolder.h') diff --git a/src/server/database/Database/QueryHolder.h b/src/server/database/Database/QueryHolder.h new file mode 100644 index 00000000000..657496e8196 --- /dev/null +++ b/src/server/database/Database/QueryHolder.h @@ -0,0 +1,66 @@ +/* + * Copyright (C) 2008-2015 TrinityCore + * + * 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, see . + */ + +#ifndef _QUERYHOLDER_H +#define _QUERYHOLDER_H + +#include + +class SQLQueryHolder +{ + friend class SQLQueryHolderTask; + private: + typedef std::pair SQLResultPair; + std::vector m_queries; + public: + SQLQueryHolder() { } + virtual ~SQLQueryHolder(); + bool SetQuery(size_t index, const char* sql); + template + bool SetPQuery(size_t index, Format&& sql, Args&&... args) + { + return SetQuery(index, Trinity::StringFormat(std::forward(sql), std::forward(args)...).c_str()); + } + bool SetPreparedQuery(size_t index, PreparedStatement* stmt); + void SetSize(size_t size); + QueryResult GetResult(size_t index); + PreparedQueryResult GetPreparedResult(size_t index); + void SetResult(size_t index, ResultSet* result); + void SetPreparedResult(size_t index, PreparedResultSet* result); +}; + +typedef std::future QueryResultHolderFuture; +typedef std::promise QueryResultHolderPromise; + +class SQLQueryHolderTask : public SQLOperation +{ + private: + SQLQueryHolder* m_holder; + QueryResultHolderPromise m_result; + bool m_executed; + + public: + SQLQueryHolderTask(SQLQueryHolder* holder) + : m_holder(holder), m_executed(false) { } + + ~SQLQueryHolderTask(); + + bool Execute() override; + QueryResultHolderFuture GetFuture() { return m_result.get_future(); } +}; + +#endif -- cgit v1.2.3