blob: 6073b72e62205fcf732b38462e74ac0c5b0df691 (
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
|
#include "cont_poller.h"
#include "impl.h"
namespace NCoro {
namespace {
template <class T>
int DoExecuteEvent(T* event) noexcept {
auto* cont = event->Cont();
if (cont->Cancelled()) {
return ECANCELED;
}
cont->Executor()->ScheduleIoWait(event);
cont->Switch();
if (cont->Cancelled()) {
return ECANCELED;
}
return event->Status();
}
}
void TContPollEvent::Wake() noexcept {
UnLink();
Cont()->ReSchedule();
}
TInstant TEventWaitQueue::WakeTimedout(TInstant now) noexcept {
TIoWait::TIterator it = IoWait_.Begin();
if (it != IoWait_.End()) {
if (it->DeadLine() > now) {
return it->DeadLine();
}
do {
(it++)->Wake(ETIMEDOUT);
} while (it != IoWait_.End() && it->DeadLine() <= now);
}
return now;
}
void TEventWaitQueue::Register(NCoro::TContPollEvent* event) {
IoWait_.Insert(event);
event->Cont()->Unlink();
}
void TEventWaitQueue::Abort() noexcept {
auto visitor = [](TContPollEvent& e) {
e.Cont()->Cancel();
};
IoWait_.ForEach(visitor);
}
}
void TFdEvent::RemoveFromIOWait() noexcept {
this->Cont()->Executor()->Poller()->Remove(this);
}
int ExecuteEvent(TFdEvent* event) noexcept {
return NCoro::DoExecuteEvent(event);
}
int ExecuteEvent(TTimerEvent* event) noexcept {
return NCoro::DoExecuteEvent(event);
}
|