forked from youcaiguai/NetServer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathPoller.cpp
More file actions
153 lines (138 loc) · 3.91 KB
/
Copy pathPoller.cpp
File metadata and controls
153 lines (138 loc) · 3.91 KB
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
149
150
151
152
153
// Copyright 2019, Chen Shuaihao.
//
//Author: Chen Shuaihao
//
//
// typedef union epoll_data {
// void *ptr;
// int fd;
// uint32_t u32;
// uint64_t u64;
// } epoll_data_t;
// struct epoll_event {
// uint32_t events; /* Epoll events */
// epoll_data_t data; /* User data variable */
// };
#include "Poller.h"
#include <iostream>
#include <stdio.h> //perror
#include <stdlib.h> //exit
#include <unistd.h> //close
#include <errno.h>
#define EVENTNUM 4096 //最大触发事件数量
#define TIMEOUT 1000 //epoll_wait 超时时间设置
Poller::Poller(/* args */)
: pollfd_(-1),
eventlist_(EVENTNUM),
channelmap_(),
mutex_()
{
pollfd_ = epoll_create(256);
if(pollfd_ == -1)
{
perror("epoll_create1");
exit(1);
}
std::cout << "epoll_create" << pollfd_ << std::endl;
}
Poller::~Poller()
{
close(pollfd_);
}
//等待I/O事件
void Poller::poll(ChannelList &activechannellist)
{
int timeout = TIMEOUT;
//std::cout << "epoll_wait..." << std::endl;(int)eventlist_.capacity()
int nfds = epoll_wait(pollfd_, &*eventlist_.begin(), (int)eventlist_.capacity(), timeout);
//int nfds = epoll_wait(pollfd_, &*eventlist_.begin(), (int)channelmap_.size()*0.7+1, timeout);
if(nfds == -1)
{
//printf("epoll_wait error code is:%d", errno);
perror("epoll wait error");
//exit(1);
}
//printf("event num:%d\n", nfds);
//std::cout << "event num:" << nfds << "\n";// << std::endl;
for(int i = 0; i < nfds; ++i)
{
int events = eventlist_[i].events;
//int fd = eventlist_[i].data.fd;
Channel *pchannel = (Channel*)eventlist_[i].data.ptr;
int fd = pchannel->GetFd();
std::map<int, Channel*>::const_iterator iter;
{
std::lock_guard <std::mutex> lock(mutex_);
iter = channelmap_.find(fd);
}
if(iter != channelmap_.end())
{
pchannel->SetEvents(events);
activechannellist.push_back(pchannel);
}
else
{
std::cout << "not find channel!" << std::endl;
}
}
if(nfds == (int)eventlist_.capacity())
{
std::cout << "resize:" << nfds << std::endl;
eventlist_.resize(nfds * 2);
}
//eventlist_.clear();
}
//添加事件
void Poller::AddChannel(Channel *pchannel)
{
int fd = pchannel->GetFd();
struct epoll_event ev;
ev.events = pchannel->GetEvents();
//data是联合体
//ev.data.fd = fd;
ev.data.ptr = pchannel;
{
std::lock_guard <std::mutex> lock(mutex_);
channelmap_[fd] = pchannel;
}
if(epoll_ctl(pollfd_, EPOLL_CTL_ADD, fd, &ev) == -1)
{
perror("epoll add error");
exit(1);
}
//std::cout << "addchannel!" << std::endl;
}
//删除事件
void Poller::RemoveChannel(Channel *pchannel)
{
int fd = pchannel->GetFd();
struct epoll_event ev;
ev.events = pchannel->GetEvents();
///ev.data.fd = fd
ev.data.ptr = pchannel;
{
std::lock_guard <std::mutex> lock(mutex_);
channelmap_.erase(fd);
}
if(epoll_ctl(pollfd_, EPOLL_CTL_DEL, fd, &ev) == -1)
{
perror("epoll del error");
exit(1);
}
//std::cout << "removechannel!" << std::endl;
}
//更新事件
void Poller::UpdateChannel(Channel *pchannel)
{
int fd = pchannel->GetFd();
struct epoll_event ev;
ev.events = pchannel->GetEvents();
//ev.data.fd = fd;
ev.data.ptr = pchannel;
if(epoll_ctl(pollfd_, EPOLL_CTL_MOD, fd, &ev) == -1)
{
perror("epoll update error");
exit(1);
}
//std::cout << "updatechannel!" << std::endl;
}