From 85a3582acc3d3a6de5d24455359053dac8e34049 Mon Sep 17 00:00:00 2001
From: roc <roc@imroc.cc>
Date: Tue, 9 Aug 2022 14:52:05 +0800
Subject: [PATCH] add HeaderToString for req and resp

---
 pkg/util/util.go | 13 -------------
 req.go           | 11 +++++++++++
 request.go       |  5 +++++
 response.go      |  8 ++++++++
 4 files changed, 24 insertions(+), 13 deletions(-)
 delete mode 100644 pkg/util/util.go

diff --git a/pkg/util/util.go b/pkg/util/util.go
deleted file mode 100644
index 589e8b87..00000000
--- a/pkg/util/util.go
+++ /dev/null
@@ -1,13 +0,0 @@
-package requtil
-
-import (
-	"bytes"
-	"net/http"
-)
-
-// ConvertHeaderToString converts http header to a string.
-func ConvertHeaderToString(h http.Header) string {
-	buf := new(bytes.Buffer)
-	h.Write(buf)
-	return buf.String()
-}
diff --git a/req.go b/req.go
index 507a1ce9..eef943bb 100644
--- a/req.go
+++ b/req.go
@@ -1,6 +1,7 @@
 package req
 
 import (
+	"bytes"
 	"fmt"
 	"net/http"
 	"net/url"
@@ -145,3 +146,13 @@ func cloneMap(h map[string]string) map[string]string {
 	}
 	return m
 }
+
+// convertHeaderToString converts http header to a string.
+func convertHeaderToString(h http.Header) string {
+	if h == nil {
+		return ""
+	}
+	buf := new(bytes.Buffer)
+	h.Write(buf)
+	return buf.String()
+}
diff --git a/request.go b/request.go
index 13c7a549..591dd89e 100644
--- a/request.go
+++ b/request.go
@@ -143,6 +143,11 @@ func (r *Request) TraceInfo() TraceInfo {
 	return ti
 }
 
+// HeaderToString get all header as string.
+func (r *Request) HeaderToString() string {
+	return convertHeaderToString(r.Headers)
+}
+
 // SetURL set the url for request.
 func (r *Request) SetURL(url string) *Request {
 	r.RawURL = url
diff --git a/response.go b/response.go
index 5ccc650b..ffbfd30a 100644
--- a/response.go
+++ b/response.go
@@ -212,3 +212,11 @@ func (r *Response) GetHeaderValues(key string) []string {
 	}
 	return r.Header.Values(key)
 }
+
+// HeaderToString get all header as string.
+func (r *Response) HeaderToString() string {
+	if r.Response == nil {
+		return ""
+	}
+	return convertHeaderToString(r.Header)
+}