码迷，mamicode.com
首页 > 其他好文 > 详细

nginx请求体读取（二）

时间：2014-12-21 22:07:33      阅读：210      评论：0      收藏：0      [点我收藏+]

标签：

2，丢弃请求体

一个模块想要主动的丢弃客户端发过的请求体，可以调用nginx核心提供的ngx_http_discard_request_body()接口，主动丢弃的原因可能有很多种，如模块的业务逻辑压根不需要请求体 ，客户端发送了过大的请求体，另外为了兼容http1.1协议的pipeline请求，模块有义务主动丢弃不需要的请求体。总之为了保持良好的客户端兼容性，nginx必须主动丢弃无用的请求体。下面开始分析ngx_http_discard_request_body()函数：

 

[cpp] view plaincopy
 
  1. <span style="font-size:18px;">ngx_int_t  
  2. ngx_http_discard_request_body(ngx_http_request_t *r)  
  3. {  
  4.     ssize_t       size;  
  5.     ngx_event_t  *rev;  
  6.   
  7.     if (r != r->main || r->discard_body) {  
  8.         return NGX_OK;  
  9.     }  
  10.   
  11.     if (ngx_http_test_expect(r) != NGX_OK) {  
  12.         return NGX_HTTP_INTERNAL_SERVER_ERROR;  
  13.     }  
  14.   
  15.     rev = r->connection->read;  
  16.   
  17.     ngx_log_debug0(NGX_LOG_DEBUG_HTTP, rev->log, 0, "http set discard body");  
  18.   
  19.     if (rev->timer_set) {  
  20.         ngx_del_timer(rev);  
  21.     }  
  22.   
  23.     if (r->headers_in.content_length_n <= 0 || r->request_body) {  
  24.         return NGX_OK;  
  25.     }  
  26.   
  27.     size = r->header_in->last - r->header_in->pos;  
  28.   
  29.     if (size) {  
  30.         if (r->headers_in.content_length_n > size) {  
  31.             r->header_in->pos += size;  
  32.             r->headers_in.content_length_n -= size;  
  33.   
  34.         } else {  
  35.             r->header_in->pos += (size_t) r->headers_in.content_length_n;  
  36.             r->headers_in.content_length_n = 0;  
  37.             return NGX_OK;  
  38.         }  
  39.     }  
  40.   
  41.     r->read_event_handler = ngx_http_discarded_request_body_handler;  
  42.   
  43.     if (ngx_handle_read_event(rev, 0) != NGX_OK) {  
  44.         return NGX_HTTP_INTERNAL_SERVER_ERROR;  
  45.     }  
  46.   
  47.     if (ngx_http_read_discarded_request_body(r) == NGX_OK) {  
  48.         r->lingering_close = 0;  
  49.   
  50.     } else {  
  51.         r->count++;  
  52.         r->discard_body = 1;  
  53.     }  
  54.   
  55.     return NGX_OK;  
  56. }  
  57. </span>  


由于函数不长，这里把它完整的列出来了，函数的开始同样先判断了不需要再做处理的情况：子请求不需要处理，已经调用过此函数的也不需要再处理。接着调用ngx_http_test_expect() 处理http1.1 expect的情况，根据http1.1的expect机制，如果客户端发送了expect头，而服务端不希望接收请求体时，必须返回417(Expectation Failed)错误。nginx并没有这样做，它只是简单的让客户端把请求体发送过来，然后丢弃掉。接下来，函数删掉了读事件上的定时器，因为这时本身就不需要请求体，所以也无所谓客户端发送的快还是慢了，当然后面还会将到，当nginx已经处理完该请求但客户端还没有发送完无用的请求体时，nginx会在读事件上再挂上定时器。

 

函数同样还会检查请求头中的content-length头，客户端如果打算发送请求体，就必须发送content-length头，同时还会查看其他地方是不是已经读取了请求体。如果确实有待处理的请求体，函数接着检查请求头buffer中预读的数据，预读的数据会直接被丢掉，当然如果请求体已经被全部预读，函数就直接返回了。

接下来，如果还有剩余的请求体未处理，该函数调用ngx_handle_read_event()在事件处理机制中挂载好读事件，并把读事件的处理函数设置为ngx_http_discarded_request_body_handler。做好这些准备之后，该函数最后调用ngx_http_read_discarded_request_body()接口读取客户端过来的请求体并丢弃。如果客户端并没有一次将请求体发过来，函数会返回，剩余的数据等到下一次读事件过来时，交给ngx_http_discarded_request_body_handler()来处理，这时，请求的discard_body将被设置为1用来标识这种情况。另外请求的引用数(count)也被加1，这样做的目的是客户端可能在nginx处理完请求之后仍未完整发送待发送的请求体，增加引用是防止nginx核心在处理完请求后直接释放了请求的相关资源。

ngx_http_read_discarded_request_body()函数非常简单，它循环的从链接中读取数据并丢弃，直到读完接收缓冲区的所有数据，如果请求体已经被读完了，该函数会设置读事件的处理函数为ngx_http_block_reading，这个函数仅仅删除水平触发的读事件，防止同一事件不断被触发。

再来看一下读事件的处理函数ngx_http_discarded_request_body_handler，这个函数每次读事件来时会被调用，先看一下它的源码：

 

[cpp] view plaincopy
 
  1. <span style="font-size:18px;">void  
  2. ngx_http_discarded_request_body_handler(ngx_http_request_t *r)  
  3. {  
  4.     ...  
  5.   
  6.     c = r->connection;  
  7.     rev = c->read;  
  8.   
  9.     if (rev->timedout) {  
  10.         c->timedout = 1;  
  11.         c->error = 1;  
  12.         ngx_http_finalize_request(r, NGX_ERROR);  
  13.         return;  
  14.     }  
  15.   
  16.     if (r->lingering_time) {  
  17.         timer = (ngx_msec_t) (r->lingering_time - ngx_time());  
  18.   
  19.         if (timer <= 0) {  
  20.             r->discard_body = 0;  
  21.             r->lingering_close = 0;  
  22.             ngx_http_finalize_request(r, NGX_ERROR);  
  23.             return;  
  24.         }  
  25.   
  26.     } else {  
  27.         timer = 0;  
  28.     }  
  29.   
  30.     rc = ngx_http_read_discarded_request_body(r);  
  31.   
  32.     if (rc == NGX_OK) {  
  33.         r->discard_body = 0;  
  34.         r->lingering_close = 0;  
  35.         ngx_http_finalize_request(r, NGX_DONE);  
  36.         return;  
  37.     }  
  38.   
  39.     /* rc == NGX_AGAIN */  
  40.   
  41.     if (ngx_handle_read_event(rev, 0) != NGX_OK) {  
  42.         c->error = 1;  
  43.         ngx_http_finalize_request(r, NGX_ERROR);  
  44.         return;  
  45.     }  
  46.   
  47.     if (timer) {  
  48.   
  49.         clcf = ngx_http_get_module_loc_conf(r, ngx_http_core_module);  
  50.   
  51.         timer *= 1000;  
  52.   
  53.         if (timer > clcf->lingering_timeout) {  
  54.             timer = clcf->lingering_timeout;  
  55.         }  
  56.   
  57.         ngx_add_timer(rev, timer);  
  58.     }  
  59. }  
  60. </span>  

函数一开始就处理了读事件超时的情况，之前说到在ngx_http_discard_request_body()函数中已经删除了读事件的定时器，那么什么时候会设置定时器呢？答案就是在nginx已经处理完该请求，但是又没有完全将该请求的请求体丢弃的时候（客户端可能还没有发送过来），在ngx_http_finalize_connection()函数中，如果检查到还有未丢弃的请求体时，nginx会添加一个读事件定时器，它的时长为lingering_timeout指令所指定，默认为5秒，不过这个时间仅仅两次读事件之间的超时时间，等待请求体的总时长为lingering_time指令所指定，默认为30秒。这种情况中，该函数如果检测到超时事件则直接返回并断开连接。同样，还需要控制整个丢弃请求体的时长不能超过lingering_time设置的时间，如果超过了最大时长，也会直接返回并断开连接。

 

如果读事件发生在请求处理完之前，则不用处理超时事件，也不用设置定时器，函数只是简单的调用ngx_http_read_discarded_request_body()来读取并丢弃数据。

nginx请求体读取（二）

标签：

原文地址：http://www.cnblogs.com/lidabo/p/4177108.html
踩
(0)
赞
(0)
   
举报
评论 一句话评论（0）
登录后才能评论！
分享档案
更多>
2021年07月29日 (22)
2021年07月28日 (40)
2021年07月27日 (32)
2021年07月26日 (79)
2021年07月23日 (29)
2021年07月22日 (30)
2021年07月21日 (42)
2021年07月20日 (16)
2021年07月19日 (90)
2021年07月16日 (35)
周排行
mamicode.com排行更多图片
更多
友情链接
兰亭集智   国之画   百度统计   站长统计   阿里云   chrome插件   新版天听网
关于我们 - 联系我们 - 留言反馈
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码！