From 0b8fed7e80e86955213484e4c879d23ebb86110b Mon Sep 17 00:00:00 2001
From: M09Ic <m09ician@gmail.com>
Date: Tue, 10 Jan 2023 11:59:43 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96body=E8=AF=BB=E5=8F=96?=
 =?UTF-8?q?=E7=9A=84=E9=80=BB=E8=BE=91,=200byte=E7=9A=84body=E5=B0=86?=
 =?UTF-8?q?=E4=BC=9A=E8=87=AA=E5=8A=A8=E8=B7=B3=E8=BF=87=E8=AF=BB=E5=8F=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/pool.go |  1 -
 pkg/baseline.go  | 16 ++++++++++------
 2 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/internal/pool.go b/internal/pool.go
index d447d2c..16e27fc 100644
--- a/internal/pool.go
+++ b/internal/pool.go
@@ -294,7 +294,6 @@ func (pool *Pool) Invoke(v interface{}) {
 		return
 	}
 	req.SetHeaders(pool.Headers)
-
 	start := time.Now()
 	resp, reqerr := pool.client.Do(pool.ctx, req)
 	if pool.ClientType == ihttp.FAST {
diff --git a/pkg/baseline.go b/pkg/baseline.go
index a2459d2..62543c5 100644
--- a/pkg/baseline.go
+++ b/pkg/baseline.go
@@ -33,12 +33,16 @@ func NewBaseline(u, host string, resp *ihttp.Response) *Baseline {
 	copy(bl.Header, header)
 	bl.HeaderLength = len(bl.Header)
 
-	body := resp.Body()
-	bl.Body = make([]byte, len(body))
-	copy(bl.Body, body)
-	bl.BodyLength = resp.ContentLength()
-	if bl.BodyLength == -1 {
-		bl.BodyLength = len(bl.Body)
+	if i := resp.ContentLength(); i != 0 {
+		body := resp.Body()
+		bl.Body = make([]byte, len(body))
+		copy(bl.Body, body)
+
+		if i == -1 {
+			bl.BodyLength = len(bl.Body)
+		} else {
+			bl.BodyLength = i
+		}
 	}
 
 	if t, ok := ContentTypeMap[resp.ContentType()]; ok {