-
Notifications
You must be signed in to change notification settings - Fork 1
/
revocations_resource.go
79 lines (70 loc) · 2.2 KB
/
revocations_resource.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
package server
import (
"encoding/json"
"github.com/leanovate/microzon-auth-go/logging"
"github.com/leanovate/microzon-auth-go/revocations"
"github.com/untoldwind/routing"
"net/http"
"time"
)
type revocationssResource struct {
revocationsManager *revocations.RevocationsManager
logger logging.Logger
}
func RevocationsRoutes(revocationsManager *revocations.RevocationsManager, parent logging.Logger) routing.Matcher {
logger := parent.WithContext(map[string]interface{}{"resource": "revocations"})
resource := &revocationssResource{
revocationsManager: revocationsManager,
logger: logger,
}
return routing.PrefixSeq("/revocations",
routing.EndSeq(
routing.GETFunc(wrap(resource.logger, resource.QueryRevocations)),
SendError(logger, MethodNotAllowed()),
),
routing.PrefixSeq("/stream",
routing.EndSeq(
routing.GETFunc(resource.StreamRevocations),
SendError(logger, MethodNotAllowed()),
),
),
)
}
func (r *revocationssResource) QueryRevocations(req *http.Request) (interface{}, error) {
sinceVersion, err := queryParamUint(req, "since_version", 0)
if err != nil {
return nil, BadRequest()
}
wait, err := queryParamBool(req, "wait", false)
if err != nil {
return nil, BadRequest()
}
timeout, err := queryParamUint(req, "timeout", 0)
if err != nil {
return nil, BadRequest()
}
if wait {
<-r.revocationsManager.Observe.AddObserverWithTimeout(revocations.ObserveState(sinceVersion), time.Duration(timeout)*time.Second)
}
return r.revocationsManager.GetRevocationsSinceVersion(sinceVersion, 200), nil
}
func (r *revocationssResource) StreamRevocations(resp http.ResponseWriter, req *http.Request) {
sinceVersion, err := queryParamUint(req, "since_version", 0)
if err != nil {
encodeError(r.logger, resp, req, BadRequest())
return
}
resp.WriteHeader(200)
encoder := json.NewEncoder(resp)
for {
<-r.revocationsManager.Observe.AddObserver(revocations.ObserveState(sinceVersion))
chunk := r.revocationsManager.GetRevocationsSinceVersion(sinceVersion, 10)
sinceVersion = chunk.LastVersion
for _, revocation := range chunk.Revocations {
encoder.Encode(revocation)
}
if flusher, ok := resp.(http.Flusher); ok {
flusher.Flush()
}
}
}