From 8349a6f84dc18ad7553788fce64b206ca74a2a91 Mon Sep 17 00:00:00 2001
From: Elle Mouton <elle.mouton@gmail.com>
Date: Wed, 26 Nov 2025 07:58:08 +0200
Subject: [PATCH] sqldb: add UpsertSelfNode query

This query is less strict in terms of the latest update timestamp field.
We want to be less strict with our own node data since we always want
our own updates recorded.
---
 sqldb/sqlc/graph.sql.go      | 45 ++++++++++++++++++++++++++++++++++++
 sqldb/sqlc/querier.go        |  4 ++++
 sqldb/sqlc/queries/graph.sql | 21 +++++++++++++++++
 3 files changed, 70 insertions(+)

diff --git a/sqldb/sqlc/graph.sql.go b/sqldb/sqlc/graph.sql.go
index 9c2702737..8ed9333d7 100644
--- a/sqldb/sqlc/graph.sql.go
+++ b/sqldb/sqlc/graph.sql.go
@@ -3735,6 +3735,51 @@ func (q *Queries) UpsertPruneLogEntry(ctx context.Context, arg UpsertPruneLogEnt
 	return err
 }
 
+const upsertSourceNode = `-- name: UpsertSourceNode :one
+INSERT INTO graph_nodes (
+    version, pub_key, alias, last_update, color, signature
+) VALUES (
+    $1, $2, $3, $4, $5, $6
+)
+ON CONFLICT (pub_key, version)
+    -- Update the following fields if a conflict occurs on pub_key
+    -- and version.
+    DO UPDATE SET
+        alias = EXCLUDED.alias,
+        last_update = EXCLUDED.last_update,
+        color = EXCLUDED.color,
+        signature = EXCLUDED.signature
+WHERE graph_nodes.last_update IS NULL
+    OR EXCLUDED.last_update >= graph_nodes.last_update
+RETURNING id
+`
+
+type UpsertSourceNodeParams struct {
+	Version    int16
+	PubKey     []byte
+	Alias      sql.NullString
+	LastUpdate sql.NullInt64
+	Color      sql.NullString
+	Signature  []byte
+}
+
+// We use a separate upsert for our own node since we want to be less strict
+// about the last_update field. For our own node, we always want to
+// update the record even if the last_update is the same as what we have.
+func (q *Queries) UpsertSourceNode(ctx context.Context, arg UpsertSourceNodeParams) (int64, error) {
+	row := q.db.QueryRowContext(ctx, upsertSourceNode,
+		arg.Version,
+		arg.PubKey,
+		arg.Alias,
+		arg.LastUpdate,
+		arg.Color,
+		arg.Signature,
+	)
+	var id int64
+	err := row.Scan(&id)
+	return id, err
+}
+
 const upsertZombieChannel = `-- name: UpsertZombieChannel :exec
 /* ─────────────────────────────────────────────
    graph_zombie_channels table queries
diff --git a/sqldb/sqlc/querier.go b/sqldb/sqlc/querier.go
index 0087559be..7b7b06495 100644
--- a/sqldb/sqlc/querier.go
+++ b/sqldb/sqlc/querier.go
@@ -147,6 +147,10 @@ type Querier interface {
 	UpsertNodeAddress(ctx context.Context, arg UpsertNodeAddressParams) error
 	UpsertNodeExtraType(ctx context.Context, arg UpsertNodeExtraTypeParams) error
 	UpsertPruneLogEntry(ctx context.Context, arg UpsertPruneLogEntryParams) error
+	// We use a separate upsert for our own node since we want to be less strict
+	// about the last_update field. For our own node, we always want to
+	// update the record even if the last_update is the same as what we have.
+	UpsertSourceNode(ctx context.Context, arg UpsertSourceNodeParams) (int64, error)
 	UpsertZombieChannel(ctx context.Context, arg UpsertZombieChannelParams) error
 }
 
diff --git a/sqldb/sqlc/queries/graph.sql b/sqldb/sqlc/queries/graph.sql
index 19087fc1b..b9bee1822 100644
--- a/sqldb/sqlc/queries/graph.sql
+++ b/sqldb/sqlc/queries/graph.sql
@@ -21,6 +21,27 @@ WHERE graph_nodes.last_update IS NULL
     OR EXCLUDED.last_update > graph_nodes.last_update
 RETURNING id;
 
+-- We use a separate upsert for our own node since we want to be less strict
+-- about the last_update field. For our own node, we always want to
+-- update the record even if the last_update is the same as what we have.
+-- name: UpsertSourceNode :one
+INSERT INTO graph_nodes (
+    version, pub_key, alias, last_update, color, signature
+) VALUES (
+    $1, $2, $3, $4, $5, $6
+)
+ON CONFLICT (pub_key, version)
+    -- Update the following fields if a conflict occurs on pub_key
+    -- and version.
+    DO UPDATE SET
+        alias = EXCLUDED.alias,
+        last_update = EXCLUDED.last_update,
+        color = EXCLUDED.color,
+        signature = EXCLUDED.signature
+WHERE graph_nodes.last_update IS NULL
+    OR EXCLUDED.last_update >= graph_nodes.last_update
+RETURNING id;
+
 -- name: GetNodesByIDs :many
 SELECT *
 FROM graph_nodes