htlcswitch/control_tower: use one db txn for transitions

Composes the new payment status helper methods such that we only require one db txn per state transition. This also allows us to remove the exclusive lock from the control tower, and enable more concurrent requests.
2024-11-19 18:10:34 +01:00 · 2018-08-20 21:14:52 -07:00 · 2018-08-20 21:14:52 -07:00 · 5dc2a4a4b8
commit 5dc2a4a4b8
parent 86b347c996
1 changed files with 144 additions and 86 deletions
--- a/htlcswitch/control_tower.go
+++ b/htlcswitch/control_tower.go
@ -2,8 +2,8 @@ package htlcswitch

 import (
 	"errors"
-	"sync"

+	"github.com/coreos/bbolt"
 	"github.com/lightningnetwork/lnd/channeldb"
 	"github.com/lightningnetwork/lnd/lnwire"
 )
@ -59,11 +59,17 @@ type ControlTower interface {
 type paymentControl struct {
 	strict bool

-	mx sync.Mutex
 	db *channeldb.DB
 }

-// NewPaymentControl creates a new instance of the paymentControl.
+// NewPaymentControl creates a new instance of the paymentControl. The strict
+// flag indicates whether the controller should require "strict" state
+// transitions, which would be otherwise intolerant to older databases that may
+// already have duplicate payments to the same payment hash. It should be
+// enabled only after sufficient checks have been made to ensure the db does not
+// contain such payments. In the meantime, non-strict mode enforces a superset
+// of the state transitions that prevent additional payments to a given payment
+// hash from being added.
 func NewPaymentControl(strict bool, db *channeldb.DB) ControlTower {
 	return &paymentControl{
 		strict: strict,
@ -74,114 +80,166 @@ func NewPaymentControl(strict bool, db *channeldb.DB) ControlTower {
 // ClearForTakeoff checks that we don't already have an InFlight or Completed
 // payment identified by the same payment hash.
 func (p *paymentControl) ClearForTakeoff(htlc *lnwire.UpdateAddHTLC) error {
-	p.mx.Lock()
-	defer p.mx.Unlock()
+	var takeoffErr error
+	err := p.db.Batch(func(tx *bolt.Tx) error {
+		// Retrieve current status of payment from local database.
+		paymentStatus, err := channeldb.FetchPaymentStatusTx(
+			tx, htlc.PaymentHash,
+		)
+		if err != nil {
+			return err
+		}

-	// Retrieve current status of payment from local database.
-	paymentStatus, err := p.db.FetchPaymentStatus(htlc.PaymentHash)
+		// Reset the takeoff error, to avoid carrying over an error
+		// from a previous execution of the batched db transaction.
+		takeoffErr = nil
+
+		switch paymentStatus {
+
+		case channeldb.StatusGrounded:
+			// It is safe to reattempt a payment if we know that we
+			// haven't left one in flight. Since this one is
+			// grounded, Transition the payment status to InFlight
+			// to prevent others.
+			return channeldb.UpdatePaymentStatusTx(
+				tx, htlc.PaymentHash, channeldb.StatusInFlight,
+			)
+
+		case channeldb.StatusInFlight:
+			// We already have an InFlight payment on the network. We will
+			// disallow any more payment until a response is received.
+			takeoffErr = ErrPaymentInFlight
+
+		case channeldb.StatusCompleted:
+			// We've already completed a payment to this payment hash,
+			// forbid the switch from sending another.
+			takeoffErr = ErrAlreadyPaid
+
+		default:
+			takeoffErr = ErrUnknownPaymentStatus
+		}
+
+		return nil
+	})
 	if err != nil {
 		return err
 	}

-	switch paymentStatus {
-
-	case channeldb.StatusGrounded:
-		// It is safe to reattempt a payment if we know that we haven't
-		// left one in flight. Since this one is grounded, Transition
-		// the payment status to InFlight to prevent others.
-		return p.db.UpdatePaymentStatus(htlc.PaymentHash, channeldb.StatusInFlight)
-
-	case channeldb.StatusInFlight:
-		// We already have an InFlight payment on the network. We will
-		// disallow any more payment until a response is received.
-		return ErrPaymentInFlight
-
-	case channeldb.StatusCompleted:
-		// We've already completed a payment to this payment hash,
-		// forbid the switch from sending another.
-		return ErrAlreadyPaid
-
-	default:
-		return ErrUnknownPaymentStatus
-	}
+	return takeoffErr
 }

 // Success transitions an InFlight payment to Completed, otherwise it returns an
 // error. After calling Success, ClearForTakeoff should prevent any further
 // attempts for the same payment hash.
 func (p *paymentControl) Success(paymentHash [32]byte) error {
-	p.mx.Lock()
-	defer p.mx.Unlock()
+	var updateErr error
+	err := p.db.Batch(func(tx *bolt.Tx) error {
+		paymentStatus, err := channeldb.FetchPaymentStatusTx(
+			tx, paymentHash,
+		)
+		if err != nil {
+			return err
+		}

-	paymentStatus, err := p.db.FetchPaymentStatus(paymentHash)
+		// Reset the update error, to avoid carrying over an error
+		// from a previous execution of the batched db transaction.
+		updateErr = nil
+
+		switch {
+
+		case paymentStatus == channeldb.StatusGrounded && p.strict:
+			// Our records show the payment as still being grounded,
+			// meaning it never should have left the switch.
+			updateErr = ErrPaymentNotInitiated
+
+		case paymentStatus == channeldb.StatusGrounded && !p.strict:
+			// Though our records show the payment as still being
+			// grounded, meaning it never should have left the
+			// switch, we permit this transition in non-strict mode
+			// to handle inconsistent db states.
+			fallthrough
+
+		case paymentStatus == channeldb.StatusInFlight:
+			// A successful response was received for an InFlight
+			// payment, mark it as completed to prevent sending to
+			// this payment hash again.
+			return channeldb.UpdatePaymentStatusTx(
+				tx, paymentHash, channeldb.StatusCompleted,
+			)
+
+		case paymentStatus == channeldb.StatusCompleted:
+			// The payment was completed previously, alert the
+			// caller that this may be a duplicate call.
+			updateErr = ErrPaymentAlreadyCompleted
+
+		default:
+			updateErr = ErrUnknownPaymentStatus
+		}
+
+		return nil
+	})
 	if err != nil {
 		return err
 	}

-	switch {
-
-	case paymentStatus == channeldb.StatusGrounded && p.strict:
-		// Our records show the payment as still being grounded, meaning
-		// it never should have left the switch.
-		return ErrPaymentNotInitiated
-
-	case paymentStatus == channeldb.StatusGrounded && !p.strict:
-		// Our records show the payment as still being grounded, meaning
-		// it never should have left the switch.
-		fallthrough
-
-	case paymentStatus == channeldb.StatusInFlight:
-		// A successful response was received for an InFlight payment,
-		// mark it as completed to prevent sending to this payment hash
-		// again.
-		return p.db.UpdatePaymentStatus(paymentHash, channeldb.StatusCompleted)
-
-	case paymentStatus == channeldb.StatusCompleted:
-		// The payment was completed previously, alert the caller that
-		// this may be a duplicate call.
-		return ErrPaymentAlreadyCompleted
-
-	default:
-		return ErrUnknownPaymentStatus
-	}
+	return updateErr
 }

 // Fail transitions an InFlight payment to Grounded, otherwise it returns an
 // error. After calling Fail, ClearForTakeoff should fail any further attempts
 // for the same payment hash.
 func (p *paymentControl) Fail(paymentHash [32]byte) error {
-	p.mx.Lock()
-	defer p.mx.Unlock()
+	var updateErr error
+	err := p.db.Batch(func(tx *bolt.Tx) error {
+		paymentStatus, err := channeldb.FetchPaymentStatusTx(
+			tx, paymentHash,
+		)
+		if err != nil {
+			return err
+		}

-	paymentStatus, err := p.db.FetchPaymentStatus(paymentHash)
+		// Reset the update error, to avoid carrying over an error
+		// from a previous execution of the batched db transaction.
+		updateErr = nil
+
+		switch {
+
+		case paymentStatus == channeldb.StatusGrounded && p.strict:
+			// Our records show the payment as still being grounded,
+			// meaning it never should have left the switch.
+			updateErr = ErrPaymentNotInitiated
+
+		case paymentStatus == channeldb.StatusGrounded && !p.strict:
+			// Though our records show the payment as still being
+			// grounded, meaning it never should have left the
+			// switch, we permit this transition in non-strict mode
+			// to handle inconsistent db states.
+			fallthrough
+
+		case paymentStatus == channeldb.StatusInFlight:
+			// A failed response was received for an InFlight
+			// payment, mark it as Grounded again to allow
+			// subsequent attempts.
+			return channeldb.UpdatePaymentStatusTx(
+				tx, paymentHash, channeldb.StatusGrounded,
+			)
+
+		case paymentStatus == channeldb.StatusCompleted:
+			// The payment was completed previously, and we are now
+			// reporting that it has failed. Leave the status as
+			// completed, but alert the user that something is
+			// wrong.
+			updateErr = ErrPaymentAlreadyCompleted
+
+		default:
+			updateErr = ErrUnknownPaymentStatus
+		}
+
+		return nil
+	})
 	if err != nil {
 		return err
 	}

-	switch {
-
-	case paymentStatus == channeldb.StatusGrounded && p.strict:
-		// Our records show the payment as still being grounded, meaning
-		// it never should have left the switch.
-		return ErrPaymentNotInitiated
-
-	case paymentStatus == channeldb.StatusGrounded && !p.strict:
-		// Our records show the payment as still being grounded, meaning
-		// it never should have left the switch.
-		fallthrough
-
-	case paymentStatus == channeldb.StatusInFlight:
-		// A failed response was received for an InFlight payment, mark
-		// it as Grounded again to allow subsequent attempts.
-		return p.db.UpdatePaymentStatus(paymentHash, channeldb.StatusGrounded)
-
-	case paymentStatus == channeldb.StatusCompleted:
-		// The payment was completed previously, and we are now
-		// reporting that it has failed. Leave the status as completed,
-		// but alert the user that something is wrong.
-		return ErrPaymentAlreadyCompleted
-
-	default:
-		return ErrUnknownPaymentStatus
-	}
+	return updateErr
 }