From 482c0a00aace1eba75743d4eec9f3048101b8d6c Mon Sep 17 00:00:00 2001 From: link1st <562117637@qq.com> Date: Mon, 12 Aug 2024 11:00:07 +0800 Subject: [PATCH] optimize code --- model/request_model.go | 9 +++++++++ server/client/http_client.go | 2 +- 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/model/request_model.go b/model/request_model.go index 9fa8329..29d753a 100644 --- a/model/request_model.go +++ b/model/request_model.go @@ -94,6 +94,15 @@ func (r *Request) GetBody() (body io.Reader) { return strings.NewReader(r.Body) } +// CopyHeaders copy Headers +func (r *Request) CopyHeaders() map[string]string { + var result = make(map[string]string, len(r.Headers)) + for k, v := range r.Headers { + result[k] = v + } + return result +} + // getVerifyKey 获取校验 key func (r *Request) getVerifyKey() (key string) { return fmt.Sprintf("%s.%s", r.Form, r.Verify) diff --git a/server/client/http_client.go b/server/client/http_client.go index 1cf2c7f..2914937 100644 --- a/server/client/http_client.go +++ b/server/client/http_client.go @@ -29,7 +29,7 @@ func HTTPRequest(chanID uint64, request *model.Request) (resp *http.Response, re url := request.URL body := request.GetBody() timeout := request.Timeout - headers := request.Headers + headers := request.CopyHeaders() req, err := http.NewRequest(method, url, body) if err != nil {