Improving walletOfOwner efficiency (#233)

* Improving walletOfOwner efficiency * Using "_currentIndex" instead of "maxSupply"
hashlips-lab · Apr 12, 2022 · 6b2d789 · 6b2d789
1 parent 5886b1e
commit 6b2d789
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/smart-contract/contracts/YourNftToken.sol b/smart-contract/contracts/YourNftToken.sol
@@ -79,10 +79,10 @@ contract YourNftToken is ERC721A, Ownable, ReentrancyGuard {
     uint256 ownedTokenIndex = 0;
     address latestOwnerAddress;
 
-    while (ownedTokenIndex < ownerTokenCount && currentTokenId <= maxSupply) {
-      if (_exists(currentTokenId)) {
-        TokenOwnership memory ownership = _ownerships[currentTokenId];
+    while (ownedTokenIndex < ownerTokenCount && currentTokenId < _currentIndex) {
+      TokenOwnership memory ownership = _ownerships[currentTokenId];
 
+      if (!ownership.burned) {
         if (ownership.addr != address(0)) {
           latestOwnerAddress = ownership.addr;
         }