[CBO] Improve Transitive Closure + Cycle processing added (ydb-platform#10121)

Author: pashandor789

ydb/core/kqp/ut/common/kqp_ut_common.cpp

@@ -1463,6 +1463,24 @@ NJson::TJsonValue GetJoinOrder(const TString& deserializedPlan) {
     return GetJoinOrderImpl(optRoot);
 }
 
+NJson::TJsonValue GetJoinOrderFromDetailedJoinOrderImpl(const NJson::TJsonValue& opt) {
+    if (!opt.GetMapSafe().contains("table")) {
+        NJson::TJsonValue res;
+        auto args = opt.GetMapSafe().at("args").GetArraySafe();
+        for (size_t i = 0; i < args.size(); ++i) {
+            res.AppendValue(GetJoinOrderFromDetailedJoinOrderImpl(args[i]));
+        }
+        return res;
+    }
+
+    return opt.GetMapSafe().at("table");
+}
+
+NJson::TJsonValue GetJoinOrderFromDetailedJoinOrder(const TString& deserializedDetailedJoinOrder) {
+    NJson::TJsonValue optRoot;
+    NJson::ReadJsonTree(deserializedDetailedJoinOrder, &optRoot, true);
+    return GetJoinOrderFromDetailedJoinOrderImpl(optRoot);
+}
 
 } // namspace NKqp
 } // namespace NKikimr

ydb/core/kqp/ut/common/kqp_ut_common.h

@@ -350,5 +350,7 @@ NJson::TJsonValue GetDetailedJoinOrder(const TString& deserializedPlan, const TG
 /* Gets tables join order without details : only tables. */
 NJson::TJsonValue GetJoinOrder(const TString& deserializedPlan);
 
+NJson::TJsonValue GetJoinOrderFromDetailedJoinOrder(const TString& deserializedDetailedJoinOrder);
+
 } // namespace NKqp
 } // namespace NKikimr

ydb/core/kqp/ut/join/data/join_order/lookupbug.json

@@ -60,3 +60,4 @@
       }
     ]
 }
+

ydb/core/kqp/ut/join/data/join_order/tpcc.json

@@ -24,3 +24,4 @@
       }
     ]
 }
+

ydb/core/kqp/ut/join/data/join_order/tpcds64_1000s.json

@@ -52,3 +52,4 @@
       }
     ]
 }
+

ydb/core/kqp/ut/join/data/join_order/tpcds64_1000s_column_store.json

@@ -52,3 +52,4 @@
       }
     ]
 }
+

ydb/core/kqp/ut/join/data/join_order/tpcds64_small_1000s.json

@@ -82,3 +82,4 @@
       }
     ]
 }
+

ydb/core/kqp/ut/join/data/join_order/tpcds64_small_1000s_column_store.json

@@ -82,3 +82,4 @@
       }
     ]
 }
+