channeldb/db: adds FetchClosedChannel

This commit adds a FetchClosedChannel method to the channeldb, which allows querying based on a known channel point. This will be used in the nursery to load channel close summaries, which can be used to provide more accurate height hints when recovering from failures.
ProofOfKeags · Jan 24, 2018 · 926949e · 926949e
1 parent 7bf057c
commit 926949e
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/channeldb/db.go b/channeldb/db.go
@@ -9,6 +9,7 @@ import (
 	"sync"
 
 	"github.com/boltdb/bolt"
+	"github.com/go-errors/errors"
 	"github.com/roasbeef/btcd/btcec"
 	"github.com/roasbeef/btcd/wire"
 )
@@ -434,6 +435,42 @@ func (d *DB) FetchClosedChannels(pendingOnly bool) ([]*ChannelCloseSummary, erro
 	return chanSummaries, nil
 }
 
+// ErrClosedChannelNotFound signals that a closed channel could not be found in
+// the channeldb.
+var ErrClosedChannelNotFound = errors.New("unable to find closed channel summary")
+
+// FetchClosedChannel queries for a channel close summary using the channel
+// point of the channel in question.
+func (d *DB) FetchClosedChannel(chanID *wire.OutPoint) (*ChannelCloseSummary, error) {
+	var chanSummary *ChannelCloseSummary
+	if err := d.View(func(tx *bolt.Tx) error {
+		closeBucket := tx.Bucket(closedChannelBucket)
+		if closeBucket == nil {
+			return ErrClosedChannelNotFound
+		}
+
+		var b bytes.Buffer
+		var err error
+		if err = writeOutpoint(&b, chanID); err != nil {
+			return err
+		}
+
+		summaryBytes := closeBucket.Get(b.Bytes())
+		if summaryBytes == nil {
+			return ErrClosedChannelNotFound
+		}
+
+		summaryReader := bytes.NewReader(summaryBytes)
+		chanSummary, err = deserializeCloseChannelSummary(summaryReader)
+
+		return err
+	}); err != nil {
+		return nil, err
+	}
+
+	return chanSummary, nil
+}
+
 // MarkChanFullyClosed marks a channel as fully closed within the database. A
 // channel should be marked as fully closed if the channel was initially
 // cooperatively closed and it's reach a single confirmation, or after all the