36 lines
1.0 KiB
C++
36 lines
1.0 KiB
C++
/*
|
|
* Copyright (C) 2024-present ScyllaDB
|
|
*/
|
|
|
|
/*
|
|
* SPDX-License-Identifier: LicenseRef-ScyllaDB-Source-Available-1.0
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <seastar/core/abort_source.hh>
|
|
#include "utils/assert.hh"
|
|
#include "utils/small_vector.hh"
|
|
#include "seastarx.hh"
|
|
|
|
namespace utils {
|
|
// A facility to combine several abort_source-s and expose them as a single abort_source.
|
|
// The combined abort_source is aborted whenever any of the added abort_sources is aborted.
|
|
// Typical use case: there are several sources of abort signal (e.g. timeout and node shutdown)
|
|
// and we what some routine to be aborted on any of them.
|
|
class composite_abort_source {
|
|
abort_source _as;
|
|
utils::small_vector<abort_source::subscription, 2> _subscriptions;
|
|
public:
|
|
void add(abort_source& as) {
|
|
as.check();
|
|
auto sub = as.subscribe([this]() noexcept { _as.request_abort(); });
|
|
SCYLLA_ASSERT(sub);
|
|
_subscriptions.push_back(std::move(*sub));
|
|
}
|
|
abort_source& abort_source() noexcept {
|
|
return _as;
|
|
}
|
|
};
|
|
}
|