Skip to content

Commit e847937

Browse files
committed
f Account for scoring params given seperately
1 parent 649b65c commit e847937

File tree

3 files changed

+16
-7
lines changed

3 files changed

+16
-7
lines changed

src/builder.rs

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,9 @@ use lightning::ln::channelmanager::{self, ChainParameters, ChannelManagerReadArg
2323
use lightning::ln::msgs::RoutingMessageHandler;
2424
use lightning::ln::peer_handler::{IgnoringMessageHandler, MessageHandler};
2525
use lightning::routing::router::DefaultRouter;
26-
use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringParameters};
26+
use lightning::routing::scoring::{
27+
ProbabilisticScorer, ProbabilisticScoringDecayParameters, ProbabilisticScoringFeeParameters,
28+
};
2729
use lightning::sign::EntropySource;
2830

2931
use lightning::util::config::UserConfig;
@@ -505,7 +507,7 @@ fn build_with_store_internal<K: KVStore + Sync + Send + 'static>(
505507
Ok(scorer) => Arc::new(Mutex::new(scorer)),
506508
Err(e) => {
507509
if e.kind() == std::io::ErrorKind::NotFound {
508-
let params = ProbabilisticScoringParameters::default();
510+
let params = ProbabilisticScoringDecayParameters::default();
509511
Arc::new(Mutex::new(ProbabilisticScorer::new(
510512
params,
511513
Arc::clone(&network_graph),
@@ -517,11 +519,13 @@ fn build_with_store_internal<K: KVStore + Sync + Send + 'static>(
517519
}
518520
};
519521

522+
let scoring_fee_params = ProbabilisticScoringFeeParameters::default();
520523
let router = Arc::new(DefaultRouter::new(
521524
Arc::clone(&network_graph),
522525
Arc::clone(&logger),
523526
keys_manager.get_secure_random_bytes(),
524527
Arc::clone(&scorer),
528+
scoring_fee_params,
525529
));
526530

527531
// Read ChannelMonitor state from store

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};
@@ -161,7 +161,7 @@ pub(crate) fn read_scorer<
161161
where
162162
L::Target: Logger,
163163
{
164-
let params = ProbabilisticScoringParameters::default();
164+
let params = ProbabilisticScoringDecayParameters::default();
165165
let mut reader = kv_store.read(SCORER_PERSISTENCE_NAMESPACE, SCORER_PERSISTENCE_KEY)?;
166166
let args = (params, network_graph, logger.clone());
167167
ProbabilisticScorer::read(&mut reader, args).map_err(|e| {

src/types.rs

Lines changed: 8 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ use lightning::ln::msgs::RoutingMessageHandler;
88
use lightning::ln::peer_handler::IgnoringMessageHandler;
99
use lightning::routing::gossip;
1010
use lightning::routing::router::DefaultRouter;
11-
use lightning::routing::scoring::ProbabilisticScorer;
11+
use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringFeeParameters};
1212
use lightning::sign::InMemorySigner;
1313
use lightning::util::ser::{Hostname, Readable, Writeable, Writer};
1414
use lightning_net_tokio::SocketDescriptor;
@@ -56,8 +56,13 @@ pub(crate) type ChannelManager<K> = lightning::ln::channelmanager::ChannelManage
5656
pub(crate) type KeysManager =
5757
WalletKeysManager<bdk::database::SqliteDatabase, Arc<FilesystemLogger>>;
5858

59-
pub(crate) type Router =
60-
DefaultRouter<Arc<NetworkGraph>, Arc<FilesystemLogger>, Arc<Mutex<Scorer>>>;
59+
pub(crate) type Router = DefaultRouter<
60+
Arc<NetworkGraph>,
61+
Arc<FilesystemLogger>,
62+
Arc<Mutex<Scorer>>,
63+
ProbabilisticScoringFeeParameters,
64+
Scorer,
65+
>;
6166
pub(crate) type Scorer = ProbabilisticScorer<Arc<NetworkGraph>, Arc<FilesystemLogger>>;
6267

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

0 commit comments

Comments
 (0)