pezkuwi-subxt/polkadot/node/overseer/orchestra/examples/duo.rs

#![allow(dead_code)] // orchestra events are not used

//! A dummy to be used with cargo expand

use orchestra::{self as orchestra, Spawner, *};
use std::collections::HashMap;
mod misc;

pub use self::misc::*;

/// Concrete subsystem implementation for `MsgStrukt` msg type.
#[derive(Default)]
pub struct AwesomeSubSys;

#[orchestra::subsystem(Awesome, error=Yikes)]
impl<Context> AwesomeSubSys {
	fn start(self, mut ctx: Context) -> SpawnedSubsystem<Yikes> {
		let mut sender = ctx.sender().clone();
		ctx.spawn(
			"AwesomeSubsys",
			Box::pin(async move {
				sender.send_message(Plinko).await;
			}),
		)
		.unwrap();
		unimplemented!("starting yay!")
	}
}

#[derive(Default)]
pub struct Fortified;

#[orchestra::subsystem(GoblinTower, error=Yikes)]
impl<Context> Fortified {
	fn start(self, mut ctx: Context) -> SpawnedSubsystem<Yikes> {
		let mut sender = ctx.sender().clone();
		ctx.spawn(
			"GoblinTower",
			Box::pin(async move {
				sender.send_message(MsgStrukt(8u8)).await;
			}),
		)
		.unwrap();
		unimplemented!("welcum")
	}
}

#[orchestra(signal=SigSigSig, event=EvX, error=Yikes, gen=AllMessages)]
struct Duo<T> {
	#[subsystem(consumes: MsgStrukt, sends: [Plinko])]
	sub0: Awesome,

	#[subsystem(blocking, consumes: Plinko, sends: [MsgStrukt])]
	plinkos: GoblinTower,

	i_like_pi: f64,
	i_like_generic: T,
	i_like_hash: HashMap<f64, f64>,
}

fn main() {
	use futures::{executor, pin_mut};

	executor::block_on(async move {
		let (orchestra, _handle): (Duo<_, f64>, _) = Duo::builder()
			.sub0(AwesomeSubSys::default())
			.plinkos(Fortified::default())
			.i_like_pi(::std::f64::consts::PI)
			.i_like_generic(42.0)
			.i_like_hash(HashMap::new())
			.spawner(DummySpawner)
			.build()
			.unwrap();

		assert_eq!(orchestra.i_like_pi.floor() as i8, 3);
		assert_eq!(orchestra.i_like_generic.floor() as i8, 42);
		assert_eq!(orchestra.i_like_hash.len() as i8, 0);

		let orchestra_fut = orchestra
			.running_subsystems
			.into_future()
			.timeout(std::time::Duration::from_millis(300))
			.fuse();

		pin_mut!(orchestra_fut);

		orchestra_fut.await
	});
}