1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
|
#pragma once
#if !defined(INCLUDE_LIBRARY_THREADING_FUTURE_WAIT_ALL_OR_EXCEPTION_INL_H)
#error "you should never include wait_all_or_exception_inl.h directly"
#endif
#include "subscription.h"
#include <initializer_list>
namespace NThreading::NWait {
namespace NPrivate {
class TWaitAllOrException final : public NThreading::NPrivate::TWait<TWaitAllOrException>
{
private:
size_t Count = 0;
static constexpr bool RevertOnSignaled = false;
using TBase = NThreading::NPrivate::TWait<TWaitAllOrException>;
friend TBase;
private:
TWaitAllOrException(TSubscriptionManagerPtr manager)
: TBase(std::move(manager))
, Count(0)
{
}
template <typename TFutures>
void BeforeSubscribe(TFutures const& futures) {
Count = std::size(futures);
Y_ENSURE(Count > 0, "It is meaningless to use this class with empty futures set");
}
template <typename T>
void Set(TFuture<T> const& future) {
with_lock (TBase::Lock) {
try {
future.TryRethrow();
if (--Count == 0) {
// there is no need to call Unsubscribe here since all futures are signaled
TBase::Promise.SetValue();
}
} catch (...) {
Y_ASSERT(!TBase::Promise.HasValue());
TBase::Unsubscribe();
if (!TBase::Promise.HasException()) {
TBase::Promise.SetException(std::current_exception());
}
}
}
}
};
}
template <typename TFutures, typename TCallbackExecutor = NThreading::NPrivate::TNoexceptExecutor>
TFuture<void> WaitAllOrException(TFutures const& futures, TSubscriptionManagerPtr manager, TCallbackExecutor&& executor) {
return NThreading::NPrivate::Wait<NPrivate::TWaitAllOrException>(futures, std::move(manager), std::forward<TCallbackExecutor>(executor));
}
template <typename T, typename TCallbackExecutor = NThreading::NPrivate::TNoexceptExecutor>
TFuture<void> WaitAllOrException(std::initializer_list<TFuture<T> const> futures, TSubscriptionManagerPtr manager
, TCallbackExecutor&& executor)
{
return NThreading::NPrivate::Wait<NPrivate::TWaitAllOrException>(futures, std::move(manager), std::forward<TCallbackExecutor>(executor));
}
template <typename T, typename TCallbackExecutor = NThreading::NPrivate::TNoexceptExecutor>
TFuture<void> WaitAllOrException(TFuture<T> const& future1, TFuture<T> const& future2, TSubscriptionManagerPtr manager
, TCallbackExecutor&& executor)
{
return NThreading::NPrivate::Wait<NPrivate::TWaitAllOrException>(future1, future2, std::move(manager)
, std::forward<TCallbackExecutor>(executor));
}
}
|
