Merge pull request #10858 from yyforyongyu/benchmark-fetch-inflight-payments

payments: Add `FetchInFlightPayments` Benchmarks

Author: ziggie1984

payments/db/fetch_inflight_benchmark_postgres_test.go

@@ -0,0 +1,60 @@
+//go:build test_db_postgres && !test_db_sqlite
+
+package paymentsdb
+
+import (
+	"database/sql"
+	"testing"
+
+	"github.com/lightningnetwork/lnd/sqldb"
+)
+
+// postgresFetchInFlightBenchBackend creates Postgres-backed benchmark stores.
+type postgresFetchInFlightBenchBackend struct {
+	fixture *sqldb.TestPgFixture
+}
+
+// newFetchInFlightBenchBackend creates the benchmark backend selected by the
+// active build tags.
+func newFetchInFlightBenchBackend(b *testing.B) fetchInFlightBenchBackend {
+	b.Helper()
+
+	fixture := sqldb.NewTestPgFixture(
+		b, sqldb.DefaultPostgresFixtureLifetime,
+	)
+	b.Cleanup(func() {
+		fixture.TearDown(b)
+	})
+
+	return postgresFetchInFlightBenchBackend{fixture: fixture}
+}
+
+// newStore creates a migrated Postgres SQLStore and seeds it with one benchmark
+// scenario.
+func (backend postgresFetchInFlightBenchBackend) newStore(b *testing.B,
+	scenario fetchInFlightBenchScenario,
+	totalPayments int) (*SQLStore, fetchInFlightBenchSeedStats) {
+
+	b.Helper()
+
+	pgDB := sqldb.NewTestPostgresDB(b, backend.fixture)
+	baseDB := pgDB.BaseDB
+
+	withTx := func(tx *sql.Tx) SQLQueries {
+		return baseDB.WithTx(tx)
+	}
+	queries := sqldb.NewTransactionExecutor(baseDB, withTx)
+
+	store, err := NewSQLStore(&SQLStoreConfig{
+		QueryCfg: sqldb.DefaultPostgresConfig(),
+	}, queries)
+	if err != nil {
+		b.Fatalf("new SQL store: %v", err)
+	}
+
+	stats := seedFetchInFlightBenchDB(
+		b, baseDB.DB, withTx, scenario, totalPayments,
+	)
+
+	return store, stats
+}

payments/db/fetch_inflight_benchmark_sqlite_test.go

@@ -0,0 +1,49 @@
+//go:build test_db_sqlite && !test_db_postgres
+
+package paymentsdb
+
+import (
+	"database/sql"
+	"testing"
+
+	"github.com/lightningnetwork/lnd/sqldb"
+)
+
+// sqliteFetchInFlightBenchBackend creates SQLite-backed benchmark stores.
+type sqliteFetchInFlightBenchBackend struct{}
+
+// newFetchInFlightBenchBackend creates the benchmark backend selected by the
+// active build tags.
+func newFetchInFlightBenchBackend(_ *testing.B) fetchInFlightBenchBackend {
+	return sqliteFetchInFlightBenchBackend{}
+}
+
+// newStore creates a migrated SQLite SQLStore and seeds it with one benchmark
+// scenario.
+func (sqliteFetchInFlightBenchBackend) newStore(b *testing.B,
+	scenario fetchInFlightBenchScenario,
+	totalPayments int) (*SQLStore, fetchInFlightBenchSeedStats) {
+
+	b.Helper()
+
+	sqliteDB := sqldb.NewTestSqliteDB(b)
+	baseDB := sqliteDB.BaseDB
+
+	withTx := func(tx *sql.Tx) SQLQueries {
+		return baseDB.WithTx(tx)
+	}
+	queries := sqldb.NewTransactionExecutor(baseDB, withTx)
+
+	store, err := NewSQLStore(&SQLStoreConfig{
+		QueryCfg: sqldb.DefaultSQLiteConfig(),
+	}, queries)
+	if err != nil {
+		b.Fatalf("new SQL store: %v", err)
+	}
+
+	stats := seedFetchInFlightBenchDB(
+		b, baseDB.DB, withTx, scenario, totalPayments,
+	)
+
+	return store, stats
+}