fix local ddl (#257)

* Update rocksdb_scan_node.cpp

fix local ddl

* Update ddl_work_planner.cpp

* Update full_export_node.cpp

* Update access_path.cpp

* Update select_manager_node.cpp

* Update schema_factory.cpp

* Update schema_factory.cpp

Author: lgqss

src/common/schema_factory.cpp

@@ -3035,21 +3035,23 @@ int SchemaFactory::get_region_by_key(int64_t main_table_id,
         template_primary.mutable_sort_index()->CopyFrom(primary->sort_index());
     }
     template_primary.mutable_index_conjuncts()->CopyFrom(primary->index_conjuncts());
-    template_primary.set_is_eq(primary->is_eq());
-    template_primary.set_left_field_cnt(primary->left_field_cnt());
-    template_primary.set_right_field_cnt(primary->right_field_cnt());
-    template_primary.set_left_open(primary->left_open());
-    template_primary.set_right_open(primary->right_open());
-    template_primary.set_like_prefix(primary->like_prefix());
+    if (primary->has_left_field_cnt() || primary->has_right_field_cnt()) {
+        template_primary.set_is_eq(primary->is_eq());
+        template_primary.set_left_field_cnt(primary->left_field_cnt());
+        template_primary.set_right_field_cnt(primary->right_field_cnt());
+        template_primary.set_left_open(primary->left_open());
+        template_primary.set_right_open(primary->right_open());
+        template_primary.set_like_prefix(primary->like_prefix());
+    }
 
     std::map<int64_t, std::vector<int>> region_idx_map;
     auto record_template = TableRecord::new_record(main_table_id);
     int range_size = primary->ranges_size();
     for (int i = 0; i < range_size; ++i) {
         const auto& range = primary->ranges(i);
-        bool like_prefix = template_primary.has_like_prefix();
-        bool left_open = template_primary.has_left_open();
-        bool right_open = template_primary.has_right_open();
+        bool like_prefix = template_primary.has_like_prefix() ? template_primary.like_prefix() : range.like_prefix();
+        bool left_open = template_primary.has_left_open() ? template_primary.left_open() : range.left_open();
+        bool right_open = template_primary.has_right_open() ? template_primary.right_open() : range.right_open();
         MutTableKey  start;
         MutTableKey  end;
         if (!range.left_key().empty()) {

src/exec/access_path.cpp

@@ -508,7 +508,7 @@ void AccessPath::calc_index_range(
     pos_index.set_left_open(_left_open);
     pos_index.set_right_open(_right_open);
     if (_left_field_cnt == 0 && _right_field_cnt == 0) {
-        pos_index.clear_is_eq();//无命中条件非eq
+        pos_index.set_is_eq(false);//无命中条件非eq
         pos_index.add_ranges();
     } else if (in_records.size() > 0) {
         is_possible = true;

src/exec/full_export_node.cpp

@@ -105,10 +105,10 @@ int FullExportNode::get_next_region_infos() {
     if (scan_index_info.router_index->ranges_size() == 0) {
         scan_index_info.router_index->add_ranges();
     }
+    scan_index_info.router_index->set_left_field_cnt(index_ptr->fields.size());
+    scan_index_info.router_index->set_left_open(false);
     scan_index_info.router_index->mutable_ranges(0)->set_left_key(_last_router_key);
     scan_index_info.router_index->mutable_ranges(0)->set_left_full(true);
-    scan_index_info.router_index->mutable_ranges(0)->set_left_field_cnt(index_ptr->fields.size());
-    scan_index_info.router_index->mutable_ranges(0)->set_left_open(false);
     scan_index_info.router_index->mutable_ranges(0)->mutable_partition_ids()->Clear();
     scan_index_info.router_index->mutable_ranges(0)->add_partition_ids(_current_partition);
     int ret = schema_factory->get_region_by_key(main_table_id, 
@@ -168,14 +168,15 @@ int FullExportNode::calc_last_key(RuntimeState* state, MemRow* mem_row) {
     pos_index.ParseFromString(scan_index_info.raw_index);
     pos_index.set_index_id(main_table_id);
     // fullexport 非eq
-    pos_index.clear_is_eq();
+    pos_index.set_is_eq(false);
     if (pos_index.ranges_size() == 0) {
         pos_index.add_ranges();
     }
+    pos_index.set_left_field_cnt(pri_info->fields.size());
+    pos_index.set_left_open(true);
     pos_index.mutable_ranges(0)->set_left_key(key.data());
     pos_index.mutable_ranges(0)->set_left_full(key.get_full());
-    pos_index.mutable_ranges(0)->set_left_field_cnt(pri_info->fields.size());
-    pos_index.mutable_ranges(0)->set_left_open(true);
+
     pos_index.mutable_ranges(0)->mutable_partition_ids()->Clear();
     pos_index.mutable_ranges(0)->add_partition_ids(_current_partition);
     pos_index.SerializeToString(&scan_index_info.raw_index);

src/exec/rocksdb_scan_node.cpp

@@ -205,7 +205,7 @@ int RocksdbScanNode::choose_index(RuntimeState* state) {
         check_memory = true;
     }
     bool has_global_param = false;
-    if (pos_index.has_is_eq() && pos_index.has_left_field_cnt()) {
+    if (pos_index.has_left_field_cnt() || pos_index.has_right_field_cnt()) {
         has_global_param = true;
         _is_eq = pos_index.is_eq();
         _like_prefix = pos_index.like_prefix();

src/exec/select_manager_node.cpp

@@ -737,14 +737,12 @@ int SelectManagerNode::construct_primary_possible_index(
                 DB_FATAL("Fail to encode_key left, table:%ld", pri_info->id);
                 return -1;
             }
+            pos_index.set_is_eq(true);
             range->set_left_key(key.data());
             range->set_left_full(key.get_full());
-            range->set_right_key(key.data());
-            range->set_right_full(key.get_full());
-            range->set_left_field_cnt(pri_info->fields.size());
-            range->set_right_field_cnt(pri_info->fields.size());
-            range->set_left_open(false);
-            range->set_right_open(false);
+            pos_index.set_left_field_cnt(pri_info->fields.size());
+            pos_index.set_left_open(false);
+            
             range->add_partition_ids(mem_row->get_partition_id());
             limit_cnt --;
             if(!limit_cnt) {
@@ -869,14 +867,11 @@ int SelectManagerNode::construct_primary_possible_index_vectorize(
             DB_FATAL("Fail to encode_key left, table:%ld", pri_info->id);
             return -1;
         }
+        pos_index.set_is_eq(true);
         range->set_left_key(key.data());
         range->set_left_full(key.get_full());
-        range->set_right_key(key.data());
-        range->set_right_full(key.get_full());
-        range->set_left_field_cnt(pri_info->fields.size());
-        range->set_right_field_cnt(pri_info->fields.size());
-        range->set_left_open(false);
-        range->set_right_open(false);
+        pos_index.set_left_field_cnt(pri_info->fields.size());
+        pos_index.set_left_open(false);
         // [ARROW todo] 全局索引反查partition分发
         // range->set_partition_id(mem_row->get_partition_id());
     }

src/logical_plan/ddl_work_planner.cpp

@@ -211,6 +211,7 @@ std::unique_ptr<ScanNode> DDLWorkPlanner::create_scan_node() {
     _pos_index.Clear();
     _pos_index.set_index_id(_table_id);
     auto range_index = _pos_index.add_ranges();
+    _pos_index.set_is_eq(false);
     _pos_index.set_left_field_cnt(_field_num);
     _pos_index.set_left_open(true);
     if (_start_key != "") {