diff --git a/be/src/agent/task_worker_pool.cpp b/be/src/agent/task_worker_pool.cpp index 330f3e916e..3d398d6425 100644 --- a/be/src/agent/task_worker_pool.cpp +++ b/be/src/agent/task_worker_pool.cpp @@ -1735,6 +1735,7 @@ void storage_medium_migrate_callback(StorageEngine& engine, const TAgentTaskRequ auto status = check_migrate_request(engine, storage_medium_migrate_req, tablet, &dest_store); if (status.ok()) { EngineStorageMigrationTask engine_task(tablet, dest_store); + SCOPED_ATTACH_TASK(engine_task.mem_tracker()); status = engine_task.execute(); } // fe should ignore this err diff --git a/be/src/http/action/tablet_migration_action.cpp b/be/src/http/action/tablet_migration_action.cpp index 241f63c61a..5558a34bfd 100644 --- a/be/src/http/action/tablet_migration_action.cpp +++ b/be/src/http/action/tablet_migration_action.cpp @@ -224,6 +224,7 @@ Status TabletMigrationAction::_execute_tablet_migration(TabletSharedPtr tablet, int32_t schema_hash = tablet->schema_hash(); string dest_disk = dest_store->path(); EngineStorageMigrationTask engine_task(tablet, dest_store); + SCOPED_ATTACH_TASK(engine_task.mem_tracker()); Status res = engine_task.execute(); if (!res.ok()) { LOG(WARNING) << "tablet migrate failed. tablet_id=" << tablet_id diff --git a/be/src/olap/snapshot_manager.h b/be/src/olap/snapshot_manager.h index 4f06c3c87c..9781d4f69e 100644 --- a/be/src/olap/snapshot_manager.h +++ b/be/src/olap/snapshot_manager.h @@ -63,7 +63,8 @@ public: private: SnapshotManager() : _snapshot_base_id(0) { - _mem_tracker = MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::OTHER, "SnapshotManager"); + _mem_tracker = + MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::OTHER, "SnapshotManager"); } Status _calc_snapshot_id_path(const TabletSharedPtr& tablet, int64_t timeout_s, diff --git a/be/src/olap/task/engine_storage_migration_task.cpp b/be/src/olap/task/engine_storage_migration_task.cpp index f9227a903f..4aa27e4523 100644 --- a/be/src/olap/task/engine_storage_migration_task.cpp +++ b/be/src/olap/task/engine_storage_migration_task.cpp @@ -53,6 +53,8 @@ EngineStorageMigrationTask::EngineStorageMigrationTask(const TabletSharedPtr& ta DataDir* dest_store) : _tablet(tablet), _dest_store(dest_store) { _task_start_time = time(nullptr); + _mem_tracker = MemTrackerLimiter::create_shared(MemTrackerLimiter::Type::OTHER, + "EngineStorageMigrationTask"); } Status EngineStorageMigrationTask::execute() {