mirror of
https://github.com/eolinker/apinto
synced 2025-10-05 08:47:04 +08:00
141 lines
2.6 KiB
Go
141 lines
2.6 KiB
Go
package http_context_copy
|
|
|
|
import (
|
|
"bytes"
|
|
"net/http"
|
|
"strings"
|
|
|
|
http_service "github.com/eolinker/eosc/eocontext/http-context"
|
|
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
var _ http_service.IHeaderWriter = (*RequestHeader)(nil)
|
|
|
|
type RequestHeader struct {
|
|
header *fasthttp.RequestHeader
|
|
tmp http.Header
|
|
}
|
|
|
|
func (h *RequestHeader) RawHeader() string {
|
|
return h.header.String()
|
|
}
|
|
|
|
func (h *RequestHeader) reset(header *fasthttp.RequestHeader) {
|
|
h.header = header
|
|
h.tmp = nil
|
|
}
|
|
|
|
func (h *RequestHeader) initHeader() {
|
|
if h.tmp == nil {
|
|
h.tmp = make(http.Header)
|
|
h.header.VisitAll(func(key, value []byte) {
|
|
bytes.SplitN(value, []byte(":"), 2)
|
|
h.tmp[string(key)] = []string{string(value)}
|
|
})
|
|
}
|
|
}
|
|
|
|
func (h *RequestHeader) Host() string {
|
|
return string(h.header.Host())
|
|
}
|
|
|
|
func (h *RequestHeader) GetHeader(name string) string {
|
|
return h.Headers().Get(name)
|
|
}
|
|
|
|
func (h *RequestHeader) Headers() http.Header {
|
|
h.initHeader()
|
|
return h.tmp
|
|
}
|
|
|
|
func (h *RequestHeader) SetHeader(key, value string) {
|
|
if h.tmp != nil {
|
|
h.tmp.Set(key, value)
|
|
}
|
|
h.header.Set(key, value)
|
|
}
|
|
|
|
func (h *RequestHeader) AddHeader(key, value string) {
|
|
if h.tmp != nil {
|
|
h.tmp.Add(key, value)
|
|
}
|
|
h.header.Add(key, value)
|
|
}
|
|
|
|
func (h *RequestHeader) DelHeader(key string) {
|
|
if h.tmp != nil {
|
|
h.tmp.Del(key)
|
|
}
|
|
h.header.Del(key)
|
|
}
|
|
|
|
func (h *RequestHeader) SetHost(host string) {
|
|
if h.tmp != nil {
|
|
h.tmp.Set("Host", host)
|
|
}
|
|
h.header.SetHost(host)
|
|
}
|
|
|
|
type ResponseHeader struct {
|
|
header *fasthttp.ResponseHeader
|
|
tmp http.Header
|
|
}
|
|
|
|
func (r *ResponseHeader) reset(header *fasthttp.ResponseHeader) {
|
|
r.header = header
|
|
r.tmp = nil
|
|
}
|
|
func NewResponseHeader(header *fasthttp.ResponseHeader) *ResponseHeader {
|
|
return &ResponseHeader{header: header}
|
|
}
|
|
|
|
func (r *ResponseHeader) GetHeader(name string) string {
|
|
return r.Headers().Get(name)
|
|
}
|
|
|
|
func (r *ResponseHeader) Headers() http.Header {
|
|
|
|
if r.tmp == nil {
|
|
r.tmp = make(http.Header)
|
|
hs := strings.Split(r.header.String(), "\r\n")
|
|
for _, t := range hs {
|
|
vs := strings.Split(t, ":")
|
|
if len(vs) < 2 {
|
|
if vs[0] == "" {
|
|
continue
|
|
}
|
|
r.tmp[vs[0]] = []string{""}
|
|
continue
|
|
}
|
|
r.tmp[vs[0]] = []string{strings.TrimSpace(vs[1])}
|
|
}
|
|
}
|
|
return r.tmp
|
|
}
|
|
|
|
func (r *ResponseHeader) SetHeader(key, value string) {
|
|
if r.tmp != nil {
|
|
r.tmp.Set(key, value)
|
|
}
|
|
r.header.Set(key, value)
|
|
}
|
|
|
|
func (r *ResponseHeader) AddHeader(key, value string) {
|
|
if r.tmp != nil {
|
|
r.tmp.Add(key, value)
|
|
}
|
|
r.header.Add(key, value)
|
|
}
|
|
|
|
func (r *ResponseHeader) DelHeader(key string) {
|
|
if r.tmp != nil {
|
|
r.tmp.Del(key)
|
|
}
|
|
r.header.Del(key)
|
|
}
|
|
|
|
func (h *RequestHeader) GetCookie(key string) string {
|
|
return string(h.header.Cookie(key))
|
|
}
|