diff options
Diffstat (limited to 'system/api/json.go')
-rw-r--r-- | system/api/json.go | 57 |
1 files changed, 57 insertions, 0 deletions
diff --git a/system/api/json.go b/system/api/json.go new file mode 100644 index 0000000..e9d448e --- /dev/null +++ b/system/api/json.go @@ -0,0 +1,57 @@ +package api + +import ( + "bytes" + "encoding/json" + "log" + "net/http" +) + +func fmtJSON(data ...json.RawMessage) ([]byte, error) { + var msg = []json.RawMessage{} + for _, d := range data { + msg = append(msg, d) + } + + resp := map[string][]json.RawMessage{ + "data": msg, + } + + var buf = &bytes.Buffer{} + enc := json.NewEncoder(buf) + err := enc.Encode(resp) + if err != nil { + log.Println("Failed to encode data to JSON:", err) + return nil, err + } + + return buf.Bytes(), nil +} + +func toJSON(data []string) ([]byte, error) { + var buf = &bytes.Buffer{} + enc := json.NewEncoder(buf) + resp := map[string][]string{ + "data": data, + } + + err := enc.Encode(resp) + if err != nil { + log.Println("Failed to encode data to JSON:", err) + return nil, err + } + + return buf.Bytes(), nil +} + +// sendData should be used any time you want to communicate +// data back to a foreign client +func sendData(res http.ResponseWriter, req *http.Request, data []byte) { + res.Header().Set("Content-Type", "application/json") + res.Header().Set("Vary", "Accept-Encoding") + + _, err := res.Write(data) + if err != nil { + log.Println("Error writing to response in sendData") + } +} |