quick fix to be able to set the caches path via env variables like we do for the...
authorMarc Tyndel <mtyndel@squareup.com>
Mon, 5 Jun 2023 17:54:51 +0000 (13:54 -0400)
committerMarc Tyndel <mtyndel@squareup.com>
Mon, 5 Jun 2023 19:02:28 +0000 (15:02 -0400)
src/config.rs
src/snapshot.rs

index 1f968d6418760b880cd1db6862ecf892b9446dc7..bf5a171e4502f015044fef70a75af6c64b94d20c 100644 (file)
@@ -30,8 +30,13 @@ pub(crate) fn network() -> Network {
        }
 }
 
-pub(crate) fn network_graph_cache_path() -> &'static str {
-       "./res/network_graph.bin"
+pub(crate) fn network_graph_cache_path() -> String {
+       format!("{}/network_graph.bin", cache_path())
+}
+
+pub(crate) fn cache_path() -> String {
+       let path = env::var("RAPID_GOSSIP_SYNC_SERVER_CACHES_PATH").unwrap_or("./res".to_string()).to_lowercase();
+       path
 }
 
 pub(crate) fn db_connection_config() -> Config {
index 86115ea667d287003dc4cb2362f510857ba52d6d..bbe94a93618037a227dac3cad4aab0c815658b78 100644 (file)
@@ -7,6 +7,7 @@ use std::time::{Duration, SystemTime, UNIX_EPOCH};
 use lightning::routing::gossip::NetworkGraph;
 
 use crate::{config, TestLogger};
+use crate::config::cache_path;
 
 pub(crate) struct Snapshotter {
        network_graph: Arc<NetworkGraph<TestLogger>>,
@@ -23,10 +24,10 @@ impl Snapshotter {
                let snapshot_sync_day_factors = [1, 2, 3, 4, 5, 6, 7, 14, 21, u64::MAX];
                let round_day_seconds = config::SNAPSHOT_CALCULATION_INTERVAL as u64;
 
-               let pending_snapshot_directory = "./res/snapshots_pending";
-               let pending_symlink_directory = "./res/symlinks_pending";
-               let finalized_snapshot_directory = "./res/snapshots";
-               let finalized_symlink_directory = "./res/symlinks";
+               let pending_snapshot_directory = format!("{}/snapshots_pending", cache_path());
+               let pending_symlink_directory = format!("{}/symlinks_pending", cache_path());
+               let finalized_snapshot_directory = format!("{}/snapshots", cache_path());
+               let finalized_symlink_directory = format!("{}/symlinks", cache_path());
                let relative_symlink_to_snapshot_path = "../snapshots";
 
                // this is gonna be a never-ending background job