helpers.rs

use futures_channel::oneshot;
use jsonrpsee::{
	http_server::HttpServerBuilder,
	ws_server::{RpcModule, WsServerBuilder},
};

/// Run jsonrpsee HTTP server for benchmarks.
pub async fn http_server() -> String {
	let (server_started_tx, server_started_rx) = oneshot::channel();
	tokio::spawn(async move {
		let mut server =
			HttpServerBuilder::default().max_request_body_size(u32::MAX).build("127.0.0.1:0".parse().unwrap()).unwrap();
		let mut module = RpcModule::new(());
		module.register_method("say_hello", |_, _| Ok("lo")).unwrap();
		server.register_module(module).unwrap();
		server_started_tx.send(server.local_addr().unwrap()).unwrap();
		server.start().await
	});
	format!("http://{}", server_started_rx.await.unwrap())
}

/// Run jsonrpsee WebSocket server for benchmarks.
pub async fn ws_server() -> String {
	let (server_started_tx, server_started_rx) = oneshot::channel();
	tokio::spawn(async move {
		let mut server = WsServerBuilder::default().build("127.0.0.1:0").await.unwrap();
		let mut module = RpcModule::new(());
		module.register_method("say_hello", |_, _| Ok("lo")).unwrap();
		server.register_module(module).unwrap();
		server_started_tx.send(server.local_addr().unwrap()).unwrap();
		server.start().await
	});
	format!("ws://{}", server_started_rx.await.unwrap())
}

/// Get number of concurrent tasks based on the num_cpus.
pub fn concurrent_tasks() -> Vec<usize> {
	let cores = num_cpus::get();
	vec![cores / 4, cores / 2, cores, cores * 2, cores * 4]
}