mirror of
https://github.com/lightninglabs/lightning-terminal.git
synced 2026-08-13 12:33:36 +02:00
261c547622
In this commit, we add a `DB` struct in the `firewalldb` package. This struct will be responsible for housing abstract implementations of the various stores in the `firewalldb`. For now, we start with just the RulesDB. We also add Start&Stop methods for the struct in preparation for future additions here - for now, these do nothing. In the main LiT setup, we move the firewall.BoltDB and the new firewalldb.DB to the `stores` struct and implement them in the two `config_` files. For now, both varients create the Bbolt version of the firewallDB and this is used to init the `firewalldb.DB` struct. This will be changed in future commits where we will add a sql implementation.
86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
//go:build !dev
|
|
|
|
package terminal
|
|
|
|
import (
|
|
"fmt"
|
|
"path/filepath"
|
|
|
|
"github.com/lightninglabs/lightning-terminal/accounts"
|
|
"github.com/lightninglabs/lightning-terminal/firewalldb"
|
|
"github.com/lightninglabs/lightning-terminal/session"
|
|
"github.com/lightningnetwork/lnd/clock"
|
|
)
|
|
|
|
// DevConfig is an empty shell struct that allows us to build without the dev
|
|
// tag. This struct is embedded in the main Config struct, and it adds no new
|
|
// functionality in a production build.
|
|
type DevConfig struct{}
|
|
|
|
// defaultDevConfig returns an empty DevConfig struct.
|
|
func defaultDevConfig() *DevConfig {
|
|
return &DevConfig{}
|
|
}
|
|
|
|
// Validate is a no-op function during a production build.
|
|
func (c *DevConfig) Validate(_, _ string) error {
|
|
return nil
|
|
}
|
|
|
|
// NewStores creates a new instance of the stores struct using the default Bolt
|
|
// backend since in production, this is currently the only backend supported.
|
|
func NewStores(cfg *Config, clock clock.Clock) (*stores, error) {
|
|
networkDir := filepath.Join(cfg.LitDir, cfg.Network)
|
|
|
|
acctStore, err := accounts.NewBoltStore(
|
|
filepath.Dir(cfg.MacaroonPath), accounts.DBFilename, clock,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
sessStore, err := session.NewDB(
|
|
networkDir, session.DBFilename, clock, acctStore,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error creating session BoltStore: %v",
|
|
err)
|
|
}
|
|
|
|
firewallDB, err := firewalldb.NewBoltDB(
|
|
networkDir, firewalldb.DBFilename, sessStore,
|
|
)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error creating firewall DB: %v", err)
|
|
}
|
|
|
|
return &stores{
|
|
accounts: acctStore,
|
|
sessions: sessStore,
|
|
firewallBolt: firewallDB,
|
|
firewall: firewalldb.NewDB(firewallDB),
|
|
close: func() error {
|
|
var returnErr error
|
|
if err := acctStore.Close(); err != nil {
|
|
returnErr = fmt.Errorf("error closing "+
|
|
"account store: %v", err)
|
|
|
|
log.Error(returnErr.Error())
|
|
}
|
|
if err := sessStore.Close(); err != nil {
|
|
returnErr = fmt.Errorf("error closing "+
|
|
"session store: %v", err)
|
|
|
|
log.Error(returnErr.Error())
|
|
}
|
|
if err := firewallDB.Close(); err != nil {
|
|
returnErr = fmt.Errorf("error closing "+
|
|
"firewall DB: %v", err)
|
|
|
|
log.Error(returnErr.Error())
|
|
}
|
|
|
|
return returnErr
|
|
},
|
|
}, nil
|
|
}
|