From 14857c2db0882db1597826e7cef8cc2c8290859d Mon Sep 17 00:00:00 2001 From: Matthew Di Ferrante Date: Mon, 20 Jan 2025 19:20:44 -0800 Subject: [PATCH] helper func for later --- src/ComputeRegistry.sol | 4 ++++ src/interfaces/IComputeRegistry.sol | 1 + 2 files changed, 5 insertions(+) diff --git a/src/ComputeRegistry.sol b/src/ComputeRegistry.sol index e058e80..6a3de67 100644 --- 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 index a304eee..0311774 100644 --- 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