Linux高性能服务器编程第十五章线程池
Linux高性能服务器编程第十五章线程池
Hoshea Zhang我们这里不介绍进程池
概述
创建一组子线程,运行着相同的代码和属性,任务到来时,主线程通过某种方式选择一个子线程为之服务,方法有两种:
- 随机算法或轮流选取算法
- 工作队列
处理多客户
这里用进程的概念去解释:
在使用进程池处理多客户任务时,首先要考虑的一个问题是﹔监听 socket和连接socket是否都由主进程来统一管理。回忆第8章中我们介绍过的几种并发模式,其中半同步/半反应堆模式是由主进程统一管理这两种 socket的;而半同步/半异步模式,以及领导者/追随者模式,则是由主进程管理所有监听socket,而各个子进程分别管理属于自己的连接socket 的。对于前一种情况,主进程接受新的连接以得到连接socket,然后它需要将该socket传递给子进程(对于线程池而言,父线程将socket传递给子线程是很简单的,因为它们可以很容易地共享该socket。)。后一种情况的灵活性更大一些,因为子进程可以自己调用accept来接受新的连接,这样父进程就无须向子进程传递socket,而只需要简单地通知一声:“我检测到新的连接,你来接受它。”
如果客户任务存在上下文关系,最好使用同一个子线程去服务,多线程中可以使用EPOLLONESHOT来处理
线程池的实现
具体看注释
同步机制包装类
1 | /线程同步机制包装类 |
线程池类
1 |
|
http_conn
头文件
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
class http_conn
{
public:
static const int FILENAME_LEN = 200;//文件名的最大长度
static const int READ_BUFFER_SIZE = 2048;//缓冲区的大小
static const int WRITE_BUFFER_SIZE = 1024;//写缓冲区的大小
//HTTP请求方法
//HTTP请求方法,但本代码中仅仅支持GET
enum METHOD{
GET = 0,
POST,
HEAD,
PUT,
DELETE,
TRACE,
OPTIONS,
CONNECT,
PATCH
};
//主状态机可能的状态
enum CHECK_STATE
{
CHECK_STATE_REQUESTION = 0,//正在分析当前请求行
CHECK_STATE_HEADER,//正在分析头部字段
CHECK_STATE_CONTENT
};
//从状态机可能的状态
enum LINE_STATUS
{
LINE_OK = 0,//读取到一个完整的行
LINE_BAD,//行出错
LINE_OPEN//行数据尚且不完整
};
//服务器处理http请求的结果
enum HTTP_CODE
{
NO_REQUEST,//请求不完整需要继续读取
GET_REQUEST,//得到了一个完整的请求
BAD_REQUEST,//请求有语法错误
NO_RESOURCE,//没有资源
FORBIDDEN_REQUEST,//没有足够的权限
FILE_REQUEST,//文件已被请求
INTERNAL_ERROR,//服务器内部错误
CLOSED_CONNECTION//客户端连接已关闭
};
public:
http_conn(){}
~http_conn(){}
public:
void init(int sockfd,const sockaddr_in& addr);
void close_conn(bool real_close = true);
bool read();
bool write();
void process();
private:
void init();
HTTP_CODE process_read();
bool process_write(HTTP_CODE ret);
//下面一组函数被process_read调用以分析HTTP请求
HTTP_CODE parse_request_line(char* text);
HTTP_CODE parse_header(char* text);
HTTP_CODE parse_content(char* text);
HTTP_CODE do_request();
char* get_line(){ return m_read_buf + m_start_line; }
LINE_STATUS parse_line();
//下面一组函数被process_write调用以填充HTTP应答
void unmap();
bool add_response(const char* format, ...);
bool add_content(const char* content);
bool add_status_line(int status, const char* title);
bool add_headers(int content_length);
bool add_content_length(int content_length);
bool add_linger();
bool add_blank_line();
public:
static int m_epollfd;
static int m_user_count;
private:
int m_sockfd;
sockaddr_in m_address;
char m_read_buf[READ_BUFFER_SIZE];
int m_read_idx;
int m_check_idx;
int m_start_line;
char m_write_buf[WRITE_BUFFER_SIZE];
int m_write_idx;
CHECK_STATE m_check_state;
METHOD m_method;
char m_real_file[FILENAME_LEN];
char* m_url;
char* m_version;
char* m_host;
int m_content_length;
bool m_linger;
char* m_file_address;
struct stat m_file_stat;
struct iovec m_iv[2];
int m_iv_count;
};
源文件
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
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
//定义一些HTTP响应的状态信息
const char* ok_200_title = "OK";
const char* error_400_title = "Bad Request";
const char* error_400_form = "Your request has bad syntax or is inherently impossible to satisfy.\n";
const char* error_403_title = "Forbidden";
const char* error_403_form = "you do not have permisson to get file from this server.\n";
const char* error_404_title = "Not Found";
const char* error_404_form = "The requested file was not found on this server.\n";
const char* error_500_title = "Internal Error";
const char* error_500_form = "There was an unusual problem serving the requested file.\n";
//网站的根目录
const char* doc_root = "/home/ross/Documents";
int setnonblocking(int fd){
int old_option = fcntl(fd,F_GETFL);
int new_option = old_option | O_NONBLOCK;
fcntl(fd,F_SETFL,new_option);
return old_option;
}
void addfd(int epollfd,int fd,bool one_shot){
epoll_event event;
event.data.fd = fd;
event.events = EPOLLIN | EPOLLET | EPOLLRDHUP;
if(one_shot){
event.events |= EPOLLONESHOT;
}
epoll_ctl(epollfd,EPOLL_CTL_ADD,fd,&event);
setnonblocking(fd);
}
void removefd(int epollfd,int fd){
epoll_ctl(epollfd,EPOLL_CTL_DEL,fd,0);
close(fd);
}
void modfd(int epollfd,int fd,int ev){
epoll_event event;
event.data.fd = fd;
event.events = ev | EPOLLET | EPOLLONESHOT | EPOLLRDHUP;
epoll_ctl(epollfd,EPOLL_CTL_MOD,fd,&event);
}
int http_conn::m_user_count = 0;
int http_conn::m_epollfd = -1;
void http_conn::close_conn(bool real_close){
if(real_close && (m_sockfd != -1)){
removefd(m_epollfd,m_sockfd);
m_sockfd = -1;
m_user_count --;
}
}
void http_conn::init(int sockfd,const sockaddr_in& addr){
m_sockfd = sockfd;
m_address = addr;
addfd(m_epollfd,sockfd,true);
m_user_count ++;
init();
}
void http_conn::init(){
m_check_state = CHECK_STATE_REQUESTION;
m_linger = false;
m_method = GET;
m_url = 0;
m_version = 0;
m_content_length = 0;
m_host = 0;
m_start_line = 0;
m_check_idx = 0;
m_read_idx = 0;
m_write_idx = 0;
memset(m_read_buf,'\0',READ_BUFFER_SIZE);
memset(m_write_buf,'\0',WRITE_BUFFER_SIZE);
memset(m_real_file,'\0',FILENAME_LEN);
}
http_conn::LINE_STATUS http_conn::parse_line(){
char temp;
for(;m_check_idx<m_read_idx;++m_check_idx){
temp = m_read_buf[m_check_idx];
if(temp == '\r'){
if(m_read_buf[m_check_idx+1] == '\n'){
m_read_buf[m_check_idx++] = '\0';
m_read_buf[m_check_idx++] = '\0';
return LINE_OK;
}
else if( (m_check_idx+1) == m_read_idx){
return LINE_OPEN;
}
else
return LINE_BAD;
}
else if(temp == '\n'){
if((m_check_idx>1) && (m_read_buf[m_check_idx-1] == '\r'))
{
m_read_buf[m_check_idx-1] = '\0';
m_read_buf[m_check_idx++] = '\0';
return LINE_OK;
}
else{
return LINE_BAD;
}
}
}
return LINE_OPEN;
}
bool http_conn::read(){
if(m_read_idx >= READ_BUFFER_SIZE){
return false;
}
int bytes_read = 0;
while(true){
bytes_read = recv(m_sockfd,m_read_buf,READ_BUFFER_SIZE-m_read_idx,0);
if(bytes_read == -1){
if( errno == EAGAIN || errno == EWOULDBLOCK){
break;
}
return false;
}
else if(bytes_read == 0){
return false;
}
m_read_idx += bytes_read;
}
return true;
}
http_conn::HTTP_CODE http_conn::parse_request_line(char* text){
m_url = strpbrk(text," \t");
if(!m_url){
return BAD_REQUEST;
}
*m_url++ = '\0';
m_url += strspn(m_url," \t");
char* method = text;
if(strcasecmp(method,"GET") == 0){
m_method = GET;
}
else{
return BAD_REQUEST;
}
m_version = strpbrk(m_url," \t");
if(!m_version){
return BAD_REQUEST;
}
*m_version++ = '\0';
m_version+=strspn(m_version," \t");
if(strcasecmp(m_version,"HTTP/1.1") != 0){
return BAD_REQUEST;
}
if(strncasecmp(m_url,"http://",7)==0){
m_url+=7;
m_url = strchr(m_url,'/');
}
if(!m_url || m_url[0]!= '/'){
return BAD_REQUEST;
}
m_check_state = CHECK_STATE_HEADER;
return NO_REQUEST;
}
http_conn::HTTP_CODE http_conn::parse_header(char* text){
if(text[0] == '\0'){
if(m_content_length != 0){
m_check_state = CHECK_STATE_CONTENT;
return NO_REQUEST;
}
printf("parse_header OK\n");
return GET_REQUEST;
}
else if(strncasecmp(text,"Connection:",11) == 0){
text+=11;
text += strspn(text," \t");
if(strcasecmp(text,"keep-alive") == 0){
m_linger = true;
}
}
else if(strncasecmp(text,"Content-Length",15)==0){
text += 15;
text += strspn(text," \t");
m_content_length += atol(text);
}
else if(strncasecmp(text,"Host:",5) == 0){
text+=5;
text+=strspn(text," \t");
m_host = text;
}
else{
printf("oh no! I can't handle this header %s\n",text);
}
return NO_REQUEST;
}
http_conn::HTTP_CODE http_conn::parse_content(char* text){
if(m_read_idx >= m_content_length+m_check_idx){
text[m_content_length] = '\0';
return GET_REQUEST;
}
return NO_REQUEST;
}
http_conn::HTTP_CODE http_conn::process_read(){
LINE_STATUS line_status = LINE_OK;
HTTP_CODE ret = NO_REQUEST;
char* text = 0;
while(((m_check_state == CHECK_STATE_CONTENT) && (line_status == LINE_OK)) || ((line_status=parse_line()) ==LINE_OK) ){
text = get_line();
m_start_line = m_check_idx;
printf("I got 1 line %s\n",text);
switch(m_check_state)
{
case CHECK_STATE_REQUESTION:{
ret = parse_request_line(text);
if(ret == BAD_REQUEST){
return BAD_REQUEST;
}
break;
}
case CHECK_STATE_HEADER:{
ret = parse_header(text);
if(ret == BAD_REQUEST){
return BAD_REQUEST;
}
else if(ret == GET_REQUEST){
return do_request();
}
break;
}
case CHECK_STATE_CONTENT:{
ret = parse_content(text);
if(ret == GET_REQUEST){
return do_request();
}
line_status = LINE_OPEN;
break;
}
default:
return INTERNAL_ERROR;
}
}
return NO_REQUEST;
}
http_conn::HTTP_CODE http_conn::do_request(){
strcpy(m_real_file,doc_root);
int len = strlen(doc_root);
strncpy(m_real_file+len,m_url,FILENAME_LEN-len-1);
if(stat(m_real_file,&m_file_stat) < 0){
return NO_RESOURCE;
}
if(!(m_file_stat.st_mode & S_IROTH)){
return FORBIDDEN_REQUEST;
}
if(S_ISDIR(m_file_stat.st_mode)){
return BAD_REQUEST;
}
int fd = open(m_real_file,O_RDONLY);
m_file_address = (char*)mmap(0,m_file_stat.st_size,PROT_READ,MAP_PRIVATE,fd,0);
close(fd);
return FILE_REQUEST;
}
void http_conn::unmap(){
if(m_file_address){
munmap(m_file_address,m_file_stat.st_size);
m_file_address = 0;
}
}
bool http_conn::add_response(const char* format, ...){
if(m_write_idx>WRITE_BUFFER_SIZE){
return false;
}
va_list arg_list;
va_start(arg_list,format);
int len = vsnprintf(m_write_buf+m_write_idx,WRITE_BUFFER_SIZE-1-m_write_idx,format,arg_list);
if(len >= (WRITE_BUFFER_SIZE-1-m_write_idx)){
return false;
}
m_write_idx+=len;
va_end(arg_list);
return true;
}
bool http_conn::add_status_line(int status,const char* title){
return add_response("%s %d %s\r\n","HTTP/1.1",status,title);
}
bool http_conn::add_headers(int content_len){
add_content_length(content_len);
add_linger();
add_blank_line();
return true;
}
bool http_conn::add_content_length(int content_len){
return add_response("Content-Length: %d\r\n",content_len);
}
bool http_conn::add_linger(){
return add_response("Connection: %s\r\n",(m_linger == true)?"keep-alive":"close");
}
bool http_conn::add_blank_line(){
return add_response("%s","\r\n");
}
bool http_conn::add_content(const char* content){
return add_response("%s",content);
}
bool http_conn::process_write(HTTP_CODE ret){
switch ((ret))
{
case INTERNAL_ERROR:
{
add_status_line(500,error_500_title);
add_headers(strlen(error_500_form));
if(!add_content(error_500_form)){
return false;
}
break;
}
case BAD_REQUEST:
{
add_status_line(400,error_400_title);
add_headers(strlen(error_400_form));
if(! add_content(error_500_form)){
return false;
}
break;
}
case NO_RESOURCE:
{
add_status_line(404,error_404_title);
add_headers(strlen(error_404_form));
if(! add_content(error_404_form)){
return false;
}
break;
}
case FORBIDDEN_REQUEST:
{
add_status_line(403,error_403_title);
add_headers(strlen(error_403_form));
if(! add_content(error_403_form)){
return false;
}
break;
}
case FILE_REQUEST:
{
add_status_line(200,ok_200_title);
if(m_file_stat.st_size !=0){
add_headers(m_file_stat.st_size);
m_iv[0].iov_base = m_write_buf;
m_iv[0].iov_len = m_write_idx;
m_iv[1].iov_base = m_file_address;
m_iv[1].iov_len = m_file_stat.st_size;
m_iv_count = 2;
printf("process_write ok\n");
return true;
}
else{
const char* ok_string = "<html><body></body></html>";
add_headers(strlen(ok_string));
if(! add_content(ok_string)){
return false;
}
break;
}
}
default:
{
return false;
}
}
m_iv[0].iov_base = m_write_buf;
m_iv[0].iov_len = m_write_idx;
m_iv_count = 1;
return true;
}
bool http_conn::write(){
int temp = 0;
int bytes_have_send = 0;
int bytes_to_send = m_write_idx;
if(bytes_to_send == 0){
modfd(m_epollfd,m_sockfd,EPOLLOUT);
init();
return true;
}
while(1)
{
temp = writev(m_sockfd,m_iv,m_iv_count);
if(temp < 0){
if(errno == EAGAIN){
modfd(m_epollfd,m_sockfd,EPOLLOUT);
return true;
}
unmap();
return false;
}
bytes_have_send+=temp;
bytes_to_send-= temp;
if(bytes_to_send<=bytes_have_send){
unmap();
if(m_linger){
init();
modfd(m_epollfd,m_sockfd,EPOLLIN);
return true;
}
else{
modfd(m_epollfd,m_sockfd,EPOLLIN);
return false;
}
}
}
}
void http_conn::process()
{
HTTP_CODE read_ret = process_read();
if(read_ret == NO_REQUEST){
modfd(m_epollfd,m_sockfd,EPOLLIN);
return;
}
bool write_ret = process_write(read_ret);
if(!write_ret){
close_conn();
}
modfd(m_epollfd,m_sockfd,EPOLLOUT);
}
main函数
1 |
|