From 138cd46c71644ba12a51140fee9f895fa6bab71c Mon Sep 17 00:00:00 2001
From: zhaoyi233 <11971297+zhao85@users.noreply.github.com>
Date: Sat, 15 Jun 2024 14:12:40 +0800
Subject: [PATCH] Update http_request_handler.go

fix http context exceptional cancel problem

Signed-off-by: zhaoyi233 <11971297+zhao85@users.noreply.github.com>
---
 server/automation/automation/http_request_handler.go | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/server/automation/automation/http_request_handler.go b/server/automation/automation/http_request_handler.go
index 03494c20cc..baf222b6b2 100644
--- a/server/automation/automation/http_request_handler.go
+++ b/server/automation/automation/http_request_handler.go
@@ -58,7 +58,14 @@ func (h httpRequestHandler) send(ctx context.Context, args *httpRequestSendArgs)
 	r.Headers = rsp.Header
 	r.ContentLength = rsp.ContentLength
 	r.ContentType = rsp.Header.Get("Content-Type")
-	r.Body = rsp.Body
+
+	defer rsp.Body.Close()
+	bodyBuffer := new(bytes.Buffer)
+	_, err = io.Copy(bodyBuffer, rsp.Body)
+	if err != nil {
+		return nil, err
+	}
+	r.Body = io.NopCloser(bodyBuffer)
 
 	return
 }