forked from synapse-garden/mf-proto
/
api.go
84 lines (72 loc) · 1.63 KB
/
api.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
package api
import (
"encoding/json"
"fmt"
"net/http"
"github.com/juju/errors"
htr "github.com/julienschmidt/httprouter"
"github.com/segmentio/go-log"
)
// API defines how a new API will be attached to the router.
// api packages should export a function such as:
//
// func Auth(d db.DB) API {
// return func(r *htr.Router) {
// r.GET("/user/auth", handleAuth(d))
// }
// }
type API func(*htr.Router) error
func SetupRoutes(r *htr.Router, apis ...API) error {
for _, a := range apis {
err := a(r)
if err != nil {
return err
}
}
return nil
}
type apiError struct {
Code int `json:"code,omitempty"`
Message string `json:"msg,omitempty"`
}
func newApiError(msg string, err error) apiError {
e := apiError{
Message: msg,
Code: http.StatusInternalServerError,
}
if errors.IsNotFound(err) {
e.Message = fmt.Sprintf("not found: %s", msg)
e.Code = http.StatusNotFound
}
return e
}
func WriteResponse(w http.ResponseWriter, values ...interface{}) {
e := json.NewEncoder(w)
response := newResponse(values...)
if err := e.Encode(response); err != nil {
code := http.StatusInternalServerError
log.Error("failed to write response", err)
http.Error(
w,
newFatalResponse("failed to write response", code),
code,
)
}
}
type response struct {
Code int `json:"code,omitempty"`
Message string `json:"msg,omitempty"`
Values []interface{} `json:"values,omitempty"`
}
func newResponse(v ...interface{}) response {
return response{
Values: v,
}
}
func newFatalResponse(msg string, code int) string {
r, _ := json.Marshal(response{
Code: code,
Message: msg,
})
return string(r)
}