Add GetActiveConnectionCount to IConnectionSet

2021-06-21 11:46:16 -07:00
parent 04387be083
commit 8f5681ffdb
6 changed files with 43 additions and 1 deletions
@@ -111,6 +111,24 @@ namespace AzNetworking
        return aznumeric_cast<uint32_t>(m_connectionIdMap.size());
    }

+    uint32_t UdpConnectionSet::GetActiveConnectionCount() const
+    {
+        uint32_t activeConnections = 0;
+        for (auto iter = m_connectionIdMap.begin(); iter != m_connectionIdMap.end(); ++iter)
+        {
+            if (iter->second.get())
+            {
+                ConnectionState state = iter->second.get()->GetConnectionState();
+                if (state == ConnectionState::Connected || state == ConnectionState::Connecting)
+                {
+                    ++activeConnections;
+                }
+            }
+        }
+
+        return activeConnections;
+    }
+
    UdpConnection* UdpConnectionSet::GetConnection(const IpAddress& address) const
    {
        RemoteAddressMap::const_iterator lookup = m_remoteAddressMap.find(address);
@@ -50,6 +50,7 @@ namespace AzNetworking
        IConnection* GetConnection(ConnectionId connectionId) const override;
        ConnectionId GetNextConnectionId() override;
        uint32_t GetConnectionCount() const override;
+        uint32_t GetActiveConnectionCount() const override;
        //! @}

        //! Retrieves a connection from this connection list instance by endpoint remote address