Просмотр исходного кода

general clean up for network upgrade

ghassmo 4 лет назад
Родитель
Сommit
8287e0195f
4 измененных файлов с 7 добавлено и 9 удалено
  1. 2 4
      bin/tau/tau-cli/src/cli.rs
  2. 1 2
      bin/tau/taud/src/jsonrpc.rs
  3. 0 2
      bin/tau/taud/src/settings.rs
  4. 4 1
      src/rpc/rpcserver.rs

+ 2 - 4
bin/tau/tau-cli/src/cli.rs

@@ -1,5 +1,3 @@
-use std::net::SocketAddr;
-
 use serde::{Deserialize, Serialize};
 use structopt::StructOpt;
 use structopt_toml::StructOptToml;
@@ -17,8 +15,8 @@ pub struct CliTau {
     #[structopt(short, parse(from_occurrences))]
     pub verbose: u8,
     /// JSON-RPC listen URL
-    #[structopt(long = "rpc", default_value = "127.0.0.1:11055")]
-    pub rpc_listen: SocketAddr,
+    #[structopt(long = "rpc", default_value = "tcp://127.0.0.1:11055")]
+    pub rpc_listen: String,
     /// Sets a custom config file
     #[structopt(short, long)]
     pub config: Option<String>,

+ 1 - 2
bin/tau/taud/src/jsonrpc.rs

@@ -1,6 +1,5 @@
-use std::{path::PathBuf, sync::Arc};
+use std::path::PathBuf;
 
-use async_executor::Executor;
 use async_trait::async_trait;
 use log::debug;
 use serde::{Deserialize, Serialize};

+ 0 - 2
bin/tau/taud/src/settings.rs

@@ -1,5 +1,3 @@
-use std::net::SocketAddr;
-
 use serde::Deserialize;
 use structopt::StructOpt;
 use structopt_toml::StructOptToml;

+ 4 - 1
src/rpc/rpcserver.rs

@@ -25,7 +25,6 @@ async fn run_accept_loop(
     listener: Box<dyn TransportListener>,
     rh: Arc<impl RequestHandler + 'static>,
 ) -> Result<()> {
-    // TODO can we spawn new task here ?
     while let Ok((stream, peer_addr)) = listener.next().await {
         info!(target: "JSON-RPC SERVER", "RPC Accepted connection {}", peer_addr);
         accept(stream, rh.clone()).await?;
@@ -107,9 +106,11 @@ pub async fn listen_and_serve(
 
             match upgrade {
                 None => {
+                    info!("RPC TCP listening to: {}", accept_url);
                     run_accept_loop(Box::new(listener), rh).await?;
                 }
                 Some(u) if u == "tls" => {
+                    info!("RPC TCP+TLS listening to: {}", accept_url);
                     let tls_listener = transport.upgrade_listener(listener)?.await?;
                     run_accept_loop(Box::new(tls_listener), rh).await?;
                 }
@@ -166,8 +167,10 @@ pub async fn listen_and_serve(
             match upgrade {
                 None => {
                     run_accept_loop(Box::new(listener), rh).await?;
+                    info!("RPC TOR listening to: {}", accept_url);
                 }
                 Some(u) if u == "tls" => {
+                    info!("RPC TOR+TLS listening to: {}", accept_url);
                     let tls_listener = transport.upgrade_listener(listener)?.await?;
                     run_accept_loop(Box::new(tls_listener), rh).await?;
                 }