63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
|
package aicra
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/json"
|
||
|
|
"net/http"
|
||
|
|
|
||
|
|
"github.com/xdrm-io/aicra/api"
|
||
|
|
)
|
||
|
|
|
||
|
|
// response for an service call
|
||
|
|
type response struct {
|
||
|
|
Data map[string]interface{}
|
||
|
|
Status int
|
||
|
|
Headers http.Header
|
||
|
|
err api.Err
|
||
|
|
}
|
||
|
|
|
||
|
|
// newResponse creates an empty response.
|
||
|
|
func newResponse() *response {
|
||
|
|
return &response{
|
||
|
|
Status: http.StatusOK,
|
||
|
|
Data: make(map[string]interface{}),
|
||
|
|
err: api.ErrFailure,
|
||
|
|
Headers: make(http.Header),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WithError sets the response error
|
||
|
|
func (res *response) WithError(err api.Err) *response {
|
||
|
|
res.err = err
|
||
|
|
return res
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetValue sets a response value
|
||
|
|
func (res *response) SetValue(name string, value interface{}) {
|
||
|
|
res.Data[name] = value
|
||
|
|
}
|
||
|
|
|
||
|
|
// MarshalJSON generates the JSON representation of the response
|
||
|
|
//
|
||
|
|
// implements json.Marshaler
|
||
|
|
func (res *response) MarshalJSON() ([]byte, error) {
|
||
|
|
fmt := make(map[string]interface{})
|
||
|
|
for k, v := range res.Data {
|
||
|
|
fmt[k] = v
|
||
|
|
}
|
||
|
|
fmt["error"] = res.err
|
||
|
|
return json.Marshal(fmt)
|
||
|
|
}
|
||
|
|
|
||
|
|
// ServeHTTP writes the response representation back to the http.ResponseWriter
|
||
|
|
//
|
||
|
|
// implements http.Handler
|
||
|
|
func (res *response) ServeHTTP(w http.ResponseWriter, r *http.Request) error {
|
||
|
|
w.WriteHeader(res.err.Status)
|
||
|
|
encoded, err := json.Marshal(res)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
w.Write(encoded)
|
||
|
|
return nil
|
||
|
|
}
|