Fix handling when there is no NextRowKey in response header (Azure#34143)

* Allow nextRowKey to be null when list next page

* Update CHANGELOG.md

* Update EntityPaged.java

The build was failing for checkstyle violations. This adjustment should fix them.

---------

Co-authored-by: Jair Myree <[email protected]>

Author: kellyyyyin

sdk/tables/azure-data-tables/CHANGELOG.md

@@ -7,8 +7,8 @@
 ### Breaking Changes
 
 ### Bugs Fixed
-
 - Fixed bug that disallowed empty strings as partition and row keys
+- Fixed handling of paging headers when Table Storage returned a `x-ms-continuation-NextPartitionKey` but no `x-ms-continuation-NextRowKey`. This was causing the subsequent pages are not fetched.
 ### Other Changes
 
 ## 12.3.9 (2023-03-08)
@@ -89,7 +89,7 @@
 
 #### Dependency Updates
 - Upgraded `azure-core` dependency to `1.29.1`
-- Upgraded `azure-core-http-netty` dependency to `1.12.2` 
+- Upgraded `azure-core-http-netty` dependency to `1.12.2`
 
 ## 12.3.0 (2022-05-11)
 
@@ -122,7 +122,7 @@
 ## 12.2.0 (2022-02-10)
 
 ### Features added
-- Implemented new traits (micro-interfaces) in `TableClientBuiler` and `TableServiceClientBuilder`. This makes the experience of using client builders more consistent across libraries in the Azure SDK for Java. 
+- Implemented new traits (micro-interfaces) in `TableClientBuiler` and `TableServiceClientBuilder`. This makes the experience of using client builders more consistent across libraries in the Azure SDK for Java.
 
 ### Other Changes
 
@@ -133,12 +133,12 @@
 ## 12.1.5 (2022-01-11)
 
 ### Bugs fixed
-- Fixed issue that made it so single quotes in entity names were not properly escaped according to OData standards. [[25066]](https://github.com/Azure/azure-sdk-for-java/pull/25066) 
+- Fixed issue that made it so single quotes in entity names were not properly escaped according to OData standards. [[25066]](https://github.com/Azure/azure-sdk-for-java/pull/25066)
 
 #### Dependency Updates
 - Upgraded `azure-core` dependency to `1.24.1`
 - Upgraded `azure-core-http-netty` dependency to `1.11.6`
-    
+
 ## 12.1.4 (2021-11-19)
 
 ### Other Changes

sdk/tables/azure-data-tables/src/main/java/com/azure/data/tables/TableAsyncClient.java

@@ -950,13 +950,13 @@ private <T extends TableEntity> Mono<PagedResponse<T>> listEntitiesNextPage(Stri
 
         String[] split = token.split(DELIMITER_CONTINUATION_TOKEN, 2);
 
-        if (split.length != 2) {
+        if (split.length == 0) {
             return monoError(logger, new RuntimeException(
-                "Split done incorrectly, must have partition and row key: " + token));
+                "Split done incorrectly, must have partition key: " + token));
         }
 
         String nextPartitionKey = split[0];
-        String nextRowKey = split[1];
+        String nextRowKey = split.length > 1 ? split[1] : null;
 
         return listEntities(nextPartitionKey, nextRowKey, context, options, resultType);
     }

sdk/tables/azure-data-tables/src/main/java/com/azure/data/tables/TableClient.java

@@ -959,13 +959,13 @@ private <T extends TableEntity> PagedResponse<T> listEntitiesNextPage(String tok
 
         String[] split = token.split(DELIMITER_CONTINUATION_TOKEN, 2);
 
-        if (split.length != 2) {
+        if (split.length == 0) {
             throw logger.logExceptionAsError(new RuntimeException(
-                "Split done incorrectly, must have partition and row key: " + token));
+                "Split done incorrectly, must have partition key: " + token));
         }
 
         String nextPartitionKey = split[0];
-        String nextRowKey = split[1];
+        String nextRowKey = split.length > 1 ? split[1] : null;
 
         return listEntities(nextPartitionKey, nextRowKey, context, options, resultType);
 

sdk/tables/azure-data-tables/src/main/java/com/azure/data/tables/implementation/EntityPaged.java

@@ -21,8 +21,10 @@ public class EntityPaged<T extends TableEntity> implements PagedResponse<T> {
 
     public EntityPaged(Response<TableEntityQueryResponse> httpResponse, List<T> entityList,
                 String nextPartitionKey, String nextRowKey) {
-        if (nextPartitionKey == null || nextRowKey == null) {
+        if (nextPartitionKey == null) {
             this.continuationToken = null;
+        } else if (nextRowKey == null) {
+            this.continuationToken = nextPartitionKey;
         } else {
             this.continuationToken = String.join(DELIMITER_CONTINUATION_TOKEN, nextPartitionKey, nextRowKey);
         }