mirror of
https://github.com/lightningnetwork/lnd.git
synced 2025-02-23 14:40:30 +01:00
6ff6c86155
This commit moves `DetermineFeePerKw` into the `Estimate` method on `FeePreference`. A few callsites previously calling `DetermineFeePerKw` without the max fee rate is now also temporarily fixed by forcing them to use `Estimate` with the default sweeper max fee rate.
80 lines
1.7 KiB
Go
80 lines
1.7 KiB
Go
package sweep
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/lightningnetwork/lnd/lnwallet/chainfee"
|
|
)
|
|
|
|
// mockFeeEstimator implements a mock fee estimator. It closely resembles
|
|
// lnwallet.StaticFeeEstimator with the addition that fees can be changed for
|
|
// testing purposes in a thread safe manner.
|
|
//
|
|
// TODO(yy): replace it with chainfee.MockEstimator once it's merged.
|
|
type mockFeeEstimator struct {
|
|
feePerKW chainfee.SatPerKWeight
|
|
|
|
relayFee chainfee.SatPerKWeight
|
|
|
|
blocksToFee map[uint32]chainfee.SatPerKWeight
|
|
|
|
// A closure that when set is used instead of the
|
|
// mockFeeEstimator.EstimateFeePerKW method.
|
|
estimateFeePerKW func(numBlocks uint32) (chainfee.SatPerKWeight, error)
|
|
|
|
lock sync.Mutex
|
|
}
|
|
|
|
func newMockFeeEstimator(feePerKW,
|
|
relayFee chainfee.SatPerKWeight) *mockFeeEstimator {
|
|
|
|
return &mockFeeEstimator{
|
|
feePerKW: feePerKW,
|
|
relayFee: relayFee,
|
|
blocksToFee: make(map[uint32]chainfee.SatPerKWeight),
|
|
}
|
|
}
|
|
|
|
func (e *mockFeeEstimator) updateFees(feePerKW,
|
|
relayFee chainfee.SatPerKWeight) {
|
|
|
|
e.lock.Lock()
|
|
defer e.lock.Unlock()
|
|
|
|
e.feePerKW = feePerKW
|
|
e.relayFee = relayFee
|
|
}
|
|
|
|
func (e *mockFeeEstimator) EstimateFeePerKW(numBlocks uint32) (
|
|
chainfee.SatPerKWeight, error) {
|
|
|
|
e.lock.Lock()
|
|
defer e.lock.Unlock()
|
|
|
|
if e.estimateFeePerKW != nil {
|
|
return e.estimateFeePerKW(numBlocks)
|
|
}
|
|
|
|
if fee, ok := e.blocksToFee[numBlocks]; ok {
|
|
return fee, nil
|
|
}
|
|
|
|
return e.feePerKW, nil
|
|
}
|
|
|
|
func (e *mockFeeEstimator) RelayFeePerKW() chainfee.SatPerKWeight {
|
|
e.lock.Lock()
|
|
defer e.lock.Unlock()
|
|
|
|
return e.relayFee
|
|
}
|
|
|
|
func (e *mockFeeEstimator) Start() error {
|
|
return nil
|
|
}
|
|
|
|
func (e *mockFeeEstimator) Stop() error {
|
|
return nil
|
|
}
|
|
|
|
var _ chainfee.Estimator = (*mockFeeEstimator)(nil)
|