From 5dfb7663e45fe337974ba86c30c0d8451bdc44a8 Mon Sep 17 00:00:00 2001
From: xiongziliang <771730766@qq.com>
Date: Tue, 23 Apr 2019 12:22:59 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96http=E6=9C=8D=E5=8A=A1?=
 =?UTF-8?q?=E5=99=A8=E5=92=8C=E5=AE=A2=E6=88=B7=E7=AB=AF=E5=86=85=E5=AD=98?=
 =?UTF-8?q?=E5=8D=A0=E7=94=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Http/HttpClient.cpp  |  2 +-
 src/Http/HttpSession.cpp | 13 +++++++++----
 2 files changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/Http/HttpClient.cpp b/src/Http/HttpClient.cpp
index 71081a3f..3181efe6 100644
--- a/src/Http/HttpClient.cpp
+++ b/src/Http/HttpClient.cpp
@@ -186,7 +186,7 @@ int64_t HttpClient::onRecvHeader(const char *data, uint64_t len) {
     //所以返回-1代表我们接下来分段接收content
     _recvedBodySize = 0;
     //根据_totalBodySize设置接收缓存大小
-    _sock->setReadBuffer(std::make_shared<BufferRaw>(MAX(_totalBodySize + 1,256 * 1024)));
+    _sock->setReadBuffer(std::make_shared<BufferRaw>(MIN(_totalBodySize + 1,256 * 1024)));
     return -1;
 }
 
diff --git a/src/Http/HttpSession.cpp b/src/Http/HttpSession.cpp
index 7200fce7..2b6a1b78 100644
--- a/src/Http/HttpSession.cpp
+++ b/src/Http/HttpSession.cpp
@@ -635,7 +635,15 @@ inline bool HttpSession::Handle_Req_POST(int64_t &content_len) {
 		return true;
 	}
 
-	if(totalContentLen > 0 && totalContentLen < maxReqSize ){
+    //根据Content-Length设置接收缓存大小
+    if(totalContentLen > 0){
+        _sock->setReadBuffer(std::make_shared<BufferRaw>(MIN(totalContentLen + 1,256 * 1024)));
+    }else{
+	    //不定长度的Content-Length
+        _sock->setReadBuffer(std::make_shared<BufferRaw>(256 * 1024));
+	}
+
+    if(totalContentLen > 0 && totalContentLen < maxReqSize ){
 		//返回固定长度的content
 		content_len = totalContentLen;
 		auto parserCopy = _parser;
@@ -652,9 +660,6 @@ inline bool HttpSession::Handle_Req_POST(int64_t &content_len) {
 			return false;
 		};
 	}else{
-		//如果是post请求 并且totalContentLen数据超过maxReqSize，那么我们加大接收缓存提升性能
-		_sock->setReadBuffer(std::make_shared<BufferRaw>(256 * 1024));
-
 		//返回不固定长度的content
 		content_len = -1;
 		auto parserCopy = _parser;