/
transports.go
85 lines (75 loc) · 2.5 KB
/
transports.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
85
package svc
import (
"context"
"encoding/json"
"errors"
"net/http"
"github.com/go-kit/kit/ratelimit"
httptransport "github.com/go-kit/kit/transport/http"
"github.com/gorilla/mux"
cache "github.com/patrickmn/go-cache"
"github.com/prometheus/client_golang/prometheus/promhttp"
"golang.org/x/time/rate"
)
var (
//ErrJSONUnMarshall indicates a bad request where json unmarshalling failed
ErrJSONUnMarshall = errors.New("failed to parse json")
)
// MakeHTTPHandler returns an http handler for the endpoints
func MakeHTTPHandler(svc NagiosParserSvc, cacher *cache.Cache, limiter *rate.Limiter) http.Handler {
r := mux.NewRouter()
ee := MakeServerEndpoints(svc, cacher, limiter)
options := []httptransport.ServerOption{
httptransport.ServerErrorEncoder(encodeError),
}
getParsedNagiosHandler := httptransport.NewServer(
ee.getParsedNagios,
decodeGetParsedNagiosRequest,
encodeGetParsedNagiosResponse,
options...,
)
r.Methods("GET").Path("/nagios").Handler(getParsedNagiosHandler)
refreshNagiosDataHandler := httptransport.NewServer(
ee.refreshNagiosData,
decodeRefreshNagiosDataRequest,
encodeRefreshNagiosDataResponse,
options...,
)
r.Methods("GET").Path("/refresh").Handler(refreshNagiosDataHandler)
r.Methods("GET").Path("/metrics").Handler(promhttp.Handler())
return r
}
func decodeRefreshNagiosDataRequest(_ context.Context, r *http.Request) (interface{}, error) {
// We return a blank request holder because no data must be taken in yet
return refreshNagiosDataRequest{}, nil
}
func encodeRefreshNagiosDataResponse(_ context.Context, w http.ResponseWriter, resp interface{}) error {
return json.NewEncoder(w).Encode(resp)
}
func decodeGetParsedNagiosRequest(_ context.Context, r *http.Request) (interface{}, error) {
// We return a blank request holder because no data must be taken in yet
return getParsedNagiosRequest{}, nil
}
func encodeGetParsedNagiosResponse(_ context.Context, w http.ResponseWriter, resp interface{}) error {
return json.NewEncoder(w).Encode(resp)
}
func encodeError(_ context.Context, err error, w http.ResponseWriter) {
if err == nil {
panic("encodeError with nil error")
}
w.Header().Set("Content-Type", "application/json; charset=utf-8")
w.WriteHeader(codeFrom(err))
json.NewEncoder(w).Encode(map[string]interface{}{
"error": err.Error(),
})
}
func codeFrom(err error) int {
switch err {
case ErrJSONUnMarshall:
return http.StatusBadRequest
case ratelimit.ErrLimited:
return http.StatusTooManyRequests
default:
return http.StatusInternalServerError
}
}