support http handler chain return HTTP_STATUS_WANT_CLOSE

Author: House-Men

http/server/HttpHandler.cpp

@@ -331,7 +331,7 @@ void HttpHandler::onMessageComplete() {
 
     if (status_code != HTTP_STATUS_NEXT) {
         // keepalive ? Reset : Close
-        if (keepalive) {
+        if (error == 0 && keepalive) {
             Reset();
         } else {
             state = WANT_CLOSE;
@@ -467,9 +467,9 @@ int HttpHandler::HandleHttpRequest() {
         pResp->status_code = (http_status)status_code;
         if (pResp->status_code >= 400 && pResp->body.size() == 0 && pReq->method != HTTP_HEAD) {
             if (service->errorHandler) {
-                customHttpHandler(service->errorHandler);
+                status_code = customHttpHandler(service->errorHandler);
             } else {
-                defaultErrorHandler();
+                status_code = defaultErrorHandler();
             }
         }
     }
@@ -481,7 +481,10 @@ int HttpHandler::HandleHttpRequest() {
         pResp->headers["Etag"] = fc->etag;
     }
     if (service->postprocessor) {
-        customHttpHandler(service->postprocessor);
+        status_code = customHttpHandler(service->postprocessor);
+    }
+    if (status_code == HTTP_STATUS_WANT_CLOSE) {
+        error = ERR_REQUEST;
     }
 
     if (writer && writer->state != hv::HttpResponseWriter::SEND_BEGIN) {
@@ -853,7 +856,7 @@ int HttpHandler::SendHttpStatusResponse(http_status status_code) {
     if (state > WANT_SEND) return 0;
     resp->status_code = status_code;
     addResponseHeaders();
-    HandleHttpRequest();
+    if (HandleHttpRequest() == HTTP_STATUS_NEXT) return 0;
     state = WANT_SEND;
     return SendHttpResponse();
 }

http/server/HttpService.h

@@ -34,6 +34,7 @@
  */
 #define HTTP_STATUS_NEXT        0
 #define HTTP_STATUS_UNFINISHED  0
+#define HTTP_STATUS_WANT_CLOSE  1
 // NOTE: http_sync_handler run on IO thread
 typedef std::function<int(HttpRequest* req, HttpResponse* resp)>                            http_sync_handler;
 // NOTE: http_async_handler run on hv::async threadpool