geo302 0.1.4

A simple geoIP-based redirect proxy
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65

use crate::Mirror;

use hyper::client::Client;
use hyper::http::uri::Uri;
use hyper::StatusCode;
use hyper_tls::HttpsConnector;
use std::sync::atomic;
use std::time::Duration;
use thiserror::Error;

const TIMEOUT: Duration = Duration::new(5, 0);

#[derive(Debug, Error)]
enum RequestError {
    #[error(transparent)]
    Http(#[from] hyper::Error),
    #[error("Connection timeout")]
    Timeout(#[from] tokio::time::error::Elapsed),
}

pub struct HealthCheck {}

impl HealthCheck {
    async fn get_status<C>(client: &Client<C>, uri: Uri) -> Result<StatusCode, RequestError>
    where
        C: hyper::client::connect::Connect + Clone + Send + Sync + 'static,
    {
        let response = tokio::time::timeout(TIMEOUT, client.get(uri)).await?;
        Ok(response?.status())
    }

    pub fn start(mirrors: &[Mirror], interval: Duration) -> Self {
        let https = HttpsConnector::new();
        let http_client = Client::builder().build::<_, hyper::Body>(https);
        for mirror in mirrors {
            let http_client = http_client.clone();
            let mirror = mirror.clone();
            tokio::spawn(async move {
                loop {
                    let status = Self::get_status(&http_client, mirror.healthcheck.clone()).await;
                    // Use Result.is_ok_and when stabilizes
                    // https://github.com/rust-lang/rust/issues/93050
                    let new_available = match status {
                        Ok(success) => success.is_success(),
                        Err(_) => false,
                    };
                    mirror
                        .available
                        .store(new_available, atomic::Ordering::Release);
                    match (new_available, status) {
                        (true, Ok(_)) => log::info!("{} is alive", mirror.healthcheck),
                        (false, Ok(status)) => {
                            log::warn!("{} is unavailable: {}", mirror.healthcheck, status)
                        }
                        (_, Err(e)) => {
                            log::warn!("{} is unavailable: {}", mirror.healthcheck, e.to_string())
                        }
                    }
                    tokio::time::sleep(interval).await;
                }
            });
        }
        Self {}
    }
}