Update version to 0.1.2 in Cargo.toml

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "easy-proxy"
-version = "0.1.1"
+version = "0.1.2"
 edition = "2021"
 
 [dependencies]

README.md

@@ -65,6 +65,8 @@ pingora:
   # upgrade_sock: /tmp/pingora_upgrade.sock
   # user: nobody
   # group: webusers
+  # grace_period_seconds: 1
+  # graceful_shutdown_timeout_seconds: 1
   # ca_file: /etc/ssl/certs/ca-certificates.crt
 ```
 

src/config/runtime.rs

@@ -28,6 +28,8 @@ pub struct Pingora {
     pub group: Option<String>,
     pub ca_file: Option<String>,
     pub upstream_keepalive_pool_size: Option<usize>,
+    pub grace_period_seconds: Option<u64>,
+    pub graceful_shutdown_timeout_seconds: Option<u64>,
 }
 
 // Initialize global configuration

src/main.rs

@@ -49,6 +49,29 @@ fn main() {
         }
     }
 
+    let conf = config::runtime::config();
+    let http = &conf.proxy.http;
+    let https = &conf.proxy.https;
+
+    // check if the proxy is running
+    match std::net::TcpListener::bind(http) {
+        Ok(_) => {}
+        Err(e) => {
+            tracing::error!("An error occurred while trying to bind to {}: {}", http, e);
+            std::process::exit(1);
+        }
+    };
+    match https {
+        Some(https) => match std::net::TcpListener::bind(https) {
+            Ok(_) => {}
+            Err(e) => {
+                tracing::error!("An error occurred while trying to bind to {}: {}", https, e);
+                std::process::exit(1);
+            }
+        },
+        None => {}
+    }
+
     let rt = tokio::runtime::Runtime::new()
         .map_err(|e| tracing::error!("Error: {:?}", e))
         .unwrap();

src/proxy/mod.rs

@@ -74,6 +74,8 @@ impl EasyProxy {
         if let Some(upstream_keepalive_pool_size) = app_conf.pingora.upstream_keepalive_pool_size {
             conf.upstream_keepalive_pool_size = upstream_keepalive_pool_size;
         }
+        conf.grace_period_seconds = conf.grace_period_seconds.or(Some(1));
+        conf.graceful_shutdown_timeout_seconds = conf.graceful_shutdown_timeout_seconds.or(Some(1));
         pingora_server.configuration = conf.into();
         pingora_server.bootstrap();
         let mut pingora_svc =