Archives
/
encrypted-dns-server
mirror of https://github.com/jedisct1/encrypted-dns-server
2
0
Fork 0
Code Issues Releases Wiki Activity

Add process feature to prometheus, and a timeout for clients

Browse Source
pull/12/head
Frank Denis 5 years ago
parent 42d361897b
commit 22d84a748c
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File

1
Cargo.toml
Unescape Escape View File

@ -46,6 +46,7 @@ default_features = false
optional = true optional = true
version = "0.7.0" version = "0.7.0"
default_features = false default_features = false
features = ["process"]
[features] [features]
default = ["metrics"] default = ["metrics"]

11
src/metrics.rs
Unescape Escape View File

@ -10,8 +10,11 @@ use hyper::{Body, Request, Response, StatusCode};
use prometheus::{self, Encoder, TextEncoder}; use prometheus::{self, Encoder, TextEncoder};
use std::sync::Arc; use std::sync::Arc;
use tokio::net::TcpListener; use tokio::net::TcpListener;
use tokio::prelude::*;
use tokio::runtime::Runtime; use tokio::runtime::Runtime;
const METRICS_CONNECTION_TIMEOUT_SECS: u64 = 10;
async fn handle_client_connection( async fn handle_client_connection(
req: Request<Body>, req: Request<Body>,
varz: Varz, varz: Varz,
@ -60,7 +63,13 @@ pub async fn prometheus_service(
let service = let service =
service_fn(move |req| handle_client_connection(req, varz.clone(), path.clone())); service_fn(move |req| handle_client_connection(req, varz.clone(), path.clone()));
let connection = Http::new().serve_connection(client, service); let connection = Http::new().serve_connection(client, service);
runtime.spawn(connection.map(|_| {})); runtime.spawn(
connection
.timeout(std::time::Duration::from_secs(
METRICS_CONNECTION_TIMEOUT_SECS,
))
.map(|_| {}),
);
} }
Ok(()) Ok(())
} }

Write Preview
Loading…
Cancel
Save