diff options
Diffstat (limited to 'http/response.go')
-rw-r--r-- | http/response.go | 42 |
1 files changed, 42 insertions, 0 deletions
diff --git a/http/response.go b/http/response.go new file mode 100644 index 00000000..d35669c4 --- /dev/null +++ b/http/response.go @@ -0,0 +1,42 @@ +package http + +import ( + "net/http" + "encoding/json" + "github.com/spiral/roadrunner" +) + +// Response handles PSR7 response logic. +type Response struct { + // Status contains response status. + Status int `json:"status"` + + // Headers contains list of response headers. + Headers map[string][]string `json:"headers"` + + // associated body payload. + body []byte +} + +// NewResponse creates new response based on given roadrunner payload. +func NewResponse(p *roadrunner.Payload) (*Response, error) { + r := &Response{body: p.Body} + if err := json.Unmarshal(p.Context, r); err != nil { + return nil, err + } + + return r, nil +} + +// Write writes response headers, status and body into ResponseWriter. +func (r *Response) Write(w http.ResponseWriter) { + for k, v := range r.Headers { + for _, h := range v { + w.Header().Add(k, h) + + } + } + + w.WriteHeader(r.Status) + w.Write(r.body) +} |