plane-dynamic-proxy 0.5.5

Dynamic proxy crate for Plane
Documentation 
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

use bytes::Bytes;
use hyper::StatusCode;
use plane_dynamic_proxy::body::to_simple_body;
use plane_dynamic_proxy::server::{HttpsConfig, SimpleHttpServer};
use std::convert::Infallible;
use std::net::SocketAddr;
use tokio::net::TcpListener;
use tokio::time::Duration;

// Ref: https://github.com/hyperium/hyper-util/blob/master/examples/server_graceful.rs

async fn slow_hello_world(
    _: hyper::Request<hyper::body::Incoming>,
) -> Result<hyper::Response<plane_dynamic_proxy::body::SimpleBody>, Infallible> {
    tokio::time::sleep(Duration::from_secs(1)).await; // emulate slow request
    let body = http_body_util::Full::<Bytes>::from("Hello, world!".to_owned());
    let body = to_simple_body(body);
    Ok(hyper::Response::new(body))
}

#[tokio::test]
async fn test_graceful_shutdown() {
    // Start the server
    let addr = SocketAddr::from(([127, 0, 0, 1], 0));
    let listener = TcpListener::bind(addr).await.unwrap();
    let addr = listener.local_addr().unwrap();
    let server = SimpleHttpServer::new(
        hyper::service::service_fn(slow_hello_world),
        listener,
        HttpsConfig::Http,
    )
    .unwrap();

    let url = format!("http://{}", addr);

    // Create a client and start a POST request without finishing the body
    let client = reqwest::Client::new();

    let response_handle = {
        let client = client.clone();
        let url = url.clone();
        tokio::spawn(async move { client.get(&url).send().await.unwrap() })
    };

    tokio::time::sleep(Duration::from_millis(100)).await;

    // Call server.graceful_shutdown()
    let shutdown_task = tokio::spawn(async move { server.graceful_shutdown().await });

    tokio::time::sleep(Duration::from_millis(100)).await;

    let response = response_handle.await.unwrap();

    // Wait for the shutdown task to complete.
    shutdown_task.await.unwrap();

    // Ensure that the result is as expected
    assert_eq!(response.status(), StatusCode::OK);
    assert_eq!(response.text().await.unwrap(), "Hello, world!");

    // Attempt to make another request, which should fail due to the server shutting down
    let result = client.get(&url).send().await;
    assert!(result.is_err());
}