mirror of
https://github.com/fafhrd91/actix-net
synced 2024-12-22 13:13:24 +01:00
78 lines
2.6 KiB
Rust
78 lines
2.6 KiB
Rust
//! simple composite service
|
|
//! build: cargo run --example basic --features "ssl"
|
|
//! to test: curl https://127.0.0.1:8443/ -k
|
|
use std::sync::{
|
|
atomic::{AtomicUsize, Ordering},
|
|
Arc,
|
|
};
|
|
use std::{env, fmt, io};
|
|
|
|
use actix_codec::{AsyncRead, AsyncWrite};
|
|
use actix_rt::System;
|
|
use actix_server::{Io, Server};
|
|
use actix_service::{service_fn, NewService};
|
|
use futures_util::{future, Future};
|
|
use openssl::ssl::{SslAcceptor, SslFiletype, SslMethod};
|
|
use tokio_openssl::SslAcceptorExt;
|
|
|
|
/// Simple logger service, it just prints fact of the new connections
|
|
fn logger<T: AsyncRead + AsyncWrite + fmt::Debug>(
|
|
stream: T,
|
|
) -> impl Future<Item = T, Error = ()> {
|
|
println!("New connection: {:?}", stream);
|
|
future::ok(stream)
|
|
}
|
|
|
|
fn main() -> io::Result<()> {
|
|
env::set_var("RUST_LOG", "actix_net=trace");
|
|
env_logger::init();
|
|
|
|
let sys = System::new("test");
|
|
|
|
// load ssl keys
|
|
let mut builder = SslAcceptor::mozilla_intermediate(SslMethod::tls()).unwrap();
|
|
builder
|
|
.set_private_key_file("./examples/key.pem", SslFiletype::PEM)
|
|
.unwrap();
|
|
builder
|
|
.set_certificate_chain_file("./examples/cert.pem")
|
|
.unwrap();
|
|
let acceptor = builder.build();
|
|
|
|
let num = Arc::new(AtomicUsize::new(0));
|
|
|
|
// bind socket address and start workers. By default server uses number of
|
|
// available logical cpu as threads count. actix net start separate
|
|
// instances of service pipeline in each worker.
|
|
Server::build()
|
|
.bind(
|
|
// configure service pipeline
|
|
"basic",
|
|
"0.0.0.0:8443",
|
|
move || {
|
|
let num = num.clone();
|
|
let acceptor = acceptor.clone();
|
|
|
|
// service for converting incoming TcpStream to a SslStream<TcpStream>
|
|
service_fn(move |stream: Io<tokio_tcp::TcpStream>| {
|
|
SslAcceptorExt::accept_async(&acceptor, stream.into_parts().0)
|
|
.map_err(|e| println!("Openssl error: {}", e))
|
|
})
|
|
// .and_then() combinator uses other service to convert incoming `Request` to a
|
|
// `Response` and then uses that response as an input for next
|
|
// service. in this case, on success we use `logger` service
|
|
.and_then(logger)
|
|
// Next service counts number of connections
|
|
.and_then(move |_| {
|
|
let num = num.fetch_add(1, Ordering::Relaxed);
|
|
println!("got ssl connection {:?}", num);
|
|
future::ok(())
|
|
})
|
|
},
|
|
)
|
|
.unwrap()
|
|
.start();
|
|
|
|
sys.run()
|
|
}
|