fixing refactoring

gateway-service-impl/src/main/java/org/hypertrace/gateway/service/explore/entity/EntityRequestHandler.java

@@ -1,14 +1,18 @@
 package org.hypertrace.gateway.service.explore.entity;
 
 import java.util.HashSet;
+import java.util.List;
 import java.util.Optional;
 import java.util.Set;
 import org.hypertrace.gateway.service.common.AttributeMetadataProvider;
 import org.hypertrace.gateway.service.common.datafetcher.QueryServiceEntityFetcher;
+import org.hypertrace.gateway.service.common.util.DataCollectionUtil;
 import org.hypertrace.gateway.service.common.util.QueryServiceClient;
 import org.hypertrace.gateway.service.entity.config.EntityIdColumnsConfigs;
 import org.hypertrace.gateway.service.explore.ExploreRequestContext;
 import org.hypertrace.gateway.service.explore.RequestHandler;
+import org.hypertrace.gateway.service.explore.RowComparator;
+import org.hypertrace.gateway.service.v1.common.OrderByExpression;
 import org.hypertrace.gateway.service.v1.explore.EntityOption;
 import org.hypertrace.gateway.service.v1.explore.ExploreRequest;
 import org.hypertrace.gateway.service.v1.explore.ExploreResponse;
@@ -90,6 +94,33 @@ public ExploreResponse.Builder handleRequest(
     return builder;
   }
 
+  @Override
+  public void sortAndPaginatePostProcess(
+      ExploreResponse.Builder builder,
+      List<OrderByExpression> orderByExpressions,
+      int limit,
+      int offset) {
+    List<org.hypertrace.gateway.service.v1.common.Row.Builder> rowBuilders =
+        builder.getRowBuilderList();
+
+    List<org.hypertrace.gateway.service.v1.common.Row.Builder> sortedRowBuilders =
+        sortAndPaginateRowBuilders(rowBuilders, orderByExpressions, limit, offset);
+
+    builder.clearRow();
+    sortedRowBuilders.forEach(builder::addRow);
+  }
+
+  protected List<org.hypertrace.gateway.service.v1.common.Row.Builder> sortAndPaginateRowBuilders(
+      List<org.hypertrace.gateway.service.v1.common.Row.Builder> rowBuilders,
+      List<OrderByExpression> orderByExpressions,
+      int limit,
+      int offset) {
+    RowComparator rowComparator = new RowComparator(orderByExpressions);
+
+    return DataCollectionUtil.limitAndSort(
+        rowBuilders.stream(), limit, offset, orderByExpressions.size(), rowComparator);
+  }
+
   private boolean requestOnLiveEntities(Optional<EntityOption> entityOption) {
     if (entityOption.isEmpty()) {
       return true;