1 files changed, 17 insertions, 1 deletions
diff --git a/src/server/bnetserver/REST/LoginHttpSession.cpp b/src/server/bnetserver/REST/LoginHttpSession.cpp
index 23a317d3726..bd8afdbcf2f 100644
--- a/src/server/bnetserver/REST/LoginHttpSession.cpp
+++ b/src/server/bnetserver/REST/LoginHttpSession.cpp
@@ -83,7 +83,7 @@ public:
         // build initializer chain
         boost::container::static_vector<std::shared_ptr<Trinity::Net::SocketConnectionInitializer>, 4> initializers;
 
-        initializers.stable_emplace_back(std::make_shared<Trinity::Net::IpBanCheckConnectionInitializer<BaseSocket>>(this));
+        initializers.stable_emplace_back(std::make_shared<Trinity::Net::IpBanCheckConnectionInitializer<Battlenet::LoginHttpSession>>(&_owner));
 
         if constexpr (std::is_same_v<BaseSocket, Trinity::Net::Http::SslSocket>)
             initializers.stable_emplace_back(std::make_shared<Trinity::Net::SslHandshakeConnectionInitializer<BaseSocket>>(this));
@@ -124,10 +124,26 @@ LoginHttpSession::LoginHttpSession(Trinity::Net::IoContextTcpSocket&& socket)
 {
 }
 
+LoginHttpSession::~LoginHttpSession() = default;
+
 void LoginHttpSession::Start()
 {
     TC_LOG_TRACE("server.http.session", "{} Accepted connection", GetClientInfo());
 
     return _socket->Start();
 }
+
+bool LoginHttpSession::Update()
+{
+    if (!_socket->Update())
+        return false;
+
+    _queryProcessor.ProcessReadyCallbacks();
+    return true;
+}
+
+void LoginHttpSession::QueueQuery(QueryCallback&& queryCallback)
+{
+    _queryProcessor.AddCallback(std::move(queryCallback));
+}
 }