Rj/col pruning (#704)

* col prune prep files

* add s24 tas :)

Author: ChaosZhai

CMakeLists.txt

@@ -351,6 +351,7 @@ set(P3_FILES
         "src/optimizer/sort_limit_as_topn.cpp"
         "src/optimizer/optimizer_internal.cpp"
         "src/optimizer/seqscan_as_indexscan.cpp"
+        "src/optimizer/column_pruning.cpp"
         "src/common/bustub_ddl.cpp"
         "src/include/execution/plans/topn_per_group_plan.h"
         ${P2_FILES}

src/common/util/string_util.cpp

@@ -42,6 +42,11 @@ void StringUtil::RTrim(std::string *str) {
   str->erase(std::find_if(str->rbegin(), str->rend(), [](int ch) { return std::isspace(ch) == 0; }).base(), str->end());
 }
 
+void StringUtil::LTrim(std::string *str) {
+  // remove leading ' ', \f, \n, \r, \t, \v
+  str->erase(str->begin(), std::find_if(str->begin(), str->end(), [](int ch) { return std::isspace(ch) == 0; }));
+}
+
 auto StringUtil::Indent(int num_indent) -> std::string { return std::string(num_indent, ' '); }  // NOLINT
 
 auto StringUtil::StartsWith(const std::string &str, const std::string &prefix) -> bool {
@@ -200,6 +205,16 @@ auto StringUtil::Split(const std::string &input, const std::string &split) -> st
   return splits;
 }
 
+auto StringUtil::Count(const std::string &input, const std::string &str) -> size_t {
+  size_t count = 0;
+  size_t n_pos = input.find(str, 0);  // first occurrence
+  while (n_pos != std::string::npos) {
+    count++;
+    n_pos = input.find(str, n_pos + 1);
+  }
+  return count;
+}
+
 auto StringUtil::Strip(const std::string &str, char c) -> std::string {
   // There's a copy here which is wasteful, so don't use this in performance-critical code!
   std::string tmp = str;

src/execution/mock_scan_executor.cpp

@@ -33,6 +33,9 @@ static const char *ta_list_2023_fall[] = {"skyzh",     "yliang412",       "ferna
                                           "anurag-23", "Mayank-Baranwal", "abigalekim",    "ChaosZhai",
                                           "aoleizhou", "averyqi115",      "kswim8"};
 
+static const char *ta_list_2024[] = {"AlSchlo",   "walkingcabbages", "averyqi115", "lanlou1554", "sweetsuro",
+                                     "ChaosZhai", "SDTheSlayer",     "xx01cyx",    "yliang412",  "thelongmarch-azx"};
+
 static const char *ta_oh_2022[] = {"Tuesday",   "Wednesday", "Monday",  "Wednesday", "Thursday", "Friday",
                                    "Wednesday", "Randomly",  "Tuesday", "Monday",    "Tuesday"};
 
@@ -42,12 +45,15 @@ static const char *ta_oh_2023[] = {"Friday",  "Thursday", "Tuesday",   "Monday",
 static const char *ta_oh_2023_fall[] = {"Randomly", "Tuesday",   "Wednesday", "Tuesday", "Thursday", "Tuesday",
                                         "Friday",   "Yesterday", "Friday",    "Friday",  "Never"};
 
+static const char *ta_oh_2024[] = {"Friday",    "Thursday", "Friday",  "Wednesday", "Thursday",
+                                   "Yesterday", "Monday",   "Tuesday", "Tuesday",   "Monday"};
+
 static const char *course_on_date[] = {"Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"};
 
 const char *mock_table_list[] = {"__mock_table_1", "__mock_table_2", "__mock_table_3", "__mock_table_tas_2022",
-                                 "__mock_table_tas_2023", "__mock_table_tas_2023_fall", "__mock_agg_input_small",
-                                 "__mock_agg_input_big", "__mock_table_schedule_2022", "__mock_table_schedule_2023",
-                                 "__mock_table_123", "__mock_graph",
+                                 "__mock_table_tas_2023", "__mock_table_tas_2023_fall", "__mock_table_tas_2024",
+                                 "__mock_agg_input_small", "__mock_agg_input_big", "__mock_table_schedule_2022",
+                                 "__mock_table_schedule", "__mock_table_123", "__mock_graph",
                                  // For leaderboard Q1
                                  "__mock_t1",
                                  // For leaderboard Q2
@@ -84,11 +90,15 @@ auto GetMockTableSchemaOf(const std::string &table) -> Schema {
     return Schema{std::vector{Column{"github_id", TypeId::VARCHAR, 128}, Column{"office_hour", TypeId::VARCHAR, 128}}};
   }
 
+  if (table == "__mock_table_tas_2024") {
+    return Schema{std::vector{Column{"github_id", TypeId::VARCHAR, 128}, Column{"office_hour", TypeId::VARCHAR, 128}}};
+  }
+
   if (table == "__mock_table_schedule_2022") {
     return Schema{std::vector{Column{"day_of_week", TypeId::VARCHAR, 128}, Column{"has_lecture", TypeId::INTEGER}}};
   }
 
-  if (table == "__mock_table_schedule_2023") {
+  if (table == "__mock_table_schedule") {
     return Schema{std::vector{Column{"day_of_week", TypeId::VARCHAR, 128}, Column{"has_lecture", TypeId::INTEGER}}};
   }
 
@@ -168,11 +178,15 @@ auto GetSizeOf(const MockScanPlanNode *plan) -> size_t {
     return sizeof(ta_list_2023_fall) / sizeof(ta_list_2023_fall[0]);
   }
 
+  if (table == "__mock_table_tas_2024") {
+    return sizeof(ta_list_2024) / sizeof(ta_list_2024[0]);
+  }
+
   if (table == "__mock_table_schedule_2022") {
     return sizeof(course_on_date) / sizeof(course_on_date[0]);
   }
 
-  if (table == "__mock_table_schedule_2023") {
+  if (table == "__mock_table_schedule") {
     return sizeof(course_on_date) / sizeof(course_on_date[0]);
   }
 
@@ -306,6 +320,15 @@ auto GetFunctionOf(const MockScanPlanNode *plan) -> std::function<Tuple(size_t)>
     };
   }
 
+  if (table == "__mock_table_tas_2024") {
+    return [plan](size_t cursor) {
+      std::vector<Value> values{};
+      values.push_back(ValueFactory::GetVarcharValue(ta_list_2024[cursor]));
+      values.push_back(ValueFactory::GetVarcharValue(ta_oh_2024[cursor]));
+      return Tuple{values, &plan->OutputSchema()};
+    };
+  }
+
   if (table == "__mock_table_schedule_2022") {
     return [plan](size_t cursor) {
       std::vector<Value> values{};
@@ -315,7 +338,7 @@ auto GetFunctionOf(const MockScanPlanNode *plan) -> std::function<Tuple(size_t)>
     };
   }
 
-  if (table == "__mock_table_schedule_2023") {
+  if (table == "__mock_table_schedule") {
     return [plan](size_t cursor) {
       std::vector<Value> values{};
       values.push_back(ValueFactory::GetVarcharValue(course_on_date[cursor]));

src/include/common/util/string_util.h

@@ -85,12 +85,21 @@ class StringUtil {
   /** @return input string split based on the split string */
   static auto Split(const std::string &input, const std::string &split) -> std::vector<std::string>;
 
+  /** @return count occurrence of specified string in input string */
+  static auto Count(const std::string &input, const std::string &str) -> size_t;
+
   /**
    * Removes the whitespace characters from the right side of the string.
    * @param[in,out] str string to be trimmed on the right
    */
   static void RTrim(std::string *str);
 
+  /**
+   * Removes the whitespace characters from the left side of the string.
+   * @param[in,out] str string to be trimmed on the left
+   */
+  static void LTrim(std::string *str);
+
   /** @return indented string */
   static auto Indent(int num_indent) -> std::string;
 

src/include/optimizer/optimizer.h

@@ -93,6 +93,13 @@ class Optimizer {
   auto MatchIndex(const std::string &table_name, uint32_t index_key_idx)
       -> std::optional<std::tuple<index_oid_t, std::string>>;
 
+  /**
+   * @brief column pruning for child plan following a projection plan
+   * @param plan the plan to optimize
+   * @return the new plan with column pruning
+   */
+  auto OptimizeColumnPruning(const AbstractPlanNodeRef &plan) -> AbstractPlanNodeRef;
+
   /**
    * @brief optimize sort + limit as top N
    */

src/optimizer/CMakeLists.txt

@@ -12,7 +12,8 @@ add_library(
         optimizer_internal.cpp
         order_by_index_scan.cpp
         sort_limit_as_topn.cpp
-        seqscan_as_indexscan.cpp)
+        seqscan_as_indexscan.cpp
+        column_pruning.cpp)
 
 set(ALL_OBJECT_FILES
         ${ALL_OBJECT_FILES} $<TARGET_OBJECTS:bustub_optimizer>

src/optimizer/column_pruning.cpp

@@ -0,0 +1,13 @@
+#include "optimizer/optimizer.h"
+
+namespace bustub {
+
+/**
+ * @note You may use this function to implement column pruning optimization.
+ */
+auto Optimizer::OptimizeColumnPruning(const bustub::AbstractPlanNodeRef &plan) -> AbstractPlanNodeRef {
+  // Your code here
+  return plan;
+}
+
+}  // namespace bustub

test/sql/p0.03-string-scan.slt

@@ -1,5 +1,5 @@
 query rowsort
-select day_of_week, upper(day_of_week), lower(day_of_week), has_lecture from __mock_table_schedule_2023;
+select day_of_week, upper(day_of_week), lower(day_of_week), has_lecture from __mock_table_schedule;
 ----
 Monday MONDAY monday 1 
 Tuesday TUESDAY tuesday 0 

test/sql/p3.00-primer.slt

@@ -1,14 +1,13 @@
 query rowsort
-select github_id, office_hour from __mock_table_tas_2023_fall;
+select github_id, office_hour from __mock_table_tas_2024;
 ----
-skyzh Randomly
-yliang412 Tuesday
-fernandolis10 Wednesday
-wiam8 Tuesday
-anurag-23 Thursday
-Mayank-Baranwal Tuesday
-abigalekim Friday
-ChaosZhai Yesterday
-aoleizhou Friday
+AlSchlo Friday
+walkingcabbages Thursday
 averyqi115 Friday
-kswim8 Never
+lanlou1554 Wednesday
+sweetsuro Thursday
+ChaosZhai Yesterday
+SDTheSlayer Monday
+xx01cyx Tuesday
+yliang412 Tuesday
+thelongmarch-azx Monday

test/sql/p3.07-simple-agg.slt

@@ -1,10 +1,10 @@
 # 4 pts
 
-# How many TAs are there in 2023 Fall?
+# How many TAs are there in 2024 Spring?
 query
-select count(*) from __mock_table_tas_2023_fall;
+select count(*) from __mock_table_tas_2024;
 ----
-11
+10
 
 # The real test process begins...