mirrors/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2024-11-19 09:53:54 +01:00
Code Issues Projects Releases Wiki Activity

channeldb: add new db field for ChannelEdgeInfo to allow new methods

Browse Source
This commit is contained in:
Olaoluwa Osuntokun 2018-08-09 20:45:33 -07:00
parent 29b6bae4ed
commit 0109672835
No known key found for this signature in database
GPG Key ID: 964EA263DD637C21
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
channeldb/graph.go
View File

@ -191,6 +191,7 @@ func (c *ChannelGraph) ForEachChannel(cb func(*ChannelEdgeInfo, *ChannelEdgePoli
if err != nil { if err != nil {
return err return err
} }
edgeInfo.db = c.db
edge1, edge2, err := fetchChanEdgePolicies( edge1, edge2, err := fetchChanEdgePolicies(
edgeIndex, edges, nodes, chanID, c.db, edgeIndex, edges, nodes, chanID, c.db,
@ -1194,6 +1195,7 @@ func (c *ChannelGraph) ChanUpdatesInHorizon(startTime, endTime time.Time) ([]Cha
if err != nil { if err != nil {
return err return err
} }
edgeInfo.db = c.db
// With the static information obtained, we'll now // With the static information obtained, we'll now
// fetch the dynamic policy info. // fetch the dynamic policy info.
@ -1432,6 +1434,7 @@ func (c *ChannelGraph) FetchChanInfos(chanIDs []uint64) ([]ChannelEdge, error) {
if err != nil { if err != nil {
return err return err
} }
edgeInfo.db = c.db
// With the static information obtained, we'll now // With the static information obtained, we'll now
// fetch the dynamic policy info. // fetch the dynamic policy info.
@ -1835,6 +1838,7 @@ func (l *LightningNode) ForEachChannel(tx *bolt.Tx,
if err != nil { if err != nil {
return err return err
} }
edgeInfo.db = l.db
outgoingPolicy, err := fetchChanEdgePolicy( outgoingPolicy, err := fetchChanEdgePolicy(
edges, chanID, nodePub, nodes, edges, chanID, nodePub, nodes,
@ -1921,6 +1925,8 @@ type ChannelEdgeInfo struct {
// Capacity is the total capacity of the channel, this is determined by // Capacity is the total capacity of the channel, this is determined by
// the value output in the outpoint that created this channel. // the value output in the outpoint that created this channel.
Capacity btcutil.Amount Capacity btcutil.Amount
db *DB
} }
// AddNodeKeys is a setter-like method that can be used to replace the set of // AddNodeKeys is a setter-like method that can be used to replace the set of
@ -2035,7 +2041,7 @@ func (c *ChannelEdgeInfo) OtherNodeKeyBytes(thisNodeKey []byte) (
case bytes.Equal(c.NodeKey2Bytes[:], thisNodeKey): case bytes.Equal(c.NodeKey2Bytes[:], thisNodeKey):
return c.NodeKey1Bytes[:], nil return c.NodeKey1Bytes[:], nil
default: default:
return nil, fmt.Errorf("Node not participating in this channel") return nil, fmt.Errorf("node not participating in this channel")
} }
} }
@ -2293,6 +2299,7 @@ func (c *ChannelGraph) FetchChannelEdgesByOutpoint(op *wire.OutPoint) (*ChannelE
return err return err
} }
edgeInfo = &edge edgeInfo = &edge
edgeInfo.db = c.db
// Once we have the information about the channels' parameters, // Once we have the information about the channels' parameters,
// we'll fetch the routing policies for each for the directed // we'll fetch the routing policies for each for the directed
@ -2356,6 +2363,7 @@ func (c *ChannelGraph) FetchChannelEdgesByID(chanID uint64) (*ChannelEdgeInfo, *
return err return err
} }
edgeInfo = &edge edgeInfo = &edge
edgeInfo.db = c.db
e1, e2, err := fetchChanEdgePolicies( e1, e2, err := fetchChanEdgePolicies(
edgeIndex, edges, nodes, channelID[:], c.db, edgeIndex, edges, nodes, channelID[:], c.db,