blob: 8a504c742518a59aa43c2e00f95a76e6040de4f7 (
plain)
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
|
#pragma once
#include <stdlib.h>
#include <coro.h>
#include <ev++.h>
#include <Exceptions.h>
#include <vector>
namespace Balau {
class TaskMan;
class Task;
namespace Events {
class BaseEvent {
public:
BaseEvent() : m_signal(false), m_task(NULL) { }
bool gotSignal() { return m_signal; }
void doSignal();
Task * taskWaiting() { Assert(m_task); return m_task; }
void registerOwner(Task * task) { Assert(m_task == NULL); m_task = task; gotOwner(task); }
protected:
virtual void gotOwner(Task * task) { }
private:
bool m_signal;
Task * m_task;
};
class Timeout : public BaseEvent {
public:
Timeout(ev_tstamp tstamp);
void evt_cb(ev::timer & w, int revents);
private:
virtual void gotOwner(Task * task);
ev::timer m_evt;
};
class TaskEvent : public BaseEvent {
public:
TaskEvent(Task * taskWaited);
Task * taskWaited() { return m_taskWaited; }
private:
Task * m_taskWaited;
};
class Custom : public BaseEvent {
public:
void doSignal() { BaseEvent::doSignal(); ev_break(m_loop, EVBREAK_ALL); }
protected:
virtual void gotOwner(Task * task);
private:
struct ev_loop * m_loop;
};
};
class Task {
public:
enum Status {
STARTING,
RUNNING,
IDLE,
STOPPED,
FAULTED,
};
Task();
virtual ~Task();
virtual const char * getName() = 0;
Status getStatus() { return m_status; }
static Task * getCurrentTask();
static void yield(Events::BaseEvent * evt) { Task * t = getCurrentTask(); t->waitFor(evt); t->yield(); }
TaskMan * getTaskMan() { return m_taskMan; }
protected:
void yield();
virtual void Do() = 0;
void waitFor(Events::BaseEvent * event);
private:
size_t stackSize() { return 128 * 1024; }
void switchTo();
static void coroutine(void *);
void * m_stack;
coro_context m_ctx;
TaskMan * m_taskMan;
Status m_status;
void * m_tls;
friend class TaskMan;
friend class Events::TaskEvent;
typedef std::vector<Events::TaskEvent *> waitedByList_t;
waitedByList_t m_waitedBy;
};
};
|
