nginx源码分析--module开发（4）

ゞ浴缸里的玫瑰 2022-08-10 11:55 228阅读 0赞

3、对HTTP body的处理（仍然在ngx\_http\_core\_run\_phases 只不过是另一个阶段）

上面我们已经开始处理http request header了，接下来，如果请求中有body内容，那么需要处理body了。这里你肯定不会想要去阻塞式的读取body吧？body的长度可大可小，网络环境也巨复杂，只要有阻塞操作肯定玩完。Nginx这时已经准备了一个现成的读取body的非阻塞模式给用户，就是ngx\_http\_read\_client\_request\_body方法。

大家看下ngx\_http\_read\_client\_request\_body方法的原型：

ngx\_int\_t

ngx\_http\_read\_client\_request\_body(ngx\_http\_request\_t \*r,ngx\_http\_client\_body\_handler\_pt post\_handler)；

参数r就是要处理的请求，post\_handler则是body接收完成后的回调方法。

所以，在worker进程中，调用ngx\_http\_read\_client\_request\_body是不会阻塞的，要么读完socket上的buffer发现不完整立刻返回，等待下一次EPOLLIN事件，要么就是读完body了，调用用户定义的post\_handler方法去处理body。

ngx\_http\_read\_client\_request\_body提供两种保存body的方式，一种是把body存储在内存中，另一种是把body存储到临时文件里。这个临时文件也有不同的处理方法，一种是请求结束后nginx便清理掉，另外就是永久保留这个临时文件。例如下面这两个参数就会设定为每个body都存放到临时文件里，并且这个临时文件在请求结束后不会被删除：

r->request\_body\_in\_persistent\_file = 1;

r->request\_body\_in\_file\_only = 1;

貌似ngx\_http\_read\_client\_request\_body已经提供了很足够的功能了，其实不然。比如，我现在实现的业务中，就要求针对不同的请求，我要把body放到不同的目录下，也就是不同的mountpoint点上。这样，如果我把临时body指定到同一个mountpoint点下，相当于已经存储到远程机器上了，但是nginx的ngx\_http\_read\_client\_request\_body方法是不提供这个功能的，它的临时文件目录早在编译时已经通过--http-client-body-temp-path=指定了，无法在运行时更改，比较恶心的实现。我想实现这个功能，只能自己重构下接收body这套方法了。这里我把我的处理方法（修改自ngx\_http\_read\_client\_request\_body处理步骤）说下，方便大家理解body的接收过程。

在我决定开始接收body后，首先调用ngx\_http\_read\_webex\_DMD\_request\_body方法，同时把处理完整body的勾子函数ngx\_webex\_DMD\_handler\_request\_body也传进去。如果需要把temp body file放到指定目录，这时需要按照自己的方式把临时目录传进去，通过参数或者通过自定义的ngx\_Module\_ctx\_t方式，这个随意。

ngx\_http\_read\_webex\_DMD\_request\_body方法首先判断是否存在body，如果不存在，立刻回调ngx\_webex\_DMD\_handler\_request\_body方法；如果存在body，那么根据已经收到的buffer判断是否已经接收完body，如果已经收到完整的body，判断是否需要写入临时文件中，若需要则调用webex\_DMD\_ngx\_http\_write\_request\_body方法写临时文件，然后回调ngx\_webex\_DMD\_handler\_request\_body。则如果没有收完，调用webex\_DMD\_ngx\_http\_do\_read\_client\_request\_body方法。

webex\_DMD\_ngx\_http\_do\_read\_client\_request\_body方法里，首先查看无阻塞的socket上是否仍有buffer，有则读出，再次判断是否body完整，如果已经收到完整的body，判断是否需要写入临时文件中，若需要则调用webex\_DMD\_ngx\_http\_write\_request\_body方法写临时文件，然后回调ngx\_webex\_DMD\_handler\_request\_body。如果不完整，注册EPOLLIN事件的回调函数为webex\_DMD\_ngx\_http\_read\_client\_request\_body\_handler，把控制权交回给nginx epoll，等待下一次的EPOLLIN事件。下次EPOLLIN事件到达时，调用webex\_DMD\_ngx\_http\_read\_client\_request\_body\_handler，然后该函数会继续调用webex\_DMD\_ngx\_http\_do\_read\_client\_request\_body读取buffer，重复上一个步骤，直到body完整。

为了大家方便理解，我再画一幅活动图吧。

![0_12753027369UbY.gif][]

[0_12753027369UbY.gif]: /images/20220810/2cc4cfa5eb16498ca474247486bc0935.png