Add |solana-validator monitor| subcommand (#15118)

validator/src/lib.rs

@@ -1,9 +1,21 @@
 pub use solana_core::test_validator;
 use {
     log::*,
-    std::{env, process::exit, thread::JoinHandle},
+    serde_derive::{Deserialize, Serialize},
+    std::{
+        env,
+        fs::{self, File},
+        io::{self, Write},
+        net::SocketAddr,
+        path::Path,
+        process::exit,
+        thread::JoinHandle,
+        time::{Duration, SystemTime},
+    },
 };
 
+pub mod dashboard;
+
 #[cfg(unix)]
 fn redirect_stderr(filename: &str) {
     use std::{fs::OpenOptions, os::unix::io::AsRawFd};
@@ -75,3 +87,49 @@ pub fn port_validator(port: String) -> Result<(), String> {
         .map(|_| ())
         .map_err(|e| format!("{:?}", e))
 }
+
+#[derive(Serialize, Deserialize, Clone, Debug)]
+struct ProcessInfo {
+    rpc_addr: Option<SocketAddr>, // RPC port to contact the validator at
+    start_time: u64,              // Seconds since the UNIX_EPOCH for when the validator was started
+}
+
+pub fn record_start(ledger_path: &Path, rpc_addr: Option<&SocketAddr>) -> Result<(), io::Error> {
+    if !ledger_path.exists() {
+        fs::create_dir_all(&ledger_path)?;
+    }
+
+    let start_info = ProcessInfo {
+        rpc_addr: rpc_addr.cloned(),
+        start_time: SystemTime::now()
+            .duration_since(SystemTime::UNIX_EPOCH)
+            .unwrap()
+            .as_secs(),
+    };
+
+    let serialized = serde_yaml::to_string(&start_info)
+        .map_err(|err| io::Error::new(io::ErrorKind::Other, format!("{:?}", err)))?;
+
+    let mut file = File::create(ledger_path.join("process-info.yml"))?;
+    file.write_all(&serialized.into_bytes())?;
+    Ok(())
+}
+
+fn get_validator_process_info(
+    ledger_path: &Path,
+) -> Result<(Option<SocketAddr>, SystemTime), io::Error> {
+    let file = File::open(ledger_path.join("process-info.yml"))?;
+    let config: ProcessInfo = serde_yaml::from_reader(file)
+        .map_err(|err| io::Error::new(io::ErrorKind::Other, format!("{:?}", err)))?;
+
+    let start_time = SystemTime::UNIX_EPOCH + Duration::from_secs(config.start_time);
+    Ok((config.rpc_addr, start_time))
+}
+
+pub fn get_validator_rpc_addr(ledger_path: &Path) -> Result<Option<SocketAddr>, io::Error> {
+    get_validator_process_info(ledger_path).map(|process_info| process_info.0)
+}
+
+pub fn get_validator_start_time(ledger_path: &Path) -> Result<SystemTime, io::Error> {
+    get_validator_process_info(ledger_path).map(|process_info| process_info.1)
+}