mirror of
https://github.com/lightningnetwork/lnd.git
synced 2024-11-19 09:53:54 +01:00
8c0df98439
In this commit, we completely remove the Router's dependence on a Graph source that requires a `kvdb.RTx`. In so doing, we are more prepared for a future where the Graph source is backed by different DB structure such as pure SQL. The two areas affected here are: the ChannelRouter's graph access that it uses for pathfinding. And the SessionSource's graph access that it uses for payments. The ChannelRouter gets given a Graph and the SessionSource is given a GraphSessionFactory which it can use to create a new session. Behind the scenes, this will acquire a kvdb.RTx that will be used for calls to the Graph's `ForEachNodeChannel` method.
66 lines
2.1 KiB
Go
66 lines
2.1 KiB
Go
package routing
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/btcsuite/btcd/btcutil"
|
|
"github.com/lightningnetwork/lnd/channeldb"
|
|
"github.com/lightningnetwork/lnd/lnwire"
|
|
"github.com/lightningnetwork/lnd/routing/route"
|
|
)
|
|
|
|
// Graph is an abstract interface that provides information about nodes and
|
|
// edges to pathfinding.
|
|
type Graph interface {
|
|
// ForEachNodeChannel calls the callback for every channel of the given
|
|
// node.
|
|
ForEachNodeChannel(nodePub route.Vertex,
|
|
cb func(channel *channeldb.DirectedChannel) error) error
|
|
|
|
// FetchNodeFeatures returns the features of the given node.
|
|
FetchNodeFeatures(nodePub route.Vertex) (*lnwire.FeatureVector, error)
|
|
}
|
|
|
|
// GraphSessionFactory can be used to produce a new Graph instance which can
|
|
// then be used for a path-finding session. Depending on the implementation,
|
|
// the Graph session will represent a DB connection where a read-lock is being
|
|
// held across calls to the backing Graph.
|
|
type GraphSessionFactory interface {
|
|
// NewGraphSession will produce a new Graph to use for a path-finding
|
|
// session. It returns the Graph along with a call-back that must be
|
|
// called once Graph access is complete. This call-back will close any
|
|
// read-only transaction that was created at Graph construction time.
|
|
NewGraphSession() (Graph, func() error, error)
|
|
}
|
|
|
|
// FetchAmountPairCapacity determines the maximal public capacity between two
|
|
// nodes depending on the amount we try to send.
|
|
func FetchAmountPairCapacity(graph Graph, source, nodeFrom, nodeTo route.Vertex,
|
|
amount lnwire.MilliSatoshi) (btcutil.Amount, error) {
|
|
|
|
// Create unified edges for all incoming connections.
|
|
//
|
|
// Note: Inbound fees are not used here because this method is only used
|
|
// by a deprecated router rpc.
|
|
u := newNodeEdgeUnifier(source, nodeTo, false, nil)
|
|
|
|
err := u.addGraphPolicies(graph)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
edgeUnifier, ok := u.edgeUnifiers[nodeFrom]
|
|
if !ok {
|
|
return 0, fmt.Errorf("no edge info for node pair %v -> %v",
|
|
nodeFrom, nodeTo)
|
|
}
|
|
|
|
edge := edgeUnifier.getEdgeNetwork(amount, 0)
|
|
if edge == nil {
|
|
return 0, fmt.Errorf("no edge for node pair %v -> %v "+
|
|
"(amount %v)", nodeFrom, nodeTo, amount)
|
|
}
|
|
|
|
return edge.capacity, nil
|
|
}
|