97d310b55c
Gin normally silently swallows errors writing to the client; however in WriteJSON (and thus context.JSON), the ResponseWriter was being passed directly into the JSON encoder, which will return an error if there's an error writing to the stream. For instance, context.JSON would panic with errors like "write tcp XXX-> YYY: write: connection reset by peer" if the client disconnected before the response was complete. This change makes JSON.Render() treat write errors the same as IndentedJSON, Data, and other renderers.
47 lines
898 B
Go
47 lines
898 B
Go
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package render
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
)
|
|
|
|
type (
|
|
JSON struct {
|
|
Data interface{}
|
|
}
|
|
|
|
IndentedJSON struct {
|
|
Data interface{}
|
|
}
|
|
)
|
|
|
|
var jsonContentType = []string{"application/json; charset=utf-8"}
|
|
|
|
func (r JSON) Render(w http.ResponseWriter) error {
|
|
return WriteJSON(w, r.Data)
|
|
}
|
|
|
|
func (r IndentedJSON) Render(w http.ResponseWriter) error {
|
|
writeContentType(w, jsonContentType)
|
|
jsonBytes, err := json.MarshalIndent(r.Data, "", " ")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
w.Write(jsonBytes)
|
|
return nil
|
|
}
|
|
|
|
func WriteJSON(w http.ResponseWriter, obj interface{}) error {
|
|
writeContentType(w, jsonContentType)
|
|
jsonBytes, err := json.Marshal(obj)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
w.Write(jsonBytes)
|
|
return nil
|
|
}
|