From 3b449820d25233627474b84aec86407fdf30b647 Mon Sep 17 00:00:00 2001
From: Elle Mouton <elle.mouton@gmail.com>
Date: Fri, 23 May 2025 11:03:45 +0200
Subject: [PATCH] graph/db: init SQLStore caches and batch schedulers

Here, we use the new options to initialise the reject and channel caches
for the SQLStore (as is done for the KVStore) and also the channel and
node batch schedulers.
---
 graph/db/sql_store.go | 31 +++++++++++++++++++++++++++----
 1 file changed, 27 insertions(+), 4 deletions(-)

diff --git a/graph/db/sql_store.go b/graph/db/sql_store.go
index 48a6509f9..3947ae0d8 100644
--- a/graph/db/sql_store.go
+++ b/graph/db/sql_store.go
@@ -2,7 +2,9 @@ package graphdb
 
 import (
 	"fmt"
+	"sync"
 
+	"github.com/lightningnetwork/lnd/batch"
 	"github.com/lightningnetwork/lnd/sqldb"
 )
 
@@ -28,6 +30,16 @@ type BatchedSQLQueries interface {
 type SQLStore struct {
 	db BatchedSQLQueries
 
+	// cacheMu guards all caches (rejectCache and chanCache). If
+	// this mutex will be acquired at the same time as the DB mutex then
+	// the cacheMu MUST be acquired first to prevent deadlock.
+	cacheMu     sync.RWMutex
+	rejectCache *rejectCache
+	chanCache   *channelCache
+
+	chanScheduler batch.Scheduler[SQLQueries]
+	nodeScheduler batch.Scheduler[SQLQueries]
+
 	// Temporary fall-back to the KVStore so that we can implement the
 	// interface incrementally.
 	*KVStore
@@ -52,8 +64,19 @@ func NewSQLStore(db BatchedSQLQueries, kvStore *KVStore,
 			"supported for SQL stores")
 	}
 
-	return &SQLStore{
-		db:      db,
-		KVStore: kvStore,
-	}, nil
+	s := &SQLStore{
+		db:          db,
+		KVStore:     kvStore,
+		rejectCache: newRejectCache(opts.RejectCacheSize),
+		chanCache:   newChannelCache(opts.ChannelCacheSize),
+	}
+
+	s.chanScheduler = batch.NewTimeScheduler(
+		db, &s.cacheMu, opts.BatchCommitInterval,
+	)
+	s.nodeScheduler = batch.NewTimeScheduler(
+		db, nil, opts.BatchCommitInterval,
+	)
+
+	return s, nil
 }