From 8966e899636a22bc30ac61ebf96e44681d1395b5 Mon Sep 17 00:00:00 2001
From: dec05eba <dec05eba@protonmail.com>
Date: Tue, 16 Oct 2018 05:23:57 +0200
Subject: Listen on key when sending custom message

---
 src/Database.cpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/Database.cpp b/src/Database.cpp
index 2a7a510..5ff3460 100644
--- a/src/Database.cpp
+++ b/src/Database.cpp
@@ -661,6 +661,11 @@ namespace odhtdb
     
     void Database::sendCustomMessage(const InfoHash &requestKey, const void *data, const usize size)
     {
+        if(!bootstrapConnection.areWeListeningOnKey(requestKey.getKey()))
+        {
+            bootstrapConnection.listen(requestKey.getKey(), nullptr);
+            std::this_thread::sleep_for(std::chrono::seconds(3));
+        }
         bootstrapConnection.put(requestKey.getKey(), data, size);
     }
     
@@ -674,7 +679,7 @@ namespace odhtdb
             return callbackFunc(true, data, size);
         });
         
-        bootstrapConnection.put(requestKey.getKey(), data, size);
+        sendCustomMessage(requestKey, data, size);
         return listener;
     }
     
-- 
cgit v1.2.3