mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-03-03 17:26:57 +01:00
Code Issues Projects Releases Wiki Activity

channeldb: add serializeAndStoreInvoice helper func

Browse source
This commit is contained in:
positiveblue 2023-03-06 16:23:38 -08:00
parent 65fc8d72c6
commit 09e87ea99b
No known key found for this signature in database
GPG key ID: 4FFF2510928804DC
1 changed files with 24 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
channeldb/invoices.go
View file

@ -1969,13 +1969,8 @@ func (d *DB) cancelHTLCs(invoices kvdb.RwBucket, invoiceNum []byte,
return nil, errors.New("cancel action on non-existent htlc(s)")
}
// Reserialize and update invoice.
var buf bytes.Buffer
if err := serializeInvoice(&buf, invoice); err != nil {
return nil, err
}
if err := invoices.Put(invoiceNum, buf.Bytes()); err != nil {
err := d.serializeAndStoreInvoice(invoices, invoiceNum, invoice)
if err != nil {
return nil, err
}
@ -1992,6 +1987,22 @@ func (d *DB) cancelHTLCs(invoices kvdb.RwBucket, invoiceNum []byte,
return invoice, nil
}
// serializeAndStoreInvoice is a helper function used to store invoices.
func (d *DB) serializeAndStoreInvoice(invoices kvdb.RwBucket, invoiceNum []byte,
invoice *invpkg.Invoice) error {
var buf bytes.Buffer
if err := serializeInvoice(&buf, invoice); err != nil {
return err
}
if err := invoices.Put(invoiceNum, buf.Bytes()); err != nil {
return err
}
return nil
}
// addHTLCs tries to add the htlcs in the given InvoiceUpdateDesc.
func (d *DB) addHTLCs(invoices, settleIndex, //nolint:funlen
setIDIndex kvdb.RwBucket, invoiceNum []byte, invoice *invpkg.Invoice,
@ -2219,13 +2230,8 @@ func (d *DB) addHTLCs(invoices, settleIndex, //nolint:funlen
}
}
// Reserialize and update invoice.
var buf bytes.Buffer
if err := serializeInvoice(&buf, invoice); err != nil {
return nil, err
}
if err := invoices.Put(invoiceNum, buf.Bytes()); err != nil {
err := d.serializeAndStoreInvoice(invoices, invoiceNum, invoice)
if err != nil {
return nil, err
}
@ -2307,13 +2313,8 @@ func (d *DB) settleHodlInvoice(invoices, settleIndex kvdb.RwBucket,
invoice.AmtPaid = amtPaid
// Reserialize and update invoice.
var buf bytes.Buffer
if err := serializeInvoice(&buf, invoice); err != nil {
return nil, err
}
if err := invoices.Put(invoiceNum, buf.Bytes()); err != nil {
err = d.serializeAndStoreInvoice(invoices, invoiceNum, invoice)
if err != nil {
return nil, err
}
@ -2369,13 +2370,8 @@ func (d *DB) cancelInvoice(invoices kvdb.RwBucket, invoiceNum []byte,
}
}
// Reserialize and update invoice.
var buf bytes.Buffer
if err := serializeInvoice(&buf, invoice); err != nil {
return nil, err
}
if err := invoices.Put(invoiceNum, buf.Bytes()); err != nil {
err = d.serializeAndStoreInvoice(invoices, invoiceNum, invoice)
if err != nil {
return nil, err
}