Skip to content

Commit cc502a2

Browse files
committed
f Account for scoring params given seperately
1 parent 81fe5ac commit cc502a2

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
@@ -24,7 +24,9 @@ use lightning::ln::channelmanager::{self, ChainParameters, ChannelManagerReadArg
2424
use lightning::ln::msgs::RoutingMessageHandler;
2525
use lightning::ln::peer_handler::{IgnoringMessageHandler, MessageHandler};
2626
use lightning::routing::router::DefaultRouter;
27-
use lightning::routing::scoring::{ProbabilisticScorer, ProbabilisticScoringParameters};
27+
use lightning::routing::scoring::{
28+
ProbabilisticScorer, ProbabilisticScoringDecayParameters, ProbabilisticScoringFeeParameters,
29+
};
2830

2931
use lightning::util::config::UserConfig;
3032
use lightning::util::ser::ReadableArgs;
@@ -494,7 +496,7 @@ fn build_with_store_internal<K: KVStore + Sync + Send + 'static>(
494496
Ok(scorer) => Arc::new(Mutex::new(scorer)),
495497
Err(e) => {
496498
if e.kind() == std::io::ErrorKind::NotFound {
497-
let params = ProbabilisticScoringParameters::default();
499+
let params = ProbabilisticScoringDecayParameters::default();
498500
Arc::new(Mutex::new(ProbabilisticScorer::new(
499501
params,
500502
Arc::clone(&network_graph),
@@ -506,11 +508,13 @@ fn build_with_store_internal<K: KVStore + Sync + Send + 'static>(
506508
}
507509
};
508510

511+
let scoring_fee_params = ProbabilisticScoringFeeParameters::default();
509512
let router = Arc::new(DefaultRouter::new(
510513
Arc::clone(&network_graph),
511514
Arc::clone(&logger),
512515
keys_manager.get_secure_random_bytes(),
513516
Arc::clone(&scorer),
517+
scoring_fee_params,
514518
));
515519

516520
// 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)