diff --git a/be/src/agent/task_worker_pool.cpp b/be/src/agent/task_worker_pool.cpp index 6e81a3efe0eb3d..6b4239ccb546ad 100644 --- a/be/src/agent/task_worker_pool.cpp +++ b/be/src/agent/task_worker_pool.cpp @@ -1324,12 +1324,14 @@ void download_callback(StorageEngine& engine, ExecEnv* env, const TAgentTaskRequ if (download_request.__isset.remote_tablet_snapshots) { std::unique_ptr loader = std::make_unique( engine, env, download_request.job_id, req.signature); + SCOPED_ATTACH_TASK(loader->resource_ctx()); status = loader->remote_http_download(download_request.remote_tablet_snapshots, &downloaded_tablet_ids); } else { std::unique_ptr loader = std::make_unique( engine, env, download_request.job_id, req.signature, download_request.broker_addr, download_request.broker_prop); + SCOPED_ATTACH_TASK(loader->resource_ctx()); status = loader->init(download_request.__isset.storage_backend ? download_request.storage_backend : TStorageBackendType::type::BROKER, @@ -1377,6 +1379,7 @@ void download_callback(CloudStorageEngine& engine, ExecEnv* env, const TAgentTas std::unique_ptr loader = std::make_unique( engine, env, download_request.job_id, req.signature, download_request.broker_addr, download_request.broker_prop); + SCOPED_ATTACH_TASK(loader->resource_ctx()); status = loader->init(download_request.__isset.storage_backend ? download_request.storage_backend : TStorageBackendType::type::BROKER, @@ -1530,6 +1533,7 @@ void move_dir_callback(StorageEngine& engine, ExecEnv* env, const TAgentTaskRequ status = Status::InvalidArgument("Could not find tablet"); } else { SnapshotLoader loader(engine, env, move_dir_req.job_id, move_dir_req.tablet_id); + SCOPED_ATTACH_TASK(loader.resource_ctx()); status = loader.move(move_dir_req.src, tablet, true); }