1

be/src/service/backend_service.cpp

@@ -50,6 +50,7 @@
 #include "olap/rowset/rowset_meta.h"
 #include "olap/storage_engine.h"
 #include "olap/tablet_manager.h"
+#include "olap/tablet_meta.h"
 #include "olap/txn_manager.h"
 #include "runtime/exec_env.h"
 #include "runtime/external_scan_context_mgr.h"
@@ -648,6 +649,28 @@ void BackendService::ingest_binlog(TIngestBinlogResult& result,
         return;
     }
 
+    auto calc_delete_bitmap_token =
+            StorageEngine::instance()->calc_delete_bitmap_executor()->create_token();
+    DeleteBitmapPtr delete_bitmap = std::make_shared<DeleteBitmap>(local_tablet_id);
+    auto pre_rowset_ids = local_tablet->all_rs_id(local_tablet->max_version_unlocked().second);
+    if (local_tablet->enable_unique_key_merge_on_write()) {
+        auto beta_rowset = reinterpret_cast<BetaRowset*>(rowset.get());
+        std::vector<segment_v2::SegmentSharedPtr> segments;
+        RETURN_IF_ERROR(beta_rowset->load_segments(&segments));
+        if (segments.size() > 1) {
+            // calculate delete bitmap between segments
+            RETURN_IF_ERROR(local_tablet->calc_delete_bitmap_between_segments(rowset, segments,
+                                                                              delete_bitmap));
+        }
+
+        local_tablet->commit_phase_update_delete_bitmap(rowset, pre_rowset_ids, delete_bitmap,
+                                                        segments, txn_id,
+                                                        calc_delete_bitmap_token.get(), nullptr);
+    }
+
+    calc_delete_bitmap_token->wait();
+    calc_delete_bitmap_token->get_delete_bitmap(delete_bitmap);
+
     // Step 6.2: commit txn
     Status commit_txn_status = StorageEngine::instance()->txn_manager()->commit_txn(
             local_tablet->data_dir()->get_meta(), rowset_meta->partition_id(),
@@ -663,6 +686,12 @@ void BackendService::ingest_binlog(TIngestBinlogResult& result,
         return;
     }
 
+    if (local_tablet->enable_unique_key_merge_on_write()) {
+        StorageEngine::instance()->txn_manager()->set_txn_related_delete_bitmap(
+                partition_id, txn_id, local_tablet_id, local_tablet->schema_hash(),
+                local_tablet->tablet_uid(), true, delete_bitmap, pre_rowset_ids);
+    }
+
     tstatus.__set_status_code(TStatusCode::OK);
 }
 } // namespace doris