diff --git a/args.go b/args.go index 9a84509..f0778ee 100644 --- a/args.go +++ b/args.go @@ -163,7 +163,7 @@ func (a *Args) SetBytesV(key string, value []byte) { // SetBytesKV sets 'key=value' argument. func (a *Args) SetBytesKV(key, value []byte) { - a.args = setArg(a.args, key, value) + a.args = setArgBytes(a.args, key, value) } // Peek returns query arg value for the given key. @@ -300,7 +300,7 @@ func delAllArgs(args []argsKV, key []byte) []argsKV { return args } -func setArg(h []argsKV, key, value []byte) []argsKV { +func setArgBytes(h []argsKV, key, value []byte) []argsKV { n := len(h) for i := 0; i < n; i++ { kv := &h[i] diff --git a/header.go b/header.go index 84c4dd1..0837d1d 100644 --- a/header.go +++ b/header.go @@ -215,7 +215,7 @@ func (h *ResponseHeader) SetContentLength(contentLength int) { h.SetConnectionClose() value = strIdentity } - h.h = setArg(h.h, strTransferEncoding, value) + h.h = setArgBytes(h.h, strTransferEncoding, value) } } @@ -256,7 +256,7 @@ func (h *RequestHeader) SetContentLength(contentLength int) { h.h = delAllArgs(h.h, strTransferEncoding) } else { h.contentLengthBytes = h.contentLengthBytes[:0] - h.h = setArg(h.h, strTransferEncoding, strChunked) + h.h = setArgBytes(h.h, strTransferEncoding, strChunked) } } @@ -882,20 +882,20 @@ func (h *ResponseHeader) SetCanonical(key, value []byte) { h.SetConnectionClose() } else { h.ResetConnectionClose() - h.h = setArg(h.h, key, value) + h.h = setArgBytes(h.h, key, value) } case "Transfer-Encoding": // Transfer-Encoding is managed automatically. case "Date": // Date is managed automatically. default: - h.h = setArg(h.h, key, value) + h.h = setArgBytes(h.h, key, value) } } // SetCookie sets the given response cookie. func (h *ResponseHeader) SetCookie(cookie *Cookie) { - h.cookies = setArg(h.cookies, cookie.Key(), cookie.Cookie()) + h.cookies = setArgBytes(h.cookies, cookie.Key(), cookie.Cookie()) } // SetCookie sets 'key: value' cookies. @@ -914,7 +914,7 @@ func (h *RequestHeader) SetCookieBytesK(key []byte, value string) { func (h *RequestHeader) SetCookieBytesKV(key, value []byte) { h.parseRawHeaders() h.collectCookies() - h.cookies = setArg(h.cookies, key, value) + h.cookies = setArgBytes(h.cookies, key, value) } // Set sets the given 'key: value' header. @@ -966,12 +966,12 @@ func (h *RequestHeader) SetCanonical(key, value []byte) { h.SetConnectionClose() } else { h.ResetConnectionClose() - h.h = setArg(h.h, key, value) + h.h = setArgBytes(h.h, key, value) } case "Transfer-Encoding": // Transfer-Encoding is managed automatically. default: - h.h = setArg(h.h, key, value) + h.h = setArgBytes(h.h, key, value) } } @@ -1594,7 +1594,7 @@ func (h *ResponseHeader) parseHeaders(buf []byte) (int, error) { case "Transfer-Encoding": if !bytes.Equal(s.value, strIdentity) { h.contentLength = -1 - h.h = setArg(h.h, strTransferEncoding, strChunked) + h.h = setArgBytes(h.h, strTransferEncoding, strChunked) } case "Set-Cookie": h.cookies, kv = allocArg(h.cookies) @@ -1620,7 +1620,7 @@ func (h *ResponseHeader) parseHeaders(buf []byte) (int, error) { h.contentLengthBytes = h.contentLengthBytes[:0] } if h.contentLength == -2 && !h.ConnectionUpgrade() && !h.mustSkipContentLength() { - h.h = setArg(h.h, strTransferEncoding, strIdentity) + h.h = setArgBytes(h.h, strTransferEncoding, strIdentity) h.connectionClose = true } if h.noHTTP11 && !h.connectionClose { @@ -1658,7 +1658,7 @@ func (h *RequestHeader) parseHeaders(buf []byte) (int, error) { case "Transfer-Encoding": if !bytes.Equal(s.value, strIdentity) { h.contentLength = -1 - h.h = setArg(h.h, strTransferEncoding, strChunked) + h.h = setArgBytes(h.h, strTransferEncoding, strChunked) } case "Connection": if bytes.Equal(s.value, strClose) {