-
Notifications
You must be signed in to change notification settings - Fork 1
/
payment.go
70 lines (59 loc) · 1.66 KB
/
payment.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
package store
import (
"github.com/dgraph-io/badger/v3"
"github.com/timshannon/badgerhold/v4"
"github.com/c13n-io/c13n-go/model"
)
// AddPayments stores a list of payments.
func (db *bhDatabase) AddPayments(payments ...*model.Payment) error {
if len(payments) <= 0 {
return nil
}
return retryConflicts(db.bh.Badger().Update, func(txn *badger.Txn) error {
for _, payment := range payments {
paymentKey := payment.PaymentIndex
switch err := db.bh.TxInsert(txn, paymentKey, payment); err {
case nil:
// On success, continue
case badgerhold.ErrKeyExists:
return alreadyExists(payment)
default:
return err
}
}
return nil
})
}
// GetLastPaymentIndex retrieves the last payment index present in the database.
func (db *bhDatabase) GetLastPaymentIndex() (paymentIdx uint64, err error) {
p := new(model.Payment)
switch result, err := db.bh.FindAggregate(p, nil); err {
case nil:
if result[0].Count() > 0 {
result[0].Max("PaymentIndex", p)
return p.PaymentIndex, nil
}
default:
return 0, err
}
return
}
// GetPayments retrieves payments, based on the provided pagination options.
func (db *bhDatabase) GetPayments(pageOpts model.PageOptions) ([]*model.Payment, error) {
q := &badgerhold.Query{}
switch {
case pageOpts.LastID != 0 && pageOpts.Reverse:
q = badgerhold.Where("PaymentIndex").Le(pageOpts.LastID)
case pageOpts.LastID != 0:
q = badgerhold.Where("PaymentIndex").Ge(pageOpts.LastID)
}
q = q.SortBy("PaymentIndex").Limit(int(pageOpts.PageSize))
if pageOpts.Reverse {
q = q.Reverse()
}
var payments []*model.Payment
if err := db.bh.Find(&payments, q); err != nil {
return nil, err
}
return payments, nil
}