From 9b806c6b5d9e83f8614523317a44742954f97744 Mon Sep 17 00:00:00 2001
From: Shauren <shauren.trinity@gmail.com>
Date: Fri, 31 Jul 2020 23:27:26 +0200
Subject: Core/DBLayer: Add nicer api for SQLQueryHolders

---
 src/server/database/Database/QueryHolder.h | 19 +++++++++++++++++++
 1 file changed, 19 insertions(+)

(limited to 'src/server/database/Database/QueryHolder.h')

diff --git a/src/server/database/Database/QueryHolder.h b/src/server/database/Database/QueryHolder.h
index fe4a0e78331..f94eff49d2c 100644
--- a/src/server/database/Database/QueryHolder.h
+++ b/src/server/database/Database/QueryHolder.h
@@ -63,4 +63,23 @@ class TC_DATABASE_API SQLQueryHolderTask : public SQLOperation
         QueryResultHolderFuture GetFuture() { return m_result.get_future(); }
 };
 
+class TC_DATABASE_API SQLQueryHolderCallback
+{
+public:
+    SQLQueryHolderCallback(QueryResultHolderFuture&& future) : m_future(std::move(future)) { }
+    SQLQueryHolderCallback(SQLQueryHolderCallback&&) = default;
+
+    SQLQueryHolderCallback& operator=(SQLQueryHolderCallback&&) = default;
+
+    void AfterComplete(std::function<void(SQLQueryHolderBase*)> callback) &
+    {
+        m_callback = std::move(callback);
+    }
+
+    bool InvokeIfReady();
+
+    QueryResultHolderFuture m_future;
+    std::function<void(SQLQueryHolderBase*)> m_callback;
+};
+
 #endif
-- 
cgit v1.2.3