-
Notifications
You must be signed in to change notification settings - Fork 1
/
actions_effects.go
192 lines (163 loc) · 4.91 KB
/
actions_effects.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
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
package aurora
import (
"fmt"
"regexp"
"github.com/diamnet/go/services/aurora/internal/actions"
"github.com/diamnet/go/services/aurora/internal/db2"
"github.com/diamnet/go/services/aurora/internal/db2/history"
"github.com/diamnet/go/services/aurora/internal/render/sse"
"github.com/diamnet/go/services/aurora/internal/resourceadapter"
"github.com/diamnet/go/support/errors"
"github.com/diamnet/go/support/render/hal"
"github.com/diamnet/go/support/render/problem"
)
// This file contains the actions:
//
// EffectIndexAction: pages of effects
// Interface verifications
var _ actions.JSONer = (*EffectIndexAction)(nil)
var _ actions.EventStreamer = (*EffectIndexAction)(nil)
// EffectIndexAction renders a page of effect resources, identified by
// a normal page query and optionally filtered by an account, ledger,
// transaction, or operation.
type EffectIndexAction struct {
Action
AccountFilter string
LedgerFilter int32
TransactionFilter string
OperationFilter int64
PagingParams db2.PageQuery
Records []history.Effect
Page hal.Page
Ledgers *history.LedgerCache
}
// JSON is a method for actions.JSON
func (action *EffectIndexAction) JSON() error {
action.Do(
action.EnsureHistoryFreshness,
action.loadParams,
action.ValidateCursorWithinHistory,
action.loadRecords,
action.loadLedgers,
action.loadPage,
func() { hal.Render(action.W, action.Page) },
)
return action.Err
}
// SSE is a method for actions.SSE
func (action *EffectIndexAction) SSE(stream *sse.Stream) error {
action.Setup(
action.EnsureHistoryFreshness,
action.loadParams,
action.ValidateCursorWithinHistory,
)
action.Do(
action.loadRecords,
action.loadLedgers,
func() {
stream.SetLimit(int(action.PagingParams.Limit))
records := action.Records[stream.SentCount():]
for _, record := range records {
ledger, found := action.Ledgers.Records[record.LedgerSequence()]
if !found {
action.Err = errors.New(fmt.Sprintf("could not find ledger data for sequence %d", record.LedgerSequence()))
return
}
res, err := resourceadapter.NewEffect(action.R.Context(), record, ledger)
if err != nil {
action.Err = err
return
}
stream.Send(sse.Event{
ID: res.PagingToken(),
Data: res,
})
}
},
)
return action.Err
}
// loadLedgers populates the ledger cache for this action
func (action *EffectIndexAction) loadLedgers() {
action.Ledgers = &history.LedgerCache{}
for _, eff := range action.Records {
action.Ledgers.Queue(eff.LedgerSequence())
}
action.Err = action.Ledgers.Load(action.HistoryQ())
}
func (action *EffectIndexAction) loadParams() {
action.ValidateCursor()
action.PagingParams = action.GetPageQuery()
action.AccountFilter = action.GetAddress("account_id")
action.LedgerFilter = action.GetInt32("ledger_id")
action.TransactionFilter = action.GetString("tx_id")
action.OperationFilter = action.GetInt64("op_id")
filters, err := countNonEmpty(
action.AccountFilter,
action.LedgerFilter,
action.TransactionFilter,
action.OperationFilter,
)
if err != nil {
action.Err = errors.Wrap(err, "Error in countNonEmpty")
return
}
if filters > 1 {
action.Err = problem.BadRequest
return
}
}
// loadRecords populates action.Records
func (action *EffectIndexAction) loadRecords() {
effects := action.HistoryQ().Effects()
switch {
case action.AccountFilter != "":
effects.ForAccount(action.AccountFilter)
case action.LedgerFilter > 0:
effects.ForLedger(action.LedgerFilter)
case action.OperationFilter > 0:
effects.ForOperation(action.OperationFilter)
case action.TransactionFilter != "":
effects.ForTransaction(action.TransactionFilter)
}
action.Err = effects.Page(action.PagingParams).Select(&action.Records)
}
// loadPage populates action.Page
func (action *EffectIndexAction) loadPage() {
for _, record := range action.Records {
ledger, found := action.Ledgers.Records[record.LedgerSequence()]
if !found {
msg := fmt.Sprintf("could not find ledger data for sequence %d", record.LedgerSequence())
action.Err = errors.New(msg)
return
}
var res hal.Pageable
res, action.Err = resourceadapter.NewEffect(action.R.Context(), record, ledger)
if action.Err != nil {
return
}
action.Page.Add(res)
}
action.Page.FullURL = action.FullURL()
action.Page.Limit = action.PagingParams.Limit
action.Page.Cursor = action.PagingParams.Cursor
action.Page.Order = action.PagingParams.Order
action.Page.PopulateLinks()
}
// ValidateCursor ensures that the provided cursor parameter is of the form
// OPERATIONID-INDEX (such as 1234-56) or is the special value "now" that
// represents the the cursor directly after the last closed ledger
func (action *EffectIndexAction) ValidateCursor() {
c := action.GetString("cursor")
if c == "" {
return
}
ok, err := regexp.MatchString("now|\\d+(-\\d+)?", c)
if err != nil {
action.Err = err
return
}
if !ok {
action.SetInvalidField("cursor", errors.New("invalid format"))
}
}