forked from nytimes/gizmo
/
get.go
62 lines (54 loc) · 1.49 KB
/
get.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
package readinglist
import (
"context"
"net/http"
"strconv"
ocontext "golang.org/x/net/context"
"github.com/NYTimes/gizmo/server/kit"
"github.com/pkg/errors"
)
// gRPC stub
func (s service) GetListLimit(ctx ocontext.Context, r *GetListLimitRequest) (*Links, error) {
res, err := s.getLinks(ctx, r)
if err != nil {
return nil, err
}
return res.(*Links), nil
}
// go-kit endpoint.Endpoint with core business logic
func (s service) getLinks(ctx context.Context, req interface{}) (interface{}, error) {
r := req.(*GetListLimitRequest)
// set request defaults
if r.Limit == 0 {
r.Limit = 50
}
// set maximum
if r.Limit > 100 {
r.Limit = 100
}
// get data from the service-injected DB interface
links, err := s.db.GetLinks(ctx, getUser(ctx), int(r.Limit))
if err != nil {
kit.LogErrorMsg(ctx, err, "error getting links from DB")
return nil, kit.NewJSONStatusResponse(
&Message{"server error"},
http.StatusInternalServerError)
}
lks := make([]*Link, len(links))
for i, l := range links {
lks[i] = &Link{Url: l}
}
return &Links{Links: lks}, errors.Wrap(err, "unable to get links")
}
// request decoder can be used for proto and JSON since there is no body
func decodeGetRequest(ctx context.Context, r *http.Request) (interface{}, error) {
limit, err := strconv.ParseInt(kit.Vars(r)["limit"], 10, 64)
if err != nil {
return nil, kit.NewJSONStatusResponse(
&Message{"bad request"},
http.StatusBadRequest)
}
return &GetListLimitRequest{
Limit: int32(limit),
}, nil
}