blob: 677ade2fffa54a10554681941a925ee8bc4ee892 (
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
|
#pragma once
#include <util/generic/object_counter.h>
#include <util/generic/ptr.h>
#include <util/network/init.h>
#include <util/network/socket.h>
namespace NEventLoop {
struct IEventHandler
: public TAtomicRefCount<IEventHandler> {
virtual void HandleEvent(SOCKET socket, void* cookie) = 0;
virtual ~IEventHandler() {
}
};
typedef TIntrusivePtr<IEventHandler> TEventHandlerPtr;
class TEventLoop;
// TODO: make TChannel itself a pointer
// to avoid confusion with Drop and Unregister
class TChannel
: public TAtomicRefCount<TChannel> {
public:
~TChannel();
void EnableRead();
void DisableRead();
void EnableWrite();
void DisableWrite();
void Unregister();
SOCKET GetSocket() const;
TSocket GetSocketPtr() const;
private:
class TImpl;
friend class TEventLoop;
TObjectCounter<TChannel> ObjectCounter;
TChannel(TImpl*);
private:
THolder<TImpl> Impl;
};
typedef TIntrusivePtr<TChannel> TChannelPtr;
class TEventLoop {
public:
TEventLoop(const char* name = nullptr);
~TEventLoop();
void Run();
void Stop();
bool IsRunning();
TChannelPtr Register(TSocket socket, TEventHandlerPtr, void* cookie = nullptr);
private:
class TImpl;
friend class TChannel;
TObjectCounter<TEventLoop> ObjectCounter;
private:
THolder<TImpl> Impl;
};
}
|
