Adding pagination support for NftRepository

Author: henrique021

hedera-base/src/main/java/com/openelements/hedera/base/NftRepository.java

@@ -20,7 +20,7 @@ public interface NftRepository {
      * @throws HederaException if the search fails
      */
     @NonNull
-    List<Nft> findByOwner(@NonNull AccountId ownerId) throws HederaException;
+    Page<Nft> findByOwner(@NonNull AccountId ownerId) throws HederaException;
 
     /**
      * Return all NFTs of a given type.
@@ -52,7 +52,7 @@ public interface NftRepository {
      * @throws HederaException if the search fails
      */
     @NonNull
-    List<Nft> findByOwnerAndType(@NonNull AccountId ownerId, @NonNull TokenId tokenId) throws HederaException;
+    Page<Nft> findByOwnerAndType(@NonNull AccountId ownerId, @NonNull TokenId tokenId) throws HederaException;
 
     /**
      * Return the NFT of a given type and serial owned by a specific account.

hedera-base/src/main/java/com/openelements/hedera/base/implementation/NftRepositoryImpl.java

@@ -22,7 +22,7 @@ public NftRepositoryImpl(@NonNull final MirrorNodeClient mirrorNodeClient) {
 
     @NonNull
     @Override
-    public List<Nft> findByOwner(@NonNull final AccountId owner) throws HederaException {
+    public Page<Nft> findByOwner(@NonNull final AccountId owner) throws HederaException {
         return mirrorNodeClient.queryNftsByAccount(owner);
     }
 
@@ -41,7 +41,7 @@ public Optional<Nft> findByTypeAndSerial(@NonNull final TokenId tokenId, final l
 
     @NonNull
     @Override
-    public List<Nft> findByOwnerAndType(@NonNull final AccountId owner, @NonNull final TokenId tokenId)
+    public Page<Nft> findByOwnerAndType(@NonNull final AccountId owner, @NonNull final TokenId tokenId)
             throws HederaException {
         return mirrorNodeClient.queryNftsByAccountAndTokenId(owner, tokenId);
     }

hedera-base/src/main/java/com/openelements/hedera/base/mirrornode/MirrorNodeClient.java

@@ -14,10 +14,10 @@
 public interface MirrorNodeClient {
 
     @NonNull
-    List<Nft> queryNftsByAccount(@NonNull AccountId accountId) throws HederaException;
+    Page<Nft> queryNftsByAccount(@NonNull AccountId accountId) throws HederaException;
 
     @NonNull
-    List<Nft> queryNftsByAccountAndTokenId(@NonNull AccountId accountId, @NonNull TokenId tokenId)
+    Page<Nft> queryNftsByAccountAndTokenId(@NonNull AccountId accountId, @NonNull TokenId tokenId)
             throws HederaException;
 
     @NonNull

hedera-spring/src/main/java/com/openelements/hedera/spring/implementation/MirrorNodeClientImpl.java

@@ -65,19 +65,38 @@ public MirrorNodeClientImpl(@NonNull final String mirrorNodeEndpoint) {
     }
 
     @Override
-    public List<Nft> queryNftsByAccount(@NonNull final AccountId accountId) throws HederaException {
+    public Page<Nft> queryNftsByAccount(@NonNull final AccountId accountId) throws HederaException {
         Objects.requireNonNull(accountId, "newAccountId must not be null");
-        final JsonNode jsonNode = doGetCall("/api/v1/accounts/" + accountId + "/nfts");
-        return jsonNodeToNftList(jsonNode);
+        final URI uri = URI.create(
+                getUriPrefix()
+                        + "/api/v1/accounts/" + accountId + "/nfts");
+        
+        final Function<JsonNode, List<Nft>> dataExtractionFunction = node -> getNfts(node);
+        final Function<JsonNode, URI> nextUriExtractionFunction = node -> getNextUri(node);
+
+        return new RestBasedPage<>(objectMapper, restClient,
+                uri,
+                dataExtractionFunction, nextUriExtractionFunction);
     }
 
     @Override
-    public List<Nft> queryNftsByAccountAndTokenId(@NonNull final AccountId accountId, @NonNull final TokenId tokenId)
+    public Page<Nft> queryNftsByAccountAndTokenId(@NonNull final AccountId accountId, @NonNull final TokenId tokenId)
             throws HederaException {
         Objects.requireNonNull(accountId, "newAccountId must not be null");
         Objects.requireNonNull(tokenId, "tokenId must not be null");
-        final JsonNode jsonNode = doGetCall("/api/v1/tokens/" + tokenId + "/nfts", Map.of("account.id", accountId));
-        return jsonNodeToNftList(jsonNode);
+        
+        final URI uri = URI.create(
+                getUriPrefix()
+                        + doGetCall ("/api/v1/tokens/" + tokenId + "/nfts", Map.of("account.id", accountId)));
+        
+        final Function<JsonNode, List<Nft>> dataExtractionFunction = node -> getNfts(node);
+        final Function<JsonNode, URI> nextUriExtractionFunction = node -> getNextUri(node);
+
+        return new RestBasedPage<>(objectMapper, restClient,
+                uri,
+                dataExtractionFunction, nextUriExtractionFunction);
+        
+        
     }
 
     @Override

hedera-spring/src/test/java/com/openelements/hedera/spring/test/NftRepositoryTests.java

@@ -146,7 +146,8 @@ void findByAccountId() throws Exception {
         hederaTestUtils.waitForMirrorNodeRecords();
 
         //when
-        final List<Nft> result = nftRepository.findByOwner(newOwner);
+        final Page<Nft> slice = nftRepository.findByOwner(newOwner);
+        final List<Nft> result = getAll(slice);
 
         //then
         Assertions.assertNotNull(result);
@@ -168,7 +169,8 @@ void findByAccountIdWIthZeroResult() throws Exception {
         hederaTestUtils.waitForMirrorNodeRecords();
 
         //when
-        final List<Nft> result = nftRepository.findByOwner(newOwner);
+        final Page<Nft> slice = nftRepository.findByOwner(newOwner);
+        final List<Nft> result = getAll(slice);
 
         //then
         Assertions.assertNotNull(result);
@@ -196,7 +198,8 @@ void findByTokenIdAndAccountId() throws Exception {
         hederaTestUtils.waitForMirrorNodeRecords();
 
         //when
-        final List<Nft> result = nftRepository.findByOwnerAndType(newOwner, tokenId);
+        final Page<Nft> slice = nftRepository.findByOwnerAndType(newOwner, tokenId);
+        final List<Nft> result = getAll(slice);
 
         //then
         Assertions.assertNotNull(result);
@@ -218,7 +221,8 @@ void findByTokenIdAndAccountIdWithZeroResult() throws Exception {
         hederaTestUtils.waitForMirrorNodeRecords();
 
         //when
-        final List<Nft> result = nftRepository.findByOwnerAndType(newOwner, tokenId);
+        final Page<Nft> slice = nftRepository.findByOwnerAndType(newOwner, tokenId);
+        final List<Nft> result = getAll(slice);
 
         //then
         Assertions.assertNotNull(result);