north 0.1.4

North Microservice Framework
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

use std::convert::Infallible;
use hyper::{Body, Request, Response};
use hyper::server::conn::AddrStream;
use crate::Error;
use crate::service::NorthServiceOptions;

pub struct NorthServer {
}

pub async fn start_server(
    service_option: &NorthServiceOptions,
) -> Result<(), Error> {
    let full_addr = format!(
        "{}:{}",
        service_option.address.clone().unwrap(),
        service_option.port.unwrap().to_string(),
    );
    println!("{}", full_addr);
    let addr: std::net::SocketAddr = full_addr.parse()?;

    let service = hyper::service::make_service_fn(|socket: &AddrStream| {
        let remote_addr = socket.remote_addr();
        async move {
            Ok::<_, Infallible>(hyper::service::service_fn(move |_: Request<Body>| async move {
                Ok::<_, Infallible>(
                    Response::new(Body::from(format!("Hello, {}!", remote_addr)))
                )
            }))
        }
    });

    let server = hyper::Server::bind(&addr)
        .serve(service);

    if !service_option.graceful_shutdown {
        if let Err(e) = server.await {
            eprintln!("server error: {}", e);
        }
        return Ok(());
    }

    let (tx, rx) = tokio::sync::oneshot::channel::<()>();
    let graceful = server
        .with_graceful_shutdown(async {
            rx.await.ok();
        });

    if let Err(e) = graceful.await {
        eprintln!("server error: {}", e);
    }

    let _ = tx.send(());
    Ok(())
}