highperfocused/lnd
1
0
Fork 0
mirror of https://github.com/lightningnetwork/lnd.git synced 2025-09-12 04:55:43 +02:00
Code Issues Packages Projects Releases Wiki Activity

graph: update proof by ID

Browse Source 
This commit restricts the graph CRUD interface such that one can only
add a proof to a channel announcement and not update any other fields.
This pattern is better suited for SQL land too.
This commit is contained in:
Elle Mouton
2025-02-01 13:26:48 +02:00
parent e58abbf0e5
commit 9df8773163
3 changed files with 15 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
graph/builder.go
View File

@@ -1441,14 +1441,7 @@ func (b *Builder) ForAllOutgoingChannels(cb func(*models.ChannelEdgeInfo,
func (b *Builder) AddProof(chanID lnwire.ShortChannelID,
proof *models.ChannelAuthProof) error {
info, _, _, err := b.cfg.Graph.FetchChannelEdgesByID(chanID.ToUint64())
if err != nil {
return err
}
info.AuthProof = proof
return b.cfg.Graph.UpdateChannelEdge(info)
return b.cfg.Graph.AddEdgeProof(chanID, proof)
}
// IsStaleNode returns true if the graph source has a node announcement for the

23
graph/db/graph.go
View File

@@ -1297,15 +1297,13 @@ func (c *ChannelGraph) HasChannelEdge(
return upd1Time, upd2Time, exists, isZombie, nil
}
// UpdateChannelEdge retrieves and update edge of the graph database. Method
// only reserved for updating an edge info after its already been created.
// In order to maintain this constraints, we return an error in the scenario
// that an edge info hasn't yet been created yet, but someone attempts to update
// it.
func (c *ChannelGraph) UpdateChannelEdge(edge *models.ChannelEdgeInfo) error {
// AddEdgeProof sets the proof of an existing edge in the graph database.
func (c *ChannelGraph) AddEdgeProof(chanID lnwire.ShortChannelID,
proof *models.ChannelAuthProof) error {
// Construct the channel's primary key which is the 8-byte channel ID.
var chanKey [8]byte
binary.BigEndian.PutUint64(chanKey[:], edge.ChannelID)
binary.BigEndian.PutUint64(chanKey[:], chanID.ToUint64())
return kvdb.Update(c.db, func(tx kvdb.RwTx) error {
edges := tx.ReadWriteBucket(edgeBucket)
@@ -1318,15 +1316,14 @@ func (c *ChannelGraph) UpdateChannelEdge(edge *models.ChannelEdgeInfo) error {
return ErrEdgeNotFound
}
if edgeInfo := edgeIndex.Get(chanKey[:]); edgeInfo == nil {
return ErrEdgeNotFound
edge, err := fetchChanEdgeInfo(edgeIndex, chanKey[:])
if err != nil {
return err
}
if c.graphCache != nil {
c.graphCache.UpdateChannel(edge)
}
edge.AuthProof = proof
return putChanEdgeInfo(edgeIndex, edge, chanKey)
return putChanEdgeInfo(edgeIndex, &edge, chanKey)
}, func() {})
}

10
graph/interfaces.go
View File

@@ -259,12 +259,10 @@ type DB interface {
*models.ChannelEdgePolicy,
*models.ChannelEdgePolicy) error) error
// UpdateChannelEdge retrieves and update edge of the graph database.
// Method only reserved for updating an edge info after its already been
// created. In order to maintain this constraints, we return an error in
// the scenario that an edge info hasn't yet been created yet, but
// someone attempts to update it.
UpdateChannelEdge(edge *models.ChannelEdgeInfo) error
// AddEdgeProof sets the proof of an existing edge in the graph
// database.
AddEdgeProof(chanID lnwire.ShortChannelID,
proof *models.ChannelAuthProof) error
// IsPublicNode is a helper method that determines whether the node with
// the given public key is seen as a public node in the graph from the