oxpanel/oxspeak_server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add event_bus_typed

Browse Source
This commit is contained in:
Nell
2026-09-23 10:36:14 +02:00
parent 3780092fa6
commit ab97dcc8d9
35 changed files with 1376 additions and 3228 deletions
Download Patch File Download Diff File Expand all files Collapse all files
event_bus/benches/event_bus_throughput.rs
+117 -160
View File
@@ -1,3 +1,5 @@
#![allow(dead_code)]
use std::sync::{
Arc,
atomic::{AtomicU64, Ordering},
@@ -8,8 +10,6 @@ use criterion::{Criterion, Throughput, criterion_group, criterion_main};
use event_bus::EventBus;
use tokio::runtime::Runtime;
const TOPIC: &str = "bench-topic";
#[derive(Clone)]
struct SmallEvent {
value: u64,
@@ -38,57 +38,46 @@ fn runtime() -> Runtime {
Runtime::new().expect("failed to create tokio runtime")
}
fn wait_until_received(
received: &AtomicU64,
expected: u64,
) -> impl std::future::Future<Output = ()> + '_ {
async move {
while received.load(Ordering::Relaxed) < expected {
tokio::task::yield_now().await;
}
async fn wait_until_received(received: &AtomicU64, expected: u64) {
while received.load(Ordering::Relaxed) < expected {
tokio::task::yield_now().await;
}
}
fn bench_emit_no_subscriber(c: &mut Criterion) {
let mut group = c.benchmark_group("event_bus/no_subscriber");
let mut group = c.benchmark_group("event_bus_typed/no_subscriber");
group.throughput(Throughput::Elements(1));
let bus = EventBus::with_capacity(1024);
group.bench_function("u64", |b| {
b.iter(|| {
bus.emit(TOPIC, 42_u64);
bus.emit(42_u64);
});
});
group.bench_function("small_struct", |b| {
b.iter(|| {
bus.emit(TOPIC, SmallEvent { value: 42 });
bus.emit(SmallEvent { value: 42 });
});
});
group.bench_function("string_struct", |b| {
b.iter(|| {
bus.emit(
TOPIC,
StringEvent {
id: 42,
name: "Alice".to_string(),
message: "hello from benchmark".to_string(),
},
);
bus.emit(StringEvent {
id: 42,
name: "Alice".to_string(),
message: "hello from benchmark".to_string(),
});
});
});
group.bench_function("vec_payload_1kb", |b| {
b.iter(|| {
bus.emit(
TOPIC,
VecEvent {
id: 42,
payload: vec![7_u8; 1024],
},
);
bus.emit(VecEvent {
id: 42,
payload: vec![7_u8; 1024],
});
});
});
@@ -96,13 +85,10 @@ fn bench_emit_no_subscriber(c: &mut Criterion) {
group.bench_function("arc_payload_1kb", |b| {
b.iter(|| {
bus.emit(
TOPIC,
ArcPayloadEvent {
id: 42,
payload: Arc::clone(&shared_payload),
},
);
bus.emit(ArcPayloadEvent {
id: 42,
payload: Arc::clone(&shared_payload),
});
});
});
@@ -112,13 +98,13 @@ fn bench_emit_no_subscriber(c: &mut Criterion) {
fn bench_raw_subscriber(c: &mut Criterion) {
let rt = runtime();
let mut group = c.benchmark_group("event_bus/raw_subscriber");
let mut group = c.benchmark_group("event_bus_typed/raw_subscriber");
group.throughput(Throughput::Elements(1));
group.bench_function("u64", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let mut rx = bus.on_raw(TOPIC);
let bus = EventBus::with_capacity(iters as usize + 1024);
let mut rx = bus.subscribe::<u64>();
let received = Arc::new(AtomicU64::new(0));
let receiver_count = Arc::clone(&received);
@@ -134,23 +120,21 @@ fn bench_raw_subscriber(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(TOPIC, i);
bus.emit(i);
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
receiver.abort();
elapsed
});
});
group.bench_function("small_struct", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let mut rx = bus.on_raw(TOPIC);
let bus = EventBus::with_capacity(iters as usize + 1024);
let mut rx = bus.subscribe::<SmallEvent>();
let received = Arc::new(AtomicU64::new(0));
let receiver_count = Arc::clone(&received);
@@ -166,23 +150,21 @@ fn bench_raw_subscriber(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(TOPIC, SmallEvent { value: i });
bus.emit(SmallEvent { value: i });
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
receiver.abort();
elapsed
});
});
group.bench_function("string_struct", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let mut rx = bus.on_raw(TOPIC);
let bus = EventBus::with_capacity(iters as usize + 1024);
let mut rx = bus.subscribe::<StringEvent>();
let received = Arc::new(AtomicU64::new(0));
let receiver_count = Arc::clone(&received);
@@ -198,30 +180,25 @@ fn bench_raw_subscriber(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(
TOPIC,
StringEvent {
id: i,
name: "Alice".to_string(),
message: "hello from benchmark".to_string(),
},
);
bus.emit(StringEvent {
id: i,
name: "Alice".to_string(),
message: "hello from benchmark".to_string(),
});
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
receiver.abort();
elapsed
});
});
group.bench_function("vec_payload_1kb", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let mut rx = bus.on_raw(TOPIC);
let bus = EventBus::with_capacity(iters as usize + 1024);
let mut rx = bus.subscribe::<VecEvent>();
let received = Arc::new(AtomicU64::new(0));
let receiver_count = Arc::clone(&received);
@@ -237,62 +214,56 @@ fn bench_raw_subscriber(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(
TOPIC,
VecEvent {
id: i,
payload: vec![7_u8; 1024],
},
);
bus.emit(VecEvent {
id: i,
payload: vec![7_u8; 1024],
});
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
receiver.abort();
elapsed
});
});
let shared_payload: Arc<[u8]> = Arc::from(vec![7_u8; 1024].into_boxed_slice());
group.bench_function("arc_payload_1kb", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let mut rx = bus.on_raw(TOPIC);
let shared = Arc::clone(&shared_payload);
b.to_async(&rt).iter_custom(|iters| {
let payload = Arc::clone(&shared);
async move {
let bus = EventBus::with_capacity(iters as usize + 1024);
let mut rx = bus.subscribe::<ArcPayloadEvent>();
let payload: Arc<[u8]> = Arc::from(vec![7_u8; 1024].into_boxed_slice());
let received = Arc::new(AtomicU64::new(0));
let receiver_count = Arc::clone(&received);
let received = Arc::new(AtomicU64::new(0));
let receiver_count = Arc::clone(&received);
let receiver = tokio::spawn(async move {
while receiver_count.load(Ordering::Relaxed) < iters {
if rx.recv().await.is_ok() {
receiver_count.fetch_add(1, Ordering::Relaxed);
let receiver = tokio::spawn(async move {
while receiver_count.load(Ordering::Relaxed) < iters {
if rx.recv().await.is_ok() {
receiver_count.fetch_add(1, Ordering::Relaxed);
}
}
}
});
});
let start = Instant::now();
let start = Instant::now();
for i in 0..iters {
bus.emit(
TOPIC,
ArcPayloadEvent {
for i in 0..iters {
bus.emit(ArcPayloadEvent {
id: i,
payload: Arc::clone(&payload),
},
);
});
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
receiver.abort();
elapsed
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
receiver.abort();
elapsed
});
});
@@ -302,17 +273,17 @@ fn bench_raw_subscriber(c: &mut Criterion) {
fn bench_typed_callback(c: &mut Criterion) {
let rt = runtime();
let mut group = c.benchmark_group("event_bus/typed_callback");
let mut group = c.benchmark_group("event_bus_typed/typed_callback");
group.throughput(Throughput::Elements(1));
group.bench_function("u64", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let bus = EventBus::with_capacity(iters as usize + 1024);
let received = Arc::new(AtomicU64::new(0));
let handler_count = Arc::clone(&received);
let subscription = bus.on::<u64, _>(TOPIC, move |event| {
let subscription = bus.on::<u64>(move |event| {
let _ = event;
handler_count.fetch_add(1, Ordering::Relaxed);
});
@@ -320,27 +291,25 @@ fn bench_typed_callback(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(TOPIC, i);
bus.emit(i);
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
subscription.abort();
elapsed
});
});
group.bench_function("small_struct", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let bus = EventBus::with_capacity(iters as usize + 1024);
let received = Arc::new(AtomicU64::new(0));
let handler_count = Arc::clone(&received);
let subscription = bus.on::<SmallEvent, _>(TOPIC, move |event| {
let subscription = bus.on::<SmallEvent>(move |event| {
let _ = event.value;
handler_count.fetch_add(1, Ordering::Relaxed);
});
@@ -348,27 +317,25 @@ fn bench_typed_callback(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(TOPIC, SmallEvent { value: i });
bus.emit(SmallEvent { value: i });
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
subscription.abort();
elapsed
});
});
group.bench_function("string_struct", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let bus = EventBus::with_capacity(iters as usize + 1024);
let received = Arc::new(AtomicU64::new(0));
let handler_count = Arc::clone(&received);
let subscription = bus.on::<StringEvent, _>(TOPIC, move |event| {
let subscription = bus.on::<StringEvent>(move |event| {
let _ = event.id;
let _ = event.name.len();
let _ = event.message.len();
@@ -378,34 +345,29 @@ fn bench_typed_callback(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(
TOPIC,
StringEvent {
id: i,
name: "Alice".to_string(),
message: "hello from benchmark".to_string(),
},
);
bus.emit(StringEvent {
id: i,
name: "Alice".to_string(),
message: "hello from benchmark".to_string(),
});
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
subscription.abort();
elapsed
});
});
group.bench_function("vec_payload_1kb", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let bus = EventBus::with_capacity(iters as usize + 1024);
let received = Arc::new(AtomicU64::new(0));
let handler_count = Arc::clone(&received);
let subscription = bus.on::<VecEvent, _>(TOPIC, move |event| {
let subscription = bus.on::<VecEvent>(move |event| {
let _ = event.id;
let _ = event.payload.len();
handler_count.fetch_add(1, Ordering::Relaxed);
@@ -414,58 +376,53 @@ fn bench_typed_callback(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(
TOPIC,
VecEvent {
id: i,
payload: vec![7_u8; 1024],
},
);
bus.emit(VecEvent {
id: i,
payload: vec![7_u8; 1024],
});
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
subscription.abort();
elapsed
});
});
let shared_payload: Arc<[u8]> = Arc::from(vec![7_u8; 1024].into_boxed_slice());
group.bench_function("arc_payload_1kb", |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let payload: Arc<[u8]> = Arc::from(vec![7_u8; 1024].into_boxed_slice());
let shared = Arc::clone(&shared_payload);
b.to_async(&rt).iter_custom(|iters| {
let payload = Arc::clone(&shared);
async move {
let bus = EventBus::with_capacity(iters as usize + 1024);
let received = Arc::new(AtomicU64::new(0));
let handler_count = Arc::clone(&received);
let received = Arc::new(AtomicU64::new(0));
let handler_count = Arc::clone(&received);
let subscription = bus.on::<ArcPayloadEvent, _>(TOPIC, move |event| {
let _ = event.id;
let _ = event.payload.len();
handler_count.fetch_add(1, Ordering::Relaxed);
});
let subscription = bus.on::<ArcPayloadEvent>(move |event| {
let _ = event.id;
let _ = event.payload.len();
handler_count.fetch_add(1, Ordering::Relaxed);
});
let start = Instant::now();
let start = Instant::now();
for i in 0..iters {
bus.emit(
TOPIC,
ArcPayloadEvent {
for i in 0..iters {
bus.emit(ArcPayloadEvent {
id: i,
payload: Arc::clone(&payload),
},
);
});
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
subscription.abort();
elapsed
}
wait_until_received(&received, iters).await;
let elapsed = start.elapsed();
subscription.abort();
elapsed
});
});
@@ -475,13 +432,13 @@ fn bench_typed_callback(c: &mut Criterion) {
fn bench_multiple_subscribers(c: &mut Criterion) {
let rt = runtime();
let mut group = c.benchmark_group("event_bus/multiple_subscribers");
let mut group = c.benchmark_group("event_bus_typed/multiple_subscribers");
group.throughput(Throughput::Elements(1));
for subscriber_count in [1_u64, 2, 4, 8, 16, 32] {
group.bench_function(format!("{subscriber_count}_subscribers"), |b| {
b.to_async(&rt).iter_custom(|iters| async move {
let bus = Arc::new(EventBus::with_capacity(iters as usize + 1024));
let bus = EventBus::with_capacity(iters as usize + 1024);
let expected = iters * subscriber_count;
let received = Arc::new(AtomicU64::new(0));
@@ -490,7 +447,7 @@ fn bench_multiple_subscribers(c: &mut Criterion) {
for _ in 0..subscriber_count {
let handler_count = Arc::clone(&received);
let subscription = bus.on::<SmallEvent, _>(TOPIC, move |event| {
let subscription = bus.on::<SmallEvent>(move |event| {
let _ = event.value;
handler_count.fetch_add(1, Ordering::Relaxed);
});
@@ -501,7 +458,7 @@ fn bench_multiple_subscribers(c: &mut Criterion) {
let start = Instant::now();
for i in 0..iters {
bus.emit(TOPIC, SmallEvent { value: i });
bus.emit(SmallEvent { value: i });
}
wait_until_received(&received, expected).await;