0e2f1fe3f5
* QueryMetricsV2: Stream response Signed-off-by: Arve Knudsen <arve.knudsen@gmail.com> * API: Use jsoniter instead of standard JSON package Signed-off-by: Arve Knudsen <arve.knudsen@gmail.com>
117 lines
2.6 KiB
Go
117 lines
2.6 KiB
Go
package api
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/grafana/grafana/pkg/models"
|
|
jsoniter "github.com/json-iterator/go"
|
|
)
|
|
|
|
// Response is an HTTP response interface.
|
|
type Response interface {
|
|
// WriteTo writes to a context.
|
|
WriteTo(ctx *models.ReqContext)
|
|
// Body gets the response's body.
|
|
Body() []byte
|
|
// Status gets the response's status.
|
|
Status() int
|
|
}
|
|
|
|
type NormalResponse struct {
|
|
status int
|
|
body []byte
|
|
header http.Header
|
|
errMessage string
|
|
err error
|
|
}
|
|
|
|
// Status gets the response's status.
|
|
func (r *NormalResponse) Status() int {
|
|
return r.status
|
|
}
|
|
|
|
// Body gets the response's body.
|
|
func (r *NormalResponse) Body() []byte {
|
|
return r.body
|
|
}
|
|
|
|
func (r *NormalResponse) WriteTo(ctx *models.ReqContext) {
|
|
if r.err != nil {
|
|
ctx.Logger.Error(r.errMessage, "error", r.err, "remote_addr", ctx.RemoteAddr())
|
|
}
|
|
|
|
header := ctx.Resp.Header()
|
|
for k, v := range r.header {
|
|
header[k] = v
|
|
}
|
|
ctx.Resp.WriteHeader(r.status)
|
|
if _, err := ctx.Resp.Write(r.body); err != nil {
|
|
ctx.Logger.Error("Error writing to response", "err", err)
|
|
}
|
|
}
|
|
|
|
func (r *NormalResponse) Header(key, value string) *NormalResponse {
|
|
r.header.Set(key, value)
|
|
return r
|
|
}
|
|
|
|
// Empty creates an empty NormalResponse.
|
|
func Empty(status int) *NormalResponse {
|
|
return Respond(status, nil)
|
|
}
|
|
|
|
// streamingResponse is a response that streams itself back to the client.
|
|
type streamingResponse struct {
|
|
body interface{}
|
|
status int
|
|
header http.Header
|
|
}
|
|
|
|
// Status gets the response's status.
|
|
// Required to implement api.Response.
|
|
func (r streamingResponse) Status() int {
|
|
return r.status
|
|
}
|
|
|
|
// Body gets the response's body.
|
|
// Required to implement api.Response.
|
|
func (r streamingResponse) Body() []byte {
|
|
return nil
|
|
}
|
|
|
|
// WriteTo writes the response to the provided context.
|
|
// Required to implement api.Response.
|
|
func (r streamingResponse) WriteTo(ctx *models.ReqContext) {
|
|
header := ctx.Resp.Header()
|
|
for k, v := range r.header {
|
|
header[k] = v
|
|
}
|
|
ctx.Resp.WriteHeader(r.status)
|
|
enc := jsoniter.NewEncoder(ctx.Resp)
|
|
if err := enc.Encode(r.body); err != nil {
|
|
ctx.Logger.Error("Error writing to response", "err", err)
|
|
}
|
|
}
|
|
|
|
// RedirectResponse represents a redirect response.
|
|
type RedirectResponse struct {
|
|
location string
|
|
}
|
|
|
|
// WriteTo writes to a response.
|
|
func (r *RedirectResponse) WriteTo(ctx *models.ReqContext) {
|
|
ctx.Redirect(r.location)
|
|
}
|
|
|
|
// Status gets the response's status.
|
|
// Required to implement api.Response.
|
|
func (*RedirectResponse) Status() int {
|
|
return http.StatusFound
|
|
}
|
|
|
|
// Body gets the response's body.
|
|
// Required to implement api.Response.
|
|
func (r *RedirectResponse) Body() []byte {
|
|
return nil
|
|
}
|