55fde70f8d
task_status contains a vector of children identities. If the number of children is large, we may hit oversized allocation. Change all types of children-related containers to chunked_vector. Modify the children type returned from task manager API. Fixes: scylladb#25795. Closes scylladb/scylladb#25923
89 lines
2.0 KiB
C++
89 lines
2.0 KiB
C++
/*
|
|
* Copyright (C) 2024-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include "gms/inet_address.hh"
|
|
#include "tasks/task_manager.hh"
|
|
#include "utils/chunked_vector.hh"
|
|
|
|
namespace tasks {
|
|
|
|
struct task_identity {
|
|
locator::host_id host_id;
|
|
task_id task_id;
|
|
};
|
|
|
|
struct task_status {
|
|
tasks::task_id task_id;
|
|
std::string type;
|
|
task_kind kind;
|
|
std::string scope;
|
|
task_manager::task_state state;
|
|
is_abortable is_abortable;
|
|
db_clock::time_point start_time;
|
|
db_clock::time_point end_time;
|
|
std::string error;
|
|
tasks::task_id parent_id;
|
|
uint64_t sequence_number;
|
|
unsigned shard;
|
|
std::string keyspace;
|
|
std::string table;
|
|
std::string entity;
|
|
std::string progress_units;
|
|
task_manager::task::progress progress;
|
|
utils::chunked_vector<task_identity> children;
|
|
};
|
|
|
|
struct task_stats {
|
|
tasks::task_id task_id;
|
|
std::string type;
|
|
task_kind kind;
|
|
std::string scope;
|
|
task_manager::task_state state;
|
|
uint64_t sequence_number;
|
|
std::string keyspace;
|
|
std::string table;
|
|
std::string entity;
|
|
unsigned shard;
|
|
db_clock::time_point start_time;
|
|
db_clock::time_point end_time;
|
|
};
|
|
|
|
struct status_helper;
|
|
|
|
class task_handler {
|
|
private:
|
|
task_manager& _tm;
|
|
task_id _id;
|
|
public:
|
|
explicit task_handler(task_manager& tm, task_id id) noexcept
|
|
: _tm(tm)
|
|
, _id(id)
|
|
{}
|
|
future<task_stats> get_stats();
|
|
future<task_status> get_status();
|
|
future<task_status> wait_for_task(std::optional<std::chrono::seconds> timeout);
|
|
future<utils::chunked_vector<task_status>> get_status_recursively(bool local);
|
|
future<> abort();
|
|
private:
|
|
future<status_helper> get_status_helper();
|
|
};
|
|
|
|
class task_not_abortable : public std::exception {
|
|
sstring _cause;
|
|
public:
|
|
explicit task_not_abortable(task_id tid)
|
|
: _cause(format("task with id {} cannot be aborted", tid))
|
|
{ }
|
|
|
|
virtual const char* what() const noexcept override { return _cause.c_str(); }
|
|
};
|
|
|
|
}
|