add more golden explain plan tests

Author: dang-stripe

pinot-query-planner/src/test/resources/queries/PhysicalOptimizerPlans.json

@@ -1125,6 +1125,76 @@
           "\n          PhysicalTableScan(table=[[default, a]])",
           "\n"
         ]
+      },
+      {
+        "description": "Two lookup joins on different dimension tables in one query",
+        "sql": "SET usePhysicalOptimizer=true; EXPLAIN PLAN FOR SELECT /*+ joinOptions(join_strategy='lookup') */ a.col1, b.col2, c.col2 FROM a JOIN b ON a.col1 = b.col1 JOIN c ON a.col2 = c.col1",
+        "output": [
+          "Execution Plan",
+          "\nPhysicalExchange(exchangeStrategy=[SINGLETON_EXCHANGE])",
+          "\n  PhysicalProject(col1=[$0], col2=[$3], col20=[$5])",
+          "\n    PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n      PhysicalJoin(condition=[=($1, $4)], joinType=[inner])",
+          "\n        PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n          PhysicalJoin(condition=[=($0, $2)], joinType=[inner])",
+          "\n            PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n              PhysicalProject(col1=[$0], col2=[$1])",
+          "\n                PhysicalTableScan(table=[[default, a]])",
+          "\n            PhysicalExchange(exchangeStrategy=[LOOKUP_LOCAL_EXCHANGE])",
+          "\n              PhysicalProject(col1=[$0], col2=[$1])",
+          "\n                PhysicalTableScan(table=[[default, b]])",
+          "\n        PhysicalExchange(exchangeStrategy=[LOOKUP_LOCAL_EXCHANGE])",
+          "\n          PhysicalProject(col1=[$0], col2=[$1])",
+          "\n            PhysicalTableScan(table=[[default, c]])",
+          "\n"
+        ]
+      },
+      {
+        "description": "Hash join then lookup join on same table with aggregation",
+        "sql": "SET usePhysicalOptimizer=true; EXPLAIN PLAN FOR SELECT /*+ joinOptions(join_strategy='lookup') */ t.col1, SUM(t.col3) FROM (SELECT /*+ joinOptions(join_strategy='hash') */ a.col1, a.col3, b.col2 AS dim1 FROM a JOIN b ON a.col2 = b.col1) t JOIN b dim2 ON t.col1 = dim2.col1 GROUP BY t.col1",
+        "output": [
+          "Execution Plan",
+          "\nPhysicalExchange(exchangeStrategy=[SINGLETON_EXCHANGE])",
+          "\n  PhysicalAggregate(group=[{0}], agg#0=[$SUM0($1)], aggType=[FINAL])",
+          "\n    PhysicalExchange(exchangeStrategy=[PARTITIONING_EXCHANGE], distKeys=[[0]])",
+          "\n      PhysicalAggregate(group=[{0}], agg#0=[$SUM0($1)], aggType=[LEAF])",
+          "\n        PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n          PhysicalJoin(condition=[=($0, $2)], joinType=[inner])",
+          "\n            PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n              PhysicalProject(col1=[$0], col3=[$2])",
+          "\n                PhysicalJoin(condition=[=($1, $3)], joinType=[inner])",
+          "\n                  PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n                    PhysicalProject(col1=[$0], col2=[$1], col3=[$2])",
+          "\n                      PhysicalTableScan(table=[[default, a]])",
+          "\n                  PhysicalExchange(exchangeStrategy=[PARTITIONING_EXCHANGE], distKeys=[[0]])",
+          "\n                    PhysicalProject(col1=[$0])",
+          "\n                      PhysicalTableScan(table=[[default, b]])",
+          "\n            PhysicalExchange(exchangeStrategy=[LOOKUP_LOCAL_EXCHANGE])",
+          "\n              PhysicalProject(col1=[$0])",
+          "\n                PhysicalTableScan(table=[[default, b]])",
+          "\n"
+        ]
+      },
+      {
+        "description": "Window function feeding into lookup join",
+        "sql": "SET usePhysicalOptimizer=true; EXPLAIN PLAN FOR SELECT /*+ joinOptions(join_strategy='lookup') */ t.col1, t.running_total, b.col2 FROM (SELECT col1, col2, SUM(col3) OVER (PARTITION BY col1 ORDER BY col2) AS running_total FROM a) t JOIN b ON t.col1 = b.col1",
+        "output": [
+          "Execution Plan",
+          "\nPhysicalExchange(exchangeStrategy=[SINGLETON_EXCHANGE])",
+          "\n  PhysicalProject(col1=[$0], running_total=[$1], col2=[$3])",
+          "\n    PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n      PhysicalJoin(condition=[=($0, $2)], joinType=[inner])",
+          "\n        PhysicalExchange(exchangeStrategy=[IDENTITY_EXCHANGE])",
+          "\n          PhysicalProject(col1=[$0], $1=[$3])",
+          "\n            PhysicalWindow(window#0=[window(partition {0} order by [1] aggs [SUM($2)])])",
+          "\n              PhysicalExchange(exchangeStrategy=[PARTITIONING_EXCHANGE], distKeys=[[0]], collation=[[1]])",
+          "\n                PhysicalProject(col1=[$0], col2=[$1], col3=[$2])",
+          "\n                  PhysicalTableScan(table=[[default, a]])",
+          "\n        PhysicalExchange(exchangeStrategy=[LOOKUP_LOCAL_EXCHANGE])",
+          "\n          PhysicalProject(col1=[$0], col2=[$1])",
+          "\n            PhysicalTableScan(table=[[default, b]])",
+          "\n"
+        ]
       }
     ]
   }