helper func for later

PrimeIntellect-ai · Jan 21, 2025 · 14857c2 · 14857c2
1 parent aee1a3f
commit 14857c2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/ComputeRegistry.sol b/src/ComputeRegistry.sol
@@ -171,6 +171,10 @@ contract ComputeRegistry is IComputeRegistry, AccessControlEnumerable {
         return providers[provider].activeNodes;
     }
 
+    function getProviderTotalNodes(address provider) external view returns (uint32) {
+        return uint32(providers[provider].nodes.length);
+    }
+
     function getNodes(address provider, uint256 page, uint256 limit) external view returns (ComputeNode[] memory) {
         if (page == 0 && limit == 0) {
             return providers[provider].nodes;

diff --git a/src/interfaces/IComputeRegistry.sol b/src/interfaces/IComputeRegistry.sol
@@ -54,6 +54,7 @@ interface IComputeRegistry is IAccessControlEnumerable {
     function getNodeValidationStatus(address provider, address subkey) external returns (bool);
     function getProvider(address provider) external view returns (ComputeProvider memory);
     function getProviderActiveNodes(address provider) external view returns (uint32);
+    function getProviderTotalNodes(address provider) external view returns (uint32);
     function getProviderAddressList() external view returns (address[] memory);
     function getProviderValidatedNodes(address provider, bool filterForActive)
         external