graph+config: add sql pagination config to ChannelGraph

This commit is contained in:
Elle Mouton
2025-07-16 08:33:02 +02:00
parent f0d2d1fd0a
commit 2fa30e8735
5 changed files with 17 additions and 6 deletions

View File

@@ -32,7 +32,8 @@ func (d *DefaultDatabaseBuilder) getGraphStore(baseDB *sqldb.BaseDB,
return graphdb.NewSQLStore(
&graphdb.SQLStoreConfig{
ChainHash: *d.cfg.ActiveNetParams.GenesisHash,
ChainHash: *d.cfg.ActiveNetParams.GenesisHash,
PaginationCfg: sqldb.DefaultPagedQueryConfig(),
},
graphExecutor, opts...,
)

View File

@@ -182,6 +182,9 @@ type SQLStoreConfig struct {
// ChainHash is the genesis hash for the chain that all the gossip
// messages in this store are aimed at.
ChainHash chainhash.Hash
// PaginationCfg is the configuration for paginated queries.
PaginationCfg *sqldb.PagedQueryConfig
}
// NewSQLStore creates a new SQLStore instance given an open BatchedSQLQueries

View File

@@ -19,7 +19,9 @@ func NewTestDB(t testing.TB) V1Store {
// NewTestDBFixture creates a new sqldb.TestPgFixture for testing purposes.
func NewTestDBFixture(t *testing.T) *sqldb.TestPgFixture {
pgFixture := sqldb.NewTestPgFixture(t, sqldb.DefaultPostgresFixtureLifetime)
pgFixture := sqldb.NewTestPgFixture(
t, sqldb.DefaultPostgresFixtureLifetime,
)
t.Cleanup(func() {
pgFixture.TearDown(t)
})
@@ -28,7 +30,9 @@ func NewTestDBFixture(t *testing.T) *sqldb.TestPgFixture {
// NewTestDBWithFixture is a helper function that creates a SQLStore backed by a
// SQL database for testing.
func NewTestDBWithFixture(t testing.TB, pgFixture *sqldb.TestPgFixture) V1Store {
func NewTestDBWithFixture(t testing.TB,
pgFixture *sqldb.TestPgFixture) V1Store {
var querier BatchedSQLQueries
if pgFixture == nil {
querier = newBatchQuerier(t)
@@ -38,7 +42,8 @@ func NewTestDBWithFixture(t testing.TB, pgFixture *sqldb.TestPgFixture) V1Store
store, err := NewSQLStore(
&SQLStoreConfig{
ChainHash: *chaincfg.MainNetParams.GenesisHash,
ChainHash: *chaincfg.MainNetParams.GenesisHash,
PaginationCfg: sqldb.DefaultPagedQueryConfig(),
}, querier,
)
require.NoError(t, err)

View File

@@ -27,7 +27,8 @@ func NewTestDBFixture(_ *testing.T) *sqldb.TestPgFixture {
func NewTestDBWithFixture(t testing.TB, _ *sqldb.TestPgFixture) V1Store {
store, err := NewSQLStore(
&SQLStoreConfig{
ChainHash: *chaincfg.MainNetParams.GenesisHash,
ChainHash: *chaincfg.MainNetParams.GenesisHash,
PaginationCfg: sqldb.DefaultPagedQueryConfig(),
}, newBatchQuerier(t),
)
require.NoError(t, err)

View File

@@ -144,7 +144,8 @@ func openNativeSQLGraphDB(ht *lntest.HarnessTest,
store, err := graphdb.NewSQLStore(
&graphdb.SQLStoreConfig{
ChainHash: *ht.Miner().ActiveNet.GenesisHash,
ChainHash: *ht.Miner().ActiveNet.GenesisHash,
PaginationCfg: sqldb.DefaultPagedQueryConfig(),
},
executor,
)