graph/db: gracefully handle duplicate node announcements

It can happen that we are handling 2 of the same node announcements in
the same batch transaction. In that case, our `UpsertNode` conflict
assertion may fail. We need to handle this gracefully.
This commit is contained in:
Elle Mouton committed 2025-10-06 11:28:33 +02:00
1 parent 88050da220
commit abb9654856
2 files changed
+19

No files matched your search

+8
View File
@@ -125,6 +125,14 @@ func TestNodeInsertionAndDeletion(t *testing.T) {
require.NoError(t, graph.AddNode(ctx, node))
assertNodeInCache(t, graph, node, testFeatures)
// Our AddNode implementation uses the batcher meaning that it is
// possible that two updates for the same node announcement may be
// processed in the same batch. So to avoid the conflict error (since we
// require at the DB level that the new timestamp is strictly
// greater than the previous one), we need to gracefully handle the
// case where the exact same node announcement is added twice.
require.NoError(t, graph.AddNode(ctx, node))
// Next, fetch the node from the database to ensure everything was
// serialized properly.
dbNode, err := graph.FetchNode(ctx, testPub)
+11
View File
@@ -255,6 +255,17 @@ func (s *SQLStore) AddNode(ctx context.Context,
Opts: batch.NewSchedulerOptions(opts...),
Do: func(queries SQLQueries) error {
_, err := upsertNode(ctx, queries, node)
// It is possible that two of the same node
// announcements are both being processed in the same
// batch. This may case the UpsertNode conflict to
// be hit since we require at the db layer that the
// new last_update is greater than the existing
// last_update. We need to gracefully handle this here.
if errors.Is(err, sql.ErrNoRows) {
return nil
}
return err
},
}