Listen on key when sending custom message

1 files changed, 6 insertions, 1 deletions

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;
     }