2022-01-19 13:40:50 +00:00
|
|
|
use preserves_schema::Codec;
|
|
|
|
|
2021-08-28 16:50:55 +00:00
|
|
|
use std::convert::TryFrom;
|
2021-08-27 14:19:14 +00:00
|
|
|
use std::sync::Arc;
|
|
|
|
|
|
|
|
use syndicate::actor::*;
|
2021-09-23 19:46:10 +00:00
|
|
|
use syndicate::enclose;
|
2022-01-19 13:40:50 +00:00
|
|
|
use syndicate::preserves::rec;
|
|
|
|
use syndicate::preserves::value::NestedValue;
|
2021-08-30 21:41:51 +00:00
|
|
|
use syndicate::supervise::{Supervisor, SupervisorConfiguration};
|
2022-01-19 13:40:50 +00:00
|
|
|
use syndicate::trace;
|
2021-08-27 14:19:14 +00:00
|
|
|
|
|
|
|
use tokio::net::TcpListener;
|
|
|
|
|
2021-09-19 14:53:37 +00:00
|
|
|
use crate::language::language;
|
2021-09-28 10:53:11 +00:00
|
|
|
use crate::lifecycle;
|
2021-08-27 14:19:14 +00:00
|
|
|
use crate::protocol::detect_protocol;
|
2021-09-28 10:53:11 +00:00
|
|
|
use crate::schemas::internal_services::TcpRelayListener;
|
2021-08-27 14:19:14 +00:00
|
|
|
|
2021-09-20 13:10:31 +00:00
|
|
|
use syndicate_macros::during;
|
|
|
|
|
2021-10-07 15:00:04 +00:00
|
|
|
pub fn on_demand(t: &mut Activation, ds: Arc<Cap>) {
|
2022-01-19 13:40:50 +00:00
|
|
|
t.spawn(Some(AnyValue::symbol("tcp_relay_listener")), move |t| {
|
2021-09-28 10:53:11 +00:00
|
|
|
Ok(during!(t, ds, language(), <run-service $spec: TcpRelayListener>, |t| {
|
|
|
|
Supervisor::start(
|
|
|
|
t,
|
2022-01-19 13:40:50 +00:00
|
|
|
Some(rec![AnyValue::symbol("relay"), language().unparse(&spec)]),
|
2021-09-28 10:53:11 +00:00
|
|
|
SupervisorConfiguration::default(),
|
|
|
|
enclose!((ds, spec) lifecycle::updater(ds, spec)),
|
2021-10-07 15:00:04 +00:00
|
|
|
enclose!((ds) move |t| enclose!((ds, spec) run(t, ds, spec))))
|
2021-09-28 10:53:11 +00:00
|
|
|
}))
|
2021-09-01 15:31:01 +00:00
|
|
|
});
|
2021-08-27 14:19:14 +00:00
|
|
|
}
|
|
|
|
|
2021-10-07 15:00:04 +00:00
|
|
|
fn run(t: &mut Activation, ds: Arc<Cap>, spec: TcpRelayListener) -> ActorResult {
|
2022-01-07 16:18:00 +00:00
|
|
|
lifecycle::terminate_on_service_restart(t, &ds, &spec);
|
2021-08-28 16:50:55 +00:00
|
|
|
let host = spec.addr.host.clone();
|
|
|
|
let port = u16::try_from(&spec.addr.port).map_err(|_| "Invalid TCP port number")?;
|
2021-09-28 10:53:11 +00:00
|
|
|
let facet = t.facet.clone();
|
2022-01-19 13:40:50 +00:00
|
|
|
let trace_collector = t.trace_collector();
|
|
|
|
t.linked_task(Some(AnyValue::symbol("listener")), async move {
|
2021-08-28 16:50:55 +00:00
|
|
|
let listen_addr = format!("{}:{}", host, port);
|
|
|
|
let listener = TcpListener::bind(listen_addr).await?;
|
2022-01-15 23:02:33 +00:00
|
|
|
|
|
|
|
{
|
2022-01-19 13:40:50 +00:00
|
|
|
let cause = trace_collector.as_ref().map(|_| trace::TurnCause::external("readiness"));
|
|
|
|
let account = Account::new(Some(AnyValue::symbol("readiness")), trace_collector.clone());
|
|
|
|
if !facet.activate(
|
|
|
|
&account, cause, |t| {
|
|
|
|
tracing::info!("listening");
|
|
|
|
ds.assert(t, language(), &lifecycle::ready(&spec));
|
|
|
|
Ok(())
|
|
|
|
})
|
|
|
|
{
|
|
|
|
return Ok(LinkedTaskTermination::Normal);
|
|
|
|
}
|
2022-01-15 23:02:33 +00:00
|
|
|
}
|
|
|
|
|
2021-08-28 16:50:55 +00:00
|
|
|
loop {
|
|
|
|
let (stream, addr) = listener.accept().await?;
|
2021-10-07 15:00:04 +00:00
|
|
|
let gatekeeper = spec.gatekeeper.clone();
|
2022-01-19 13:40:50 +00:00
|
|
|
let name = Some(rec![AnyValue::symbol("tcp"), AnyValue::new(format!("{}", &addr))]);
|
|
|
|
let cause = trace_collector.as_ref().map(|_| trace::TurnCause::external("connect"));
|
|
|
|
let account = Account::new(name.clone(), trace_collector.clone());
|
2022-01-15 23:02:33 +00:00
|
|
|
if !facet.activate(
|
2022-01-19 13:40:50 +00:00
|
|
|
&account, cause, enclose!((trace_collector) move |t| {
|
2022-01-15 23:02:33 +00:00
|
|
|
t.spawn(name, move |t| {
|
2022-01-19 13:40:50 +00:00
|
|
|
Ok(t.linked_task(None, {
|
2022-01-15 23:02:33 +00:00
|
|
|
let facet = t.facet.clone();
|
|
|
|
async move {
|
2022-01-19 13:40:50 +00:00
|
|
|
detect_protocol(trace_collector, facet, stream, gatekeeper, addr).await?;
|
2022-01-15 23:02:33 +00:00
|
|
|
Ok(LinkedTaskTermination::KeepFacet)
|
|
|
|
}
|
|
|
|
}))
|
|
|
|
});
|
|
|
|
Ok(())
|
2022-01-19 13:40:50 +00:00
|
|
|
}))
|
2022-01-15 23:02:33 +00:00
|
|
|
{
|
|
|
|
return Ok(LinkedTaskTermination::Normal);
|
|
|
|
}
|
2021-08-28 16:50:55 +00:00
|
|
|
}
|
|
|
|
});
|
|
|
|
Ok(())
|
2021-08-27 14:19:14 +00:00
|
|
|
}
|