add get length

onflow · Sep 6, 2023 · 719b931 · 719b931
1 parent f3c95ef
commit 719b931
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 6 deletions.
diff --git a/contracts/ExampleNFT.cdc b/contracts/ExampleNFT.cdc
@@ -220,6 +220,14 @@ access(all) contract ExampleNFT: NonFungibleToken, ViewResolver {
             return self.ownedNFTs.keys
         }
 
+        /// Helper method for getting the number of NFTs stored in the collection
+        ///
+        /// @return An Integer representing the number of NFTs
+        ///
+        access(all) view fun getLength(): Int {
+            return self.ownedNFTs.keys.length
+        }
+
         /// Gets a reference to an NFT in the collection so that
         /// the caller can read its metadata and call its methods
         ///

diff --git a/contracts/NonFungibleToken-v2.cdc b/contracts/NonFungibleToken-v2.cdc
@@ -284,6 +284,9 @@ access(all) contract NonFungibleToken {
         /// getIDs returns an array of the IDs that are in the collection
         access(all) view fun getIDs(): [UInt64]
 
+        /// Gets the amount of NFTs stored in the collection
+        access(all) view fun getLength(): Int
+
         /// getIDsWithTypes returns a list of IDs that are in the collection, keyed by type
         /// Should only be used by collections that can store multiple NFT types
         access(all) view fun getIDsWithTypes(): {Type: [UInt64]}