Skip to content

Commit 26afc39

Browse files
committed
multi: unify all payment related kv data.
All interactions related to the payment db are now part of the kvPaymentDB struct.
1 parent e714b31 commit 26afc39

File tree

2 files changed

+16
-12
lines changed

2 files changed

+16
-12
lines changed

channeldb/payments.go

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -426,7 +426,7 @@ func (p *KVPaymentsDB) InitPayment(paymentHash lntypes.Hash,
426426
func (p *KVPaymentsDB) DeleteFailedAttempts(hash lntypes.Hash) error {
427427
if !p.db.keepFailedPaymentAttempts {
428428
const failedHtlcsOnly = true
429-
err := p.db.DeletePayment(hash, failedHtlcsOnly)
429+
err := p.DeletePayment(hash, failedHtlcsOnly)
430430
if err != nil {
431431
return err
432432
}
@@ -1047,10 +1047,10 @@ func htlcBucketKey(prefix, id []byte) []byte {
10471047
// FetchPayments returns all sent payments found in the DB.
10481048
//
10491049
// nolint: dupl
1050-
func (d *DB) FetchPayments() ([]*MPPayment, error) {
1050+
func (p *KVPaymentsDB) FetchPayments() ([]*MPPayment, error) {
10511051
var payments []*MPPayment
10521052

1053-
err := kvdb.View(d, func(tx kvdb.RTx) error {
1053+
err := kvdb.View(p.db, func(tx kvdb.RTx) error {
10541054
paymentsBucket := tx.ReadBucket(paymentsRootBucket)
10551055
if paymentsBucket == nil {
10561056
return nil
@@ -1353,10 +1353,12 @@ type PaymentsResponse struct {
13531353
// QueryPayments is a query to the payments database which is restricted
13541354
// to a subset of payments by the payments query, containing an offset
13551355
// index and a maximum number of returned payments.
1356-
func (d *DB) QueryPayments(query PaymentsQuery) (PaymentsResponse, error) {
1356+
func (p *KVPaymentsDB) QueryPayments(query PaymentsQuery) (PaymentsResponse,
1357+
error) {
1358+
13571359
var resp PaymentsResponse
13581360

1359-
if err := kvdb.View(d, func(tx kvdb.RTx) error {
1361+
if err := kvdb.View(p.db, func(tx kvdb.RTx) error {
13601362
// Get the root payments bucket.
13611363
paymentsBucket := tx.ReadBucket(paymentsRootBucket)
13621364
if paymentsBucket == nil {
@@ -1570,10 +1572,10 @@ func fetchPaymentWithSequenceNumber(tx kvdb.RTx, paymentHash lntypes.Hash,
15701572
// DeletePayment deletes a payment from the DB given its payment hash. If
15711573
// failedHtlcsOnly is set, only failed HTLC attempts of the payment will be
15721574
// deleted.
1573-
func (d *DB) DeletePayment(paymentHash lntypes.Hash,
1575+
func (p *KVPaymentsDB) DeletePayment(paymentHash lntypes.Hash,
15741576
failedHtlcsOnly bool) error {
15751577

1576-
return kvdb.Update(d, func(tx kvdb.RwTx) error {
1578+
return kvdb.Update(p.db, func(tx kvdb.RwTx) error {
15771579
payments := tx.ReadWriteBucket(paymentsRootBucket)
15781580
if payments == nil {
15791581
return nil
@@ -1662,9 +1664,11 @@ func (d *DB) DeletePayment(paymentHash lntypes.Hash,
16621664
// failedHtlcsOnly is set, the payment itself won't be deleted, only failed HTLC
16631665
// attempts. The method returns the number of deleted payments, which is always
16641666
// 0 if failedHtlcsOnly is set.
1665-
func (d *DB) DeletePayments(failedOnly, failedHtlcsOnly bool) (int, error) {
1667+
func (p *KVPaymentsDB) DeletePayments(failedOnly, failedHtlcsOnly bool) (int,
1668+
error) {
1669+
16661670
var numPayments int
1667-
err := kvdb.Update(d, func(tx kvdb.RwTx) error {
1671+
err := kvdb.Update(p.db, func(tx kvdb.RwTx) error {
16681672
payments := tx.ReadWriteBucket(paymentsRootBucket)
16691673
if payments == nil {
16701674
return nil

rpcserver.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7403,7 +7403,7 @@ func (r *rpcServer) ListPayments(ctx context.Context,
74037403
query.MaxPayments = math.MaxUint64
74047404
}
74057405

7406-
paymentsQuerySlice, err := r.server.miscDB.QueryPayments(query)
7406+
paymentsQuerySlice, err := r.server.kvPaymentsDB.QueryPayments(query)
74077407
if err != nil {
74087408
return nil, err
74097409
}
@@ -7445,7 +7445,7 @@ func (r *rpcServer) DeletePayment(ctx context.Context,
74457445
rpcsLog.Infof("[DeletePayment] payment_identifier=%v, "+
74467446
"failed_htlcs_only=%v", hash, req.FailedHtlcsOnly)
74477447

7448-
err = r.server.miscDB.DeletePayment(hash, req.FailedHtlcsOnly)
7448+
err = r.server.kvPaymentsDB.DeletePayment(hash, req.FailedHtlcsOnly)
74497449
if err != nil {
74507450
return nil, err
74517451
}
@@ -7485,7 +7485,7 @@ func (r *rpcServer) DeleteAllPayments(ctx context.Context,
74857485
"failed_htlcs_only=%v", req.FailedPaymentsOnly,
74867486
req.FailedHtlcsOnly)
74877487

7488-
numDeletedPayments, err := r.server.miscDB.DeletePayments(
7488+
numDeletedPayments, err := r.server.kvPaymentsDB.DeletePayments(
74897489
req.FailedPaymentsOnly, req.FailedHtlcsOnly,
74907490
)
74917491
if err != nil {

0 commit comments

Comments
 (0)