From ffb27284dfc31e8e6ce552d8f79d692b5c6a48d2 Mon Sep 17 00:00:00 2001 From: Andras Banki-Horvath Date: Mon, 19 Oct 2020 14:24:45 +0200 Subject: [PATCH] channeldb: change walletdb dependency to kvdb --- channeldb/peers.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/channeldb/peers.go b/channeldb/peers.go index fabb5361..4dea9f2f 100644 --- a/channeldb/peers.go +++ b/channeldb/peers.go @@ -6,7 +6,7 @@ import ( "fmt" "time" - "github.com/btcsuite/btcwallet/walletdb" + "github.com/lightningnetwork/lnd/channeldb/kvdb" "github.com/lightningnetwork/lnd/routing/route" ) @@ -50,7 +50,7 @@ type FlapCount struct { // bucket for the peer's pubkey if necessary. Note that this function overwrites // the current value. func (d *DB) WriteFlapCounts(flapCounts map[route.Vertex]*FlapCount) error { - return d.Update(func(tx walletdb.ReadWriteTx) error { + return kvdb.Update(d, func(tx kvdb.RwTx) error { // Run through our set of flap counts and record them for // each peer, creating a bucket for the peer pubkey if required. for peer, flapCount := range flapCounts { @@ -88,7 +88,7 @@ func (d *DB) WriteFlapCounts(flapCounts map[route.Vertex]*FlapCount) error { func (d *DB) ReadFlapCount(pubkey route.Vertex) (*FlapCount, error) { var flapCount FlapCount - if err := d.View(func(tx walletdb.ReadTx) error { + if err := kvdb.View(d, func(tx kvdb.RTx) error { peers := tx.ReadBucket(peersBucket) peerBucket := peers.NestedReadBucket(pubkey[:])