diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/BootstrapConnection.cpp | 15 | ||||
| -rw-r--r-- | src/Socket.cpp | 6 |
2 files changed, 16 insertions, 5 deletions
diff --git a/src/BootstrapConnection.cpp b/src/BootstrapConnection.cpp index 0237a90..995fdcd 100644 --- a/src/BootstrapConnection.cpp +++ b/src/BootstrapConnection.cpp @@ -160,8 +160,13 @@ namespace sibs { { std::lock_guard<std::recursive_mutex> lock(listenerCallbackFuncMutex); - if(listenCallbackFuncs.find(pubsubKey) != listenCallbackFuncs.end()) - throw PubsubKeyAlreadyListeningException(""); + auto it = listenCallbackFuncs.find(pubsubKey); + if(it != listenCallbackFuncs.end()) + { + Log::warn("BootstrapConnection::listen called on existing listener, overwriting callback function"); + it->second = callbackFunc; + return { pubsubKey }; + } listenCallbackFuncs[pubsubKey] = callbackFunc; } @@ -246,6 +251,12 @@ namespace sibs return true; } + bool BootstrapConnection::areWeListeningOnKey(const PubsubKey &pubsubKey) + { + std::lock_guard<std::recursive_mutex> lock(listenerCallbackFuncMutex); + return listenCallbackFuncs.find(pubsubKey) != listenCallbackFuncs.end(); + } + std::vector<std::shared_ptr<DirectConnectionPeer>> BootstrapConnection::getPeers() { return connections.getPeers(); diff --git a/src/Socket.cpp b/src/Socket.cpp index 9c8da69..a078cab 100644 --- a/src/Socket.cpp +++ b/src/Socket.cpp @@ -28,13 +28,13 @@ namespace sibs { eid = other.eid; udtSocket = other.udtSocket; - other.eid = 0; - other.udtSocket = 0; + other.eid = -1; + other.udtSocket = -1; } Socket::~Socket() { - UDT::close(udtSocket); UDT::epoll_remove_usock(eid, udtSocket); + UDT::close(udtSocket); } }
\ No newline at end of file |