BplusTreeIndex::insert_entry独立性检验

HuXin0817 · Oct 3, 2024 · 3da8111 · 3da8111
1 parent f4f156f
commit 3da8111
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/observer/storage/index/bplus_tree_index.cpp b/src/observer/storage/index/bplus_tree_index.cpp
@@ -83,6 +83,19 @@ RC BplusTreeIndex::close()
 RC BplusTreeIndex::insert_entry(const char *record, const RID *rid)
 {
   char *entry = index_meta_.make_entry_from_record(record);
+  if (index_meta_.unique()) {
+    IndexScanner *scanner = create_scanner(entry, index_meta_.fields_total_len(), true, entry, index_meta_.fields_total_len(), true);
+    if (scanner != nullptr) {
+      RID existing_rid;
+      if (scanner->next_entry(&existing_rid) == RC::SUCCESS) {
+        delete scanner;
+        LOG_WARN("Key already exists, duplicate entry is not allowed.");
+        return RC::RECORD_DUPLICATE_KEY;
+      }
+      delete scanner;
+    }
+  }
+
   return index_handler_.insert_entry(entry, rid);
 }