patch 4.0

unittest/sql/optimizer/test_explain_plan_oracle_style.cpp

@@ -0,0 +1,99 @@
+/**
+ * Copyright (c) 2021 OceanBase
+ * OceanBase CE is licensed under Mulan PubL v2.
+ * You can use this software according to the terms and conditions of the Mulan PubL v2.
+ * You may obtain a copy of Mulan PubL v2 at:
+ *          http://license.coscl.org.cn/MulanPubL-2.0
+ * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
+ * EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
+ * MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
+ * See the Mulan PubL v2 for more details.
+ */
+
+#include "sql/optimizer/ob_log_plan.h"
+#include <gtest/gtest.h>
+
+
+#define BUF_LEN 102400 // 100K
+
+using namespace oceanbase::sql;
+using namespace oceanbase::common;
+
+class ObLogPlanTest: public ::testing::Test
+{
+  public:
+	  ObLogPlanTest();
+    virtual ~ObLogPlanTest();
+    virtual void SetUp();
+    virtual void TearDown();
+  private:
+    // disallow copy
+    ObLogPlanTest(const ObLogPlanTest &other);
+    ObLogPlanTest& operator=(const ObLogPlanTest &other);
+  private:
+    // data members
+};
+
+ObLogPlanTest::ObLogPlanTest()
+{
+}
+
+ObLogPlanTest::~ObLogPlanTest()
+{
+}
+
+void ObLogPlanTest::SetUp()
+{
+}
+
+void ObLogPlanTest::TearDown()
+{
+}
+
+TEST_F(ObLogPlanTest, ob_explain_test)
+{
+
+  // create a plan
+  ObSQLSessionInfo session_info
+  ASSERT_EQ(OB_SUCCESS, session_info.test_init(0, 0, 0, NULL));
+  ObOptimizerContext ctx(NULL, NULL);
+  ObSelectStmt *select_stmt = NULL;
+	ObSelectLogPlan plan(ctx, select_stmt);
+
+  // create the operator tree
+  ObLogTableScan *left =
+    reinterpret_cast<ObLogTableScan *>(log_op_factory_.allocate(LOG_TABLE_SCAN));
+  ObLogTableScan *right =
+    reinterpret_cast<ObLogTableScan *>(log_op_factory_.allocate(LOG_TABLE_SCAN));
+  ObLogJoin *join =
+    reinterpret_cast<ObLogJoin *>(log_op_factory_.allocate(LOG_JOIN));
+  ObLogGroupBy *group_by =
+    reinterpret_cast<ObLogGroupBy *>(log_op_factory_.allocate(LOG_GROUP_BY));
+  ObLogOrderBy *order_by =
+    reinterpret_cast<ObLogOrderBy *>(log_op_factory_.allocate(LOG_ORDER_BY));
+  ObLogLimit *limit =
+    reinterpret_cast<ObLogLimit *>(log_op_factory_.allocate(LOG_LIMIT));
+
+  ASSERT_EQ(OB_SUCCESS, plan.set_plan_root(limit));
+  ASSERT_EQ(OB_SUCCESS, limit.set_child(first_child, order_by));
+  ASSERT_EQ(OB_SUCCESS, order_by.set_parent(limit));
+  ASSERT_EQ(OB_SUCCESS, order_by->set_child(first_child, group_by));
+  ASSERT_EQ(OB_SUCCESS, group_by->set_parent(order_by));
+  ASSERT_EQ(OB_SUCCESS, group_by->set_child(first_child, join));
+  ASSERT_EQ(OB_SUCCESS, join->set_parent(group_by));
+  ASSERT_EQ(OB_SUCCESS, join->set_child(first_child, left));
+  ASSERT_EQ(OB_SUCCESS, left->set_parent(join));
+  ASSERT_EQ(OB_SUCCESS, join->set_child(second_child, right));
+  ASSERT_EQ(OB_SUCCESS, right->set_parent(join));
+
+  // print a plan
+  char buf[BUF_LEN];
+	ASSERT_EQ(OB_SUCCESS,	plan.to_string(buf, BUF_LEN));
+	printf("%s\n", buf);
+}
+
+int main(int argc, char **argv)
+{
+  ::testing::InitGoogleTest(&argc,argv);
+  return RUN_ALL_TESTS();
+}