-
Notifications
You must be signed in to change notification settings - Fork 0
/
Acceptor.cpp
85 lines (78 loc) · 2.29 KB
/
Acceptor.cpp
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
#include <cerrno>
#include <fcntl.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include "Acceptor.h"
#include "EventLoop.h"
#include "InetAddress.h"
#include "Logger.h"
using namespace simple_muduo;
static int createNonblocking()
{
int sockfd = ::socket(AF_INET, SOCK_STREAM | SOCK_NONBLOCK | SOCK_CLOEXEC, IPPROTO_TCP);
if(sockfd < 0)
{
LOG_ERROR("listen socket create err:%d\n",errno);
}
return sockfd;
}
Acceptor::Acceptor(EventLoop* loop, const InetAddress &listenAddr, bool reuseport)
: loop_(loop)
, acceptSocket_(createNonblocking())
, acceptChannel_(loop, acceptSocket_.fd())
, listenning_(false)
, idleFd_(::open("dev/null", O_RDONLY|O_CLOEXEC))
{
acceptSocket_.setReuseAddr(true);
acceptSocket_.setReusePort(true);
acceptSocket_.bindAddress(listenAddr);
// TcpServer::start() => Acceptor.listen() 如果有新用户连接
// 要执行一个回调(accept => connfd => 打包成Channel => 唤醒subloop)
// baseloop监听到有事件发生 => acceptChannel_(listenfd) => 执行该回调函数
acceptChannel_.setReadCallback(
std::bind(&Acceptor::handleRead, this));
}
Acceptor::~Acceptor()
{
// 把从Poller中感兴趣的事件删除掉
acceptChannel_.disableAll();
// 调用EventLoop->removeChannel => Poller->removeChannel 把Poller的ChannelMap对应的部分删除
acceptChannel_.remove();
}
void Acceptor::listen()
{
listenning_ = true;
// 监听
acceptSocket_.listen();
// acceptChannel_注册至Poller
acceptChannel_.enableReading();
}
// listenfd有事件发生了,就是有新用户连接了
// handleRead()回调执行了,就是listenfd上有事件发生了
void Acceptor::handleRead()
{
InetAddress peerAddr;
int connfd = acceptSocket_.accept(&peerAddr);
if (connfd >= 0)
{
if (NewConnectionCallback_)
{
NewConnectionCallback_(connfd, peerAddr); // 轮询找到subLoop 唤醒并分发当前的新客户端的Channel
}
else
{
::close(connfd);
}
}
else
{
// to many open file 文件描述符过多,一般设置是1024个,ulimit -a 可以查看
if (errno == EMFILE)
{
::close(idleFd_);
idleFd_ = ::accept(acceptSocket_.fd(), NULL, NULL);
::close(idleFd_);
idleFd_ = ::open("/dev/null", O_RDONLY | O_CLOEXEC);
}
}
}