fix formatting

Signed-off-by: Ritvi Bhatt <ribhatt@amazon.com>

Author: ritvibhatt

core/src/main/java/org/opensearch/sql/ast/tree/Sort.java

@@ -12,11 +12,9 @@
 
 import com.google.common.collect.ImmutableList;
 import java.util.List;
-import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.EqualsAndHashCode;
 import lombok.Getter;
-import lombok.RequiredArgsConstructor;
 import lombok.ToString;
 import org.opensearch.sql.ast.AbstractNodeVisitor;
 import org.opensearch.sql.ast.expression.Field;

core/src/main/java/org/opensearch/sql/calcite/CalciteRelNodeVisitor.java

@@ -337,7 +337,7 @@ public RelNode visitSort(Sort node, CalcitePlanContext context) {
     if (node.getCount() != null && node.getCount() > 0) {
       context.relBuilder.limit(0, node.getCount());
     }
-    
+
     return context.relBuilder.peek();
   }
 

core/src/main/java/org/opensearch/sql/planner/logical/LogicalPlanDSL.java

@@ -101,7 +101,8 @@ public static LogicalPlan sort(LogicalPlan input, Pair<SortOption, Expression>..
     return new LogicalSort(input, null, Arrays.asList(sorts));
   }
 
-  public static LogicalPlan sort(LogicalPlan input, Integer count, Pair<SortOption, Expression>... sorts) {
+  public static LogicalPlan sort(
+      LogicalPlan input, Integer count, Pair<SortOption, Expression>... sorts) {
     return new LogicalSort(input, count, Arrays.asList(sorts));
   }
 

core/src/main/java/org/opensearch/sql/planner/logical/LogicalSort.java

@@ -24,7 +24,8 @@ public class LogicalSort extends LogicalPlan {
   private final List<Pair<SortOption, Expression>> sortList;
 
   /** Constructor of LogicalSort. */
-  public LogicalSort(LogicalPlan child, Integer count, List<Pair<SortOption, Expression>> sortList) {
+  public LogicalSort(
+      LogicalPlan child, Integer count, List<Pair<SortOption, Expression>> sortList) {
     super(Collections.singletonList(child));
     this.count = count;
     this.sortList = sortList;

core/src/main/java/org/opensearch/sql/planner/optimizer/rule/PushFilterUnderSort.java

@@ -41,6 +41,7 @@ public PushFilterUnderSort() {
   @Override
   public LogicalPlan apply(LogicalFilter filter, Captures captures) {
     LogicalSort sort = captures.get(capture);
-    return new LogicalSort(filter.replaceChildPlans(sort.getChild()), sort.getCount(), sort.getSortList());
+    return new LogicalSort(
+        filter.replaceChildPlans(sort.getChild()), sort.getCount(), sort.getSortList());
   }
 }

opensearch/src/main/java/org/opensearch/sql/opensearch/storage/scan/OpenSearchIndexScanQueryBuilder.java

@@ -88,7 +88,7 @@ public boolean pushDownSort(LogicalSort sort) {
     if (sort.getCount() != null) {
       requestBuilder.pushDownLimit(sort.getCount(), 0);
     }
-    
+
     return true;
   }
 

ppl/src/main/java/org/opensearch/sql/ppl/parser/AstBuilder.java

@@ -43,12 +43,11 @@
 import org.opensearch.sql.ast.dsl.AstDSL;
 import org.opensearch.sql.ast.expression.Alias;
 import org.opensearch.sql.ast.expression.AllFieldsExcludeMeta;
+import org.opensearch.sql.ast.expression.Argument;
 import org.opensearch.sql.ast.expression.EqualTo;
 import org.opensearch.sql.ast.expression.Field;
 import org.opensearch.sql.ast.expression.Let;
 import org.opensearch.sql.ast.expression.Literal;
-import org.opensearch.sql.ast.expression.Argument;
-import org.opensearch.sql.ast.expression.DataType;
 import org.opensearch.sql.ast.expression.Map;
 import org.opensearch.sql.ast.expression.ParseMethod;
 import org.opensearch.sql.ast.expression.PatternMethod;
@@ -378,18 +377,19 @@ public UnresolvedPlan visitHeadCommand(HeadCommandContext ctx) {
   public UnresolvedPlan visitSortCommand(SortCommandContext ctx) {
     Integer count = ctx.count != null ? Integer.parseInt(ctx.count.getText()) : null;
     boolean shouldReverse = ctx.DESC() != null || ctx.D() != null;
-    
-    List<Field> sortFields = ctx.sortbyClause().sortField().stream()
-        .map(sort -> (Field) internalVisitExpression(sort))
-        .map(field -> shouldReverse ? reverseSortDirection(field) : field)
-        .collect(Collectors.toList());
-    
+
+    List<Field> sortFields =
+        ctx.sortbyClause().sortField().stream()
+            .map(sort -> (Field) internalVisitExpression(sort))
+            .map(field -> shouldReverse ? reverseSortDirection(field) : field)
+            .collect(Collectors.toList());
+
     return new Sort(count, sortFields);
   }
 
   private Field reverseSortDirection(Field field) {
     List<Argument> updatedArgs = new ArrayList<>();
-    
+
     for (Argument arg : field.getFieldArgs()) {
       if ("asc".equals(arg.getArgName())) {
         boolean isAscending = (Boolean) arg.getValue().getValue();
@@ -398,7 +398,7 @@ private Field reverseSortDirection(Field field) {
         updatedArgs.add(arg);
       }
     }
-    
+
     return new Field(field.getField(), updatedArgs);
   }
 

ppl/src/main/java/org/opensearch/sql/ppl/parser/AstExpressionBuilder.java

@@ -196,9 +196,9 @@ public UnresolvedExpression visitWcFieldExpression(WcFieldExpressionContext ctx)
   @Override
   public UnresolvedExpression visitSortField(SortFieldContext ctx) {
 
-    UnresolvedExpression fieldExpression = 
+    UnresolvedExpression fieldExpression =
         visit(ctx.sortFieldExpression().fieldExpression().qualifiedName());
-    
+
     // Apply type casting for sort field keywords - create Cast expression if needed
     if (ctx.sortFieldExpression().IP() != null) {
       fieldExpression = new Cast(fieldExpression, AstDSL.stringLiteral("ip"));

ppl/src/test/java/org/opensearch/sql/ppl/parser/AstBuilderTest.java

@@ -450,11 +450,7 @@ public void testSortCommandWithOptions() {
   @Test
   public void testSortCommandWithCount() {
     assertEqual(
-        "source=t | sort 100 f1",
-        sort(
-            relation("t"),
-            100,
-            field("f1", defaultSortFieldArgs())));
+        "source=t | sort 100 f1", sort(relation("t"), 100, field("f1", defaultSortFieldArgs())));
   }
 
   @Test
@@ -463,7 +459,10 @@ public void testSortCommandWithDesc() {
         "source=t | sort f1 desc",
         sort(
             relation("t"),
-            field("f1", exprList(argument("asc", booleanLiteral(false)), argument("type", nullLiteral())))));
+            field(
+                "f1",
+                exprList(
+                    argument("asc", booleanLiteral(false)), argument("type", nullLiteral())))));
   }
 
   @Test
@@ -472,7 +471,10 @@ public void testSortCommandWithD() {
         "source=t | sort f1 d",
         sort(
             relation("t"),
-            field("f1", exprList(argument("asc", booleanLiteral(false)), argument("type", nullLiteral())))));
+            field(
+                "f1",
+                exprList(
+                    argument("asc", booleanLiteral(false)), argument("type", nullLiteral())))));
   }
 
   @Test