proksi 0.9.0

A batteries-included reverse proxy with automatic HTTPS using Cloudflare Pingora and Let's Encrypt.
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
66
67
68
69
70
71
72
73
74
75
76
77
78
79

//! Timeouts toward clients (`server.timeouts`), for every HTTP and HTTPS connection

use std::{sync::Arc, time::Duration};

use async_trait::async_trait;
use once_cell::sync::OnceCell;
use pingora::{
    apps::{HttpServerApp, HttpServerOptions, ReusedHttpStream},
    protocols::{http::ServerSession, Stream},
    server::ShutdownWatch,
};

use crate::config::ClientTimeouts;

static TIMEOUTS: OnceCell<ClientTimeouts> = OnceCell::new();

/// Sets the timeouts of the HTTP and HTTPS apps built from now on
pub fn set_client_timeouts(timeouts: ClientTimeouts) {
    let _ = TIMEOUTS.set(timeouts);
}

/// An HTTP app whose clients get the configured timeouts: Pingora's own leave connections
/// kept alive, and responses to clients that stop reading, open forever
pub struct WithTimeouts<A> {
    app: Arc<A>,
    timeouts: ClientTimeouts,
    options: HttpServerOptions,
}

impl<A: HttpServerApp> WithTimeouts<A> {
    pub fn new(app: A) -> Self {
        let timeouts = TIMEOUTS.get().copied().unwrap_or_default();
        let mut options = app.server_options().cloned().unwrap_or_default();
        options.h2_idle_timeout = Some(Duration::from_secs(timeouts.idle_secs));
        Self {
            app: Arc::new(app),
            timeouts,
            options,
        }
    }
}

#[async_trait]
impl<A: HttpServerApp + Send + Sync + 'static> HttpServerApp for WithTimeouts<A> {
    async fn process_new_http(
        self: &Arc<Self>,
        mut session: ServerSession,
        shutdown: &ShutdownWatch,
    ) -> Option<ReusedHttpStream> {
        // Waiting for a request (the first, or the next one on a kept-alive connection), unless
        // the connection is closing
        if session.get_keepalive().is_some() {
            session.set_keepalive(Some(self.timeouts.idle_secs));
        }
        session.set_read_timeout(Some(Duration::from_secs(self.timeouts.read_secs)));
        session.set_write_timeout(Some(Duration::from_secs(self.timeouts.write_secs)));
        self.app.process_new_http(session, shutdown).await
    }

    fn h2_options(&self) -> Option<pingora::protocols::http::v2::server::H2Options> {
        self.app.h2_options()
    }

    fn server_options(&self) -> Option<&HttpServerOptions> {
        Some(&self.options)
    }

    async fn http_cleanup(&self) {
        self.app.http_cleanup().await;
    }

    async fn process_custom_session(
        self: Arc<Self>,
        stream: Stream,
        shutdown: &ShutdownWatch,
    ) -> Option<Stream> {
        self.app.clone().process_custom_session(stream, shutdown).await
    }
}