blob: 2e879999bfd250e571dfe2514b4afdfe9bffd0ed (
plain) (
blame)
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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
|
#pragma once
#include "impl.h"
#include <util/datetime/base.h>
class TContEvent {
public:
TContEvent(TCont* current) noexcept
: Cont_(current)
, Status_(0)
{
}
~TContEvent() {
}
int WaitD(TInstant deadline) {
Status_ = 0;
const int ret = Cont_->SleepD(deadline);
return Status_ ? Status_ : ret;
}
int WaitT(TDuration timeout) {
return WaitD(timeout.ToDeadLine());
}
int WaitI() {
return WaitD(TInstant::Max());
}
void Wake() noexcept {
SetStatus(EWAKEDUP);
Cont_->ReSchedule();
}
TCont* Cont() noexcept {
return Cont_;
}
int Status() const noexcept {
return Status_;
}
void SetStatus(int status) noexcept {
Status_ = status;
}
private:
TCont* Cont_;
int Status_;
};
class TContWaitQueue {
class TWaiter: public TContEvent, public TIntrusiveListItem<TWaiter> {
public:
TWaiter(TCont* current) noexcept
: TContEvent(current)
{
}
~TWaiter() {
}
};
public:
TContWaitQueue() noexcept {
}
~TContWaitQueue() {
Y_ASSERT(Waiters_.Empty());
}
int WaitD(TCont* current, TInstant deadline) {
TWaiter waiter(current);
Waiters_.PushBack(&waiter);
return waiter.WaitD(deadline);
}
int WaitT(TCont* current, TDuration timeout) {
return WaitD(current, timeout.ToDeadLine());
}
int WaitI(TCont* current) {
return WaitD(current, TInstant::Max());
}
void Signal() noexcept {
if (!Waiters_.Empty()) {
Waiters_.PopFront()->Wake();
}
}
void BroadCast() noexcept {
while (!Waiters_.Empty()) {
Waiters_.PopFront()->Wake();
}
}
void BroadCast(size_t number) noexcept {
for (size_t i = 0; i < number && !Waiters_.Empty(); ++i) {
Waiters_.PopFront()->Wake();
}
}
private:
TIntrusiveList<TWaiter> Waiters_;
};
class TContSimpleEvent {
public:
TContSimpleEvent(TContExecutor* e)
: E_(e)
{
}
TContExecutor* Executor() const noexcept {
return E_;
}
void Signal() noexcept {
Q_.Signal();
}
void BroadCast() noexcept {
Q_.BroadCast();
}
int WaitD(TInstant deadLine) noexcept {
return Q_.WaitD(E_->Running(), deadLine);
}
int WaitT(TDuration timeout) noexcept {
return WaitD(timeout.ToDeadLine());
}
int WaitI() noexcept {
return WaitD(TInstant::Max());
}
private:
TContWaitQueue Q_;
TContExecutor* E_;
};
|
