sweep+lnd: move ticker creation into sweeper

This commit is contained in:
yyforyongyu committed 2023-10-23 10:24:58 +08:00
1 parent 62b5869f87
commit 4ba09098d1
3 files changed
+8 -13

No files matched your search

+1 -3
View File
@@ -1064,9 +1064,7 @@ func newServer(cfg *Config, listenAddrs []net.Addr,
GenSweepScript: newSweepPkScriptGen(cc.Wallet), GenSweepScript: newSweepPkScriptGen(cc.Wallet),
Signer: cc.Wallet.Cfg.Signer, Signer: cc.Wallet.Cfg.Signer,
Wallet: newSweeperWallet(cc.Wallet), Wallet: newSweeperWallet(cc.Wallet),
NewBatchTimer: func() <-chan time.Time { TickerDuration: cfg.Sweeper.BatchWindowDuration,
return time.NewTimer(cfg.Sweeper.BatchWindowDuration).C
},
Notifier: cc.ChainNotifier, Notifier: cc.ChainNotifier,
Store: sweeperStore, Store: sweeperStore,
MaxInputsPerTx: sweep.DefaultMaxInputsPerTx, MaxInputsPerTx: sweep.DefaultMaxInputsPerTx,
+6 -5
View File
@@ -264,10 +264,11 @@ type UtxoSweeperConfig struct {
// Wallet contains the wallet functions that sweeper requires. // Wallet contains the wallet functions that sweeper requires.
Wallet Wallet Wallet Wallet
// NewBatchTimer creates a channel that will be sent on when a certain // TickerDuration is used to create a channel that will be sent on when
// time window has passed. During this time window, new inputs can still // a certain time window has passed. During this time window, new
// be added to the sweep tx that is about to be generated. // inputs can still be added to the sweep tx that is about to be
NewBatchTimer func() <-chan time.Time // generated.
TickerDuration time.Duration
// Notifier is an instance of a chain notifier we'll use to watch for // Notifier is an instance of a chain notifier we'll use to watch for
// certain on-chain events. // certain on-chain events.
@@ -1019,7 +1020,7 @@ func (s *UtxoSweeper) scheduleSweep(currentHeight int32) error {
// Start sweep timer to create opportunity for more inputs to be added // Start sweep timer to create opportunity for more inputs to be added
// before a tx is constructed. // before a tx is constructed.
s.timer = s.cfg.NewBatchTimer() s.timer = time.NewTicker(s.cfg.TickerDuration).C
log.Debugf("Sweep timer started") log.Debugf("Sweep timer started")
+1 -5
View File
@@ -129,11 +129,7 @@ func createSweeperTestContext(t *testing.T) *sweeperTestContext {
ctx.sweeper = New(&UtxoSweeperConfig{ ctx.sweeper = New(&UtxoSweeperConfig{
Notifier: notifier, Notifier: notifier,
Wallet: backend, Wallet: backend,
NewBatchTimer: func() <-chan time.Time { TickerDuration: 100 * time.Millisecond,
c := make(chan time.Time, 1)
ctx.timeoutChan <- c
return c
},
Store: store, Store: store,
Signer: &mock.DummySigner{}, Signer: &mock.DummySigner{},
GenSweepScript: func() ([]byte, error) { GenSweepScript: func() ([]byte, error) {