mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-12-07 03:13:18 +01:00
In this commit, we implement the postgres and sqlite versions of the NewTestDB function. We add the various build flags so that only one of the three versions of this function can be active at a time. We also introduce the SQLStore struct which is the SQL implementation of the V1Store interface. NOTE: it currently temporarily embeds the KVStore struct so that we can implement the V1Store interface incrementally. For any method not implemented, things will fall back to the KVStore. This is ONLY the case for the time being while this struct is purely used in unit tests only. Once all the methods have been implemented, the KVStore field will be removed from the SQLStore struct.
46 lines
1.3 KiB
Go
46 lines
1.3 KiB
Go
package graphdb
|
|
|
|
import (
|
|
"github.com/lightningnetwork/lnd/sqldb"
|
|
)
|
|
|
|
// SQLQueries is a subset of the sqlc.Querier interface that can be used to
|
|
// execute queries against the SQL graph tables.
|
|
type SQLQueries interface {
|
|
}
|
|
|
|
// BatchedSQLQueries is a version of SQLQueries that's capable of batched
|
|
// database operations.
|
|
type BatchedSQLQueries interface {
|
|
SQLQueries
|
|
sqldb.BatchedTx[SQLQueries]
|
|
}
|
|
|
|
// SQLStore is an implementation of the V1Store interface that uses a SQL
|
|
// database as the backend.
|
|
//
|
|
// NOTE: currently, this temporarily embeds the KVStore struct so that we can
|
|
// implement the V1Store interface incrementally. For any method not
|
|
// implemented, things will fall back to the KVStore. This is ONLY the case
|
|
// for the time being while this struct is purely used in unit tests only.
|
|
type SQLStore struct {
|
|
db BatchedSQLQueries
|
|
|
|
// Temporary fall-back to the KVStore so that we can implement the
|
|
// interface incrementally.
|
|
*KVStore
|
|
}
|
|
|
|
// A compile-time assertion to ensure that SQLStore implements the V1Store
|
|
// interface.
|
|
var _ V1Store = (*SQLStore)(nil)
|
|
|
|
// NewSQLStore creates a new SQLStore instance given an open BatchedSQLQueries
|
|
// storage backend.
|
|
func NewSQLStore(db BatchedSQLQueries, kvStore *KVStore) *SQLStore {
|
|
return &SQLStore{
|
|
db: db,
|
|
KVStore: kvStore,
|
|
}
|
|
}
|