Skip to content

Commit a7b08c7

Browse files
committed
Account for DefaultRouter taking score params
1 parent 8d2369f commit a7b08c7

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

ci/LDKSwift/Sources/LDKSwift/batteries/ChannelManagerConstructor.swift

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -42,10 +42,11 @@ public struct ChannelManagerConstructionParameters {
4242
public var txBroadcaster: BroadcasterInterface
4343
public var enableP2PGossip: Bool = false
4444
public var scorer: MultiThreadedLockableScore?
45+
public var scoreParams: ProbabilisticScoringFeeParameters?
4546
public var payerRetries: Retry = Retry.initWithAttempts(a: UInt(3))
4647
public var logger: Logger
4748

48-
public init (config: UserConfig, entropySource: EntropySource, nodeSigner: NodeSigner, signerProvider: SignerProvider, feeEstimator: FeeEstimator, chainMonitor: ChainMonitor, txBroadcaster: BroadcasterInterface, logger: Logger, enableP2PGossip: Bool = false, scorer: MultiThreadedLockableScore? = nil, payerRetries: Retry = Retry.initWithAttempts(a: UInt(3))) {
49+
public init (config: UserConfig, entropySource: EntropySource, nodeSigner: NodeSigner, signerProvider: SignerProvider, feeEstimator: FeeEstimator, chainMonitor: ChainMonitor, txBroadcaster: BroadcasterInterface, logger: Logger, enableP2PGossip: Bool = false, scorer: MultiThreadedLockableScore? = nil, scoreParams: ProbabilisticScoringFeeParameters? = nil, payerRetries: Retry = Retry.initWithAttempts(a: UInt(3))) {
4950
self.config = config
5051
self.entropySource = entropySource
5152
self.nodeSigner = nodeSigner
@@ -57,6 +58,11 @@ public struct ChannelManagerConstructionParameters {
5758

5859
self.enableP2PGossip = enableP2PGossip
5960
self.scorer = scorer
61+
if scorer != nil && scoreParams == nil {
62+
self.scoreParams = ProbabilisticScoringFeeParameters.initWithDefault()
63+
} else {
64+
self.scoreParams = scoreParams
65+
}
6066
self.payerRetries = payerRetries
6167
}
6268

@@ -74,8 +80,8 @@ public struct ChannelManagerConstructionParameters {
7480
}
7581

7682
fileprivate func router(networkGraph: NetworkGraph?) -> Router {
77-
if let netGraph = networkGraph, let scorer = self.scorer {
78-
return DefaultRouter(networkGraph: netGraph, logger: self.logger, randomSeedBytes: self.entropySource.getSecureRandomBytes(), scorer: scorer.asLockableScore()).asRouter()
83+
if let netGraph = networkGraph, let scorer = self.scorer, let scoreParams = self.scoreParams {
84+
return DefaultRouter(networkGraph: netGraph, logger: self.logger, randomSeedBytes: self.entropySource.getSecureRandomBytes(), scorer: scorer.asLockableScore(), scoreParams: scoreParams).asRouter()
7985
}
8086
return CMCRouter()
8187
}

0 commit comments

Comments
 (0)