Skip to content

Commit e13b29c

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 cedd931 commit e13b29c

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
}
@@ -1045,10 +1045,10 @@ func htlcBucketKey(prefix, id []byte) []byte {
10451045
// FetchPayments returns all sent payments found in the DB.
10461046
//
10471047
// nolint: dupl
1048-
func (d *DB) FetchPayments() ([]*MPPayment, error) {
1048+
func (p *KVPaymentsDB) FetchPayments() ([]*MPPayment, error) {
10491049
var payments []*MPPayment
10501050

1051-
err := kvdb.View(d, func(tx kvdb.RTx) error {
1051+
err := kvdb.View(p.db, func(tx kvdb.RTx) error {
10521052
paymentsBucket := tx.ReadBucket(paymentsRootBucket)
10531053
if paymentsBucket == nil {
10541054
return nil
@@ -1351,10 +1351,12 @@ type PaymentsResponse struct {
13511351
// QueryPayments is a query to the payments database which is restricted
13521352
// to a subset of payments by the payments query, containing an offset
13531353
// index and a maximum number of returned payments.
1354-
func (d *DB) QueryPayments(query PaymentsQuery) (PaymentsResponse, error) {
1354+
func (p *KVPaymentsDB) QueryPayments(query PaymentsQuery) (PaymentsResponse,
1355+
error) {
1356+
13551357
var resp PaymentsResponse
13561358

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

1574-
return kvdb.Update(d, func(tx kvdb.RwTx) error {
1576+
return kvdb.Update(p.db, func(tx kvdb.RwTx) error {
15751577
payments := tx.ReadWriteBucket(paymentsRootBucket)
15761578
if payments == nil {
15771579
return nil
@@ -1660,9 +1662,11 @@ func (d *DB) DeletePayment(paymentHash lntypes.Hash,
16601662
// failedHtlcsOnly is set, the payment itself won't be deleted, only failed HTLC
16611663
// attempts. The method returns the number of deleted payments, which is always
16621664
// 0 if failedHtlcsOnly is set.
1663-
func (d *DB) DeletePayments(failedOnly, failedHtlcsOnly bool) (int, error) {
1665+
func (p *KVPaymentsDB) DeletePayments(failedOnly, failedHtlcsOnly bool) (int,
1666+
error) {
1667+
16641668
var numPayments int
1665-
err := kvdb.Update(d, func(tx kvdb.RwTx) error {
1669+
err := kvdb.Update(p.db, func(tx kvdb.RwTx) error {
16661670
payments := tx.ReadWriteBucket(paymentsRootBucket)
16671671
if payments == nil {
16681672
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)