@@ -38,7 +38,7 @@ use lightning::ln::peer_handler::{MessageHandler,PeerManager,SocketDescriptor,Ig
38
38
use lightning:: ln:: msgs:: DecodeError ;
39
39
use lightning:: ln:: script:: ShutdownScript ;
40
40
use lightning:: routing:: router:: get_route;
41
- use lightning:: routing:: network_graph:: NetGraphMsgHandler ;
41
+ use lightning:: routing:: network_graph:: { NetGraphMsgHandler , NetworkGraph } ;
42
42
use lightning:: util:: config:: UserConfig ;
43
43
use lightning:: util:: errors:: APIError ;
44
44
use lightning:: util:: events:: Event ;
@@ -378,7 +378,8 @@ pub fn do_test(data: &[u8], logger: &Arc<dyn Logger>) {
378
378
} ;
379
379
let channelmanager = Arc :: new ( ChannelManager :: new ( fee_est. clone ( ) , monitor. clone ( ) , broadcast. clone ( ) , Arc :: clone ( & logger) , keys_manager. clone ( ) , config, params) ) ;
380
380
let our_id = PublicKey :: from_secret_key ( & Secp256k1 :: signing_only ( ) , & keys_manager. get_node_secret ( ) ) ;
381
- let net_graph_msg_handler = Arc :: new ( NetGraphMsgHandler :: new ( genesis_block ( network) . block_hash ( ) , None , Arc :: clone ( & logger) ) ) ;
381
+ let network_graph = NetworkGraph :: new ( genesis_block ( network) . block_hash ( ) ) ;
382
+ let net_graph_msg_handler = Arc :: new ( NetGraphMsgHandler :: new ( network_graph, None , Arc :: clone ( & logger) ) ) ;
382
383
383
384
let peers = RefCell :: new ( [ false ; 256 ] ) ;
384
385
let mut loss_detector = MoneyLossDetector :: new ( & peers, channelmanager. clone ( ) , monitor. clone ( ) , PeerManager :: new ( MessageHandler {
@@ -434,7 +435,7 @@ pub fn do_test(data: &[u8], logger: &Arc<dyn Logger>) {
434
435
} ,
435
436
4 => {
436
437
let value = slice_to_be24 ( get_slice ! ( 3 ) ) as u64 ;
437
- let route = match get_route ( & our_id, & net_graph_msg_handler. network_graph . read ( ) . unwrap ( ) , & get_pubkey ! ( ) , None , None , & Vec :: new ( ) , value, 42 , Arc :: clone ( & logger) ) {
438
+ let route = match get_route ( & our_id, & net_graph_msg_handler. network_graph , & get_pubkey ! ( ) , None , None , & Vec :: new ( ) , value, 42 , Arc :: clone ( & logger) ) {
438
439
Ok ( route) => route,
439
440
Err ( _) => return ,
440
441
} ;
@@ -451,7 +452,7 @@ pub fn do_test(data: &[u8], logger: &Arc<dyn Logger>) {
451
452
} ,
452
453
15 => {
453
454
let value = slice_to_be24 ( get_slice ! ( 3 ) ) as u64 ;
454
- let mut route = match get_route ( & our_id, & net_graph_msg_handler. network_graph . read ( ) . unwrap ( ) , & get_pubkey ! ( ) , None , None , & Vec :: new ( ) , value, 42 , Arc :: clone ( & logger) ) {
455
+ let mut route = match get_route ( & our_id, & net_graph_msg_handler. network_graph , & get_pubkey ! ( ) , None , None , & Vec :: new ( ) , value, 42 , Arc :: clone ( & logger) ) {
455
456
Ok ( route) => route,
456
457
Err ( _) => return ,
457
458
} ;
0 commit comments