projects
/
rapid-gossip-sync-server
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
|
inline
| side by side (parent:
32ac36a
)
Configure network
author
Andrei
<andrei.i@posteo.de>
Wed, 8 Feb 2023 00:00:00 +0000
(
00:00
+0000)
committer
Andrei
<andrei.i@posteo.de>
Wed, 8 Feb 2023 00:00:00 +0000
(
00:00
+0000)
src/config.rs
patch
|
blob
|
history
src/lib.rs
patch
|
blob
|
history
diff --git
a/src/config.rs
b/src/config.rs
index 444d54cfd10d2a18c5cf4faa65be4a0bac3ce50c..475d63a316e65adecbe274d4a574730e0b442ff6 100644
(file)
--- a/
src/config.rs
+++ b/
src/config.rs
@@
-5,6
+5,7
@@
use std::env;
use std::io::Cursor;
use std::net::{SocketAddr, ToSocketAddrs};
use std::io::Cursor;
use std::net::{SocketAddr, ToSocketAddrs};
+use bitcoin::Network;
use bitcoin::hashes::hex::FromHex;
use bitcoin::secp256k1::PublicKey;
use futures::stream::{FuturesUnordered, StreamExt};
use bitcoin::hashes::hex::FromHex;
use bitcoin::secp256k1::PublicKey;
use futures::stream::{FuturesUnordered, StreamExt};
@@
-17,6
+18,17
@@
pub(crate) const SCHEMA_VERSION: i32 = 8;
pub(crate) const SNAPSHOT_CALCULATION_INTERVAL: u32 = 3600 * 24; // every 24 hours, in seconds
pub(crate) const DOWNLOAD_NEW_GOSSIP: bool = true;
pub(crate) const SNAPSHOT_CALCULATION_INTERVAL: u32 = 3600 * 24; // every 24 hours, in seconds
pub(crate) const DOWNLOAD_NEW_GOSSIP: bool = true;
+pub(crate) fn network() -> Network {
+ let network = env::var("RAPID_GOSSIP_SYNC_SERVER_NETWORK").unwrap_or("Bitcoin".to_string());
+ match network.as_str() {
+ "Bitcoin" => Network::Bitcoin,
+ "Testnet" => Network::Testnet,
+ "Signet" => Network::Signet,
+ "Regtest" => Network::Regtest,
+ _ => panic!("Invalid network"),
+ }
+}
+
pub(crate) fn network_graph_cache_path() -> &'static str {
"./res/network_graph.bin"
}
pub(crate) fn network_graph_cache_path() -> &'static str {
"./res/network_graph.bin"
}
diff --git
a/src/lib.rs
b/src/lib.rs
index cc5a350847e2c4c42322d2bf20545d5de900059a..25ce1a2f5015c2a23dd76479fd12f831a854903d 100644
(file)
--- a/
src/lib.rs
+++ b/
src/lib.rs
@@
-16,7
+16,6
@@
use std::io::BufReader;
use std::sync::Arc;
use bitcoin::blockdata::constants::genesis_block;
use std::sync::Arc;
use bitcoin::blockdata::constants::genesis_block;
-use bitcoin::Network;
use bitcoin::secp256k1::PublicKey;
use lightning::routing::gossip::NetworkGraph;
use lightning::util::ser::{ReadableArgs, Writeable};
use bitcoin::secp256k1::PublicKey;
use lightning::routing::gossip::NetworkGraph;
use lightning::util::ser::{ReadableArgs, Writeable};
@@
-54,6
+53,7
@@
pub struct SerializedResponse {
impl RapidSyncProcessor {
pub fn new() -> Self {
impl RapidSyncProcessor {
pub fn new() -> Self {
+ let network = config::network();
let logger = TestLogger::new();
let network_graph = if let Ok(file) = File::open(&config::network_graph_cache_path()) {
println!("Initializing from cached network graph…");
let logger = TestLogger::new();
let network_graph = if let Ok(file) = File::open(&config::network_graph_cache_path()) {
println!("Initializing from cached network graph…");
@@
-65,10
+65,10
@@
impl RapidSyncProcessor {
network_graph
} else {
println!("Initialization from cached network graph failed: {}", network_graph_result.err().unwrap());
network_graph
} else {
println!("Initialization from cached network graph failed: {}", network_graph_result.err().unwrap());
- NetworkGraph::new(genesis_block(
Network::Bitcoin
).header.block_hash(), logger)
+ NetworkGraph::new(genesis_block(
network
).header.block_hash(), logger)
}
} else {
}
} else {
- NetworkGraph::new(genesis_block(
Network::Bitcoin
).header.block_hash(), logger)
+ NetworkGraph::new(genesis_block(
network
).header.block_hash(), logger)
};
let arc_network_graph = Arc::new(network_graph);
Self {
};
let arc_network_graph = Arc::new(network_graph);
Self {