net_mn: add single MN connection process.

PIVX-Project · Jan 12, 2022 · 934dd8c · 934dd8c
1 parent 15d70e7
commit 934dd8c
Show file tree

Hide file tree

Showing 2 changed files with 59 additions and 25 deletions.
diff --git a/src/tiertwo/net_masternodes.cpp b/src/tiertwo/net_masternodes.cpp
@@ -68,6 +68,16 @@ bool TierTwoConnMan::isMasternodeQuorumNode(const CNode* pnode)
     return false;
 }
 
+bool TierTwoConnMan::addPendingMasternode(const uint256& proTxHash)
+{
+    LOCK(cs_vPendingMasternodes);
+    if (std::find(vPendingMasternodes.begin(), vPendingMasternodes.end(), proTxHash) != vPendingMasternodes.end()) {
+        return false;
+    }
+    vPendingMasternodes.emplace_back(proTxHash);
+    return true;
+}
+
 void TierTwoConnMan::addPendingProbeConnections(const std::set<uint256>& proTxHashes)
 {
     LOCK(cs_vPendingMasternodes);
@@ -160,40 +170,61 @@ void TierTwoConnMan::ThreadOpenMasternodeConnections()
         bool isProbe = false;
         {
             LOCK(cs_vPendingMasternodes);
-            std::vector<CDeterministicMNCPtr> pending;
-            for (const auto& group: masternodeQuorumNodes) {
-                for (const auto& proRegTxHash: group.second) {
-                    // Skip if already have this member connected
-                    if (std::count(connectedProRegTxHashes.begin(), connectedProRegTxHashes.end(), proRegTxHash) > 0) continue;
-
-                    // Check if DMN exists in tip list
-                    const auto& dmn = mnList.GetValidMN(proRegTxHash);
-                    if (!dmn) continue;
-                    auto peerData = std::find(connectedNodes.begin(), connectedNodes.end(), dmn->pdmnState->addr);
 
-                    // Skip already connected nodes.
-                    if (peerData != std::end(connectedNodes) && (peerData->f_disconnect || peerData->f_is_mn_conn)) {
-                        continue;
+            // First try to connect to pending MNs
+            if (!vPendingMasternodes.empty()) {
+                auto dmn = mnList.GetValidMN(vPendingMasternodes.front());
+                vPendingMasternodes.erase(vPendingMasternodes.begin());
+                if (dmn) {
+                    auto peerData = std::find(connectedNodes.begin(), connectedNodes.end(), dmn->pdmnState->addr);
+                    if (peerData == std::end(connectedNodes)) {
+                        dmnToConnect = dmn;
+                        LogPrint(BCLog::NET_MN, "%s -- opening pending masternode connection to %s, service=%s\n",
+                                 __func__, dmn->proTxHash.ToString(), dmn->pdmnState->addr.ToString());
                     }
+                }
+            }
 
-                    // Check if we already tried this connection recently to not retry too often
-                    int64_t lastAttempt = g_mmetaman.GetMetaInfo(dmn->proTxHash)->GetLastOutboundAttempt();
-                    // back off trying connecting to an address if we already tried recently
-                    if (currentTime - lastAttempt < chainParams.LLMQConnectionRetryTimeout()) {
-                        continue;
+            // Secondly, try to connect quorum members
+            if (!dmnToConnect) {
+                std::vector<CDeterministicMNCPtr> pending;
+                for (const auto& group: masternodeQuorumNodes) {
+                    for (const auto& proRegTxHash: group.second) {
+                        // Skip if already have this member connected
+                        if (std::count(connectedProRegTxHashes.begin(), connectedProRegTxHashes.end(), proRegTxHash) > 0)
+                            continue;
+
+                        // Check if DMN exists in tip list
+                        const auto& dmn = mnList.GetValidMN(proRegTxHash);
+                        if (!dmn) continue;
+                        auto peerData = std::find(connectedNodes.begin(), connectedNodes.end(), dmn->pdmnState->addr);
+
+                        // Skip already connected nodes.
+                        if (peerData != std::end(connectedNodes) &&
+                            (peerData->f_disconnect || peerData->f_is_mn_conn)) {
+                            continue;
+                        }
+
+                        // Check if we already tried this connection recently to not retry too often
+                        int64_t lastAttempt = g_mmetaman.GetMetaInfo(dmn->proTxHash)->GetLastOutboundAttempt();
+                        // back off trying connecting to an address if we already tried recently
+                        if (currentTime - lastAttempt < chainParams.LLMQConnectionRetryTimeout()) {
+                            continue;
+                        }
+                        pending.emplace_back(dmn);
                     }
-                    pending.emplace_back(dmn);
                 }
-            }
-            // Select a random node to connect
-            if (!pending.empty()) {
-                dmnToConnect = pending[GetRandInt((int)pending.size())];
-                LogPrint(BCLog::NET_MN, "TierTwoConnMan::%s -- opening quorum connection to %s, service=%s\n",
-                         __func__, dmnToConnect->proTxHash.ToString(), dmnToConnect->pdmnState->addr.ToString());
+                // Select a random node to connect
+                if (!pending.empty()) {
+                    dmnToConnect = pending[GetRandInt((int) pending.size())];
+                    LogPrint(BCLog::NET_MN, "TierTwoConnMan::%s -- opening quorum connection to %s, service=%s\n",
+                             __func__, dmnToConnect->proTxHash.ToString(), dmnToConnect->pdmnState->addr.ToString());
+                }
             }
 
             // If no node was selected, let's try to probe nodes connection
             if (!dmnToConnect) {
+                std::vector<CDeterministicMNCPtr> pending;
                 for (auto it = masternodePendingProbes.begin(); it != masternodePendingProbes.end(); ) {
                     auto dmn = mnList.GetMN(*it);
                     if (!dmn) {

diff --git a/src/tiertwo/net_masternodes.h b/src/tiertwo/net_masternodes.h
@@ -39,6 +39,9 @@ class TierTwoConnMan
     // Returns true if the node has the same address as a MN.
     bool isMasternodeQuorumNode(const CNode* pnode);
 
+    // Add DMN to the pending connection list
+    bool addPendingMasternode(const uint256& proTxHash);
+
     // Adds the DMNs to the pending to probe list
     void addPendingProbeConnections(const std::set<uint256>& proTxHashes);