multi-stage-explain: Add a comparator for lists of strings

gortiz · gortiz · commit 272263125a02 · 2024-09-23T15:04:26.000+02:00
diff --git a/pinot-query-planner/src/main/java/org/apache/pinot/query/planner/explain/PlanNodeSorter.java b/pinot-query-planner/src/main/java/org/apache/pinot/query/planner/explain/PlanNodeSorter.java
@@ -211,6 +211,21 @@ public int compare(PlanNode o1, PlanNode o2) {
           case STRING:
             cmp3 = value1.getString().compareTo(value2.getString());
             break;
+          case STRINGLIST:
+            List<String> list1 = value1.getStringList().getValuesList();
+            List<String> list2 = value2.getStringList().getValuesList();
+            cmp3 = 0;
+            int min = Math.min(list1.size(), list2.size());
+            for (int i = 0; i < min; i++) {
+              cmp3 = list1.get(i).compareTo(list2.get(i));
+              if (cmp3 != 0) {
+                break;
+              }
+            }
+            if (cmp3 != 0) {
+              cmp3 = list1.size() - list2.size();
+            }
+            break;
           case VALUE_NOT_SET:
             cmp3 = 0;
             break;
