Skip to content

Commit 2b7ecf9

Browse files
committed
f Account for scoring params given seperately
1 parent 39d0dd4 commit 2b7ecf9

File tree

3 files changed

+16
-7
lines changed

3 files changed

+16
-7
lines changed

src/io/utils.rs

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ use crate::{Error, EventQueue, PaymentDetails};
77

88
use lightning::chain::channelmonitor::ChannelMonitor;
99
use lightning::routing::gossip::NetworkGraph;
10-
use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringParameters};
10+
use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringDecayParameters};
1111
use lightning::sign::{EntropySource, SignerProvider};
1212
use lightning::util::logger::Logger;
1313
use lightning::util::ser::{Readable, ReadableArgs, Writeable};
@@ -115,7 +115,7 @@ where
115115
K::Target: KVStore,
116116
L::Target: Logger,
117117
{
118-
let params = ProbabilisticScoringParameters::default();
118+
let params = ProbabilisticScoringDecayParameters::default();
119119
let mut reader = kv_store.read(SCORER_PERSISTENCE_NAMESPACE, SCORER_PERSISTENCE_KEY)?;
120120
let args = (params, network_graph, logger);
121121
let scorer = ProbabilisticScorer::read(&mut reader, args).map_err(|_| {

src/lib.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -121,7 +121,9 @@ use lightning::ln::channelmanager::{
121121
};
122122
use lightning::ln::peer_handler::{IgnoringMessageHandler, MessageHandler};
123123
use lightning::ln::{PaymentHash, PaymentPreimage, PaymentSecret};
124-
use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringParameters};
124+
use lightning::routing::scoring::{
125+
ProbabilisticScorer, ProbabilisticScoringDecayParameters, ProbabilisticScoringFeeParameters,
126+
};
125127
use lightning::sign::EntropySource;
126128

127129
use lightning::util::config::{ChannelHandshakeConfig, ChannelHandshakeLimits, UserConfig};
@@ -434,7 +436,7 @@ impl Builder {
434436
Ok(scorer) => Arc::new(Mutex::new(scorer)),
435437
Err(e) => {
436438
if e.kind() == std::io::ErrorKind::NotFound {
437-
let params = ProbabilisticScoringParameters::default();
439+
let params = ProbabilisticScoringDecayParameters::default();
438440
Arc::new(Mutex::new(ProbabilisticScorer::new(
439441
params,
440442
Arc::clone(&network_graph),
@@ -447,11 +449,13 @@ impl Builder {
447449
}
448450
};
449451

452+
let scoring_fee_params = ProbabilisticScoringFeeParameters::default();
450453
let router = Arc::new(DefaultRouter::new(
451454
Arc::clone(&network_graph),
452455
Arc::clone(&logger),
453456
keys_manager.get_secure_random_bytes(),
454457
Arc::clone(&scorer),
458+
scoring_fee_params,
455459
));
456460

457461
// Read ChannelMonitor state from store

src/types.rs

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ use lightning::ln::peer_handler::IgnoringMessageHandler;
1313
use lightning::ln::{PaymentHash, PaymentPreimage, PaymentSecret};
1414
use lightning::routing::gossip;
1515
use lightning::routing::router::DefaultRouter;
16-
use lightning::routing::scoring::ProbabilisticScorer;
16+
use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringFeeParameters};
1717
use lightning::sign::InMemorySigner;
1818
use lightning::util::ser::{Hostname, Readable, Writeable, Writer};
1919
use lightning_invoice::{Invoice, SignedRawInvoice};
@@ -64,8 +64,13 @@ pub(crate) type ChannelManager = lightning::ln::channelmanager::ChannelManager<
6464

6565
pub(crate) type KeysManager = WalletKeysManager<bdk::database::SqliteDatabase>;
6666

67-
pub(crate) type Router =
68-
DefaultRouter<Arc<NetworkGraph>, Arc<FilesystemLogger>, Arc<Mutex<Scorer>>>;
67+
pub(crate) type Router = DefaultRouter<
68+
Arc<NetworkGraph>,
69+
Arc<FilesystemLogger>,
70+
Arc<Mutex<Scorer>>,
71+
ProbabilisticScoringFeeParameters,
72+
Scorer,
73+
>;
6974
pub(crate) type Scorer = ProbabilisticScorer<Arc<NetworkGraph>, Arc<FilesystemLogger>>;
7075

7176
pub(crate) type NetworkGraph = gossip::NetworkGraph<Arc<FilesystemLogger>>;

0 commit comments

Comments
 (0)