mirror of
https://github.com/EasyTier/EasyTier.git
synced 2026-08-06 04:29:52 +00:00
fanyang
f24735a86f
* bench: add packet bytes extraction Criterion benchmark Adds a Criterion benchmark under easytier/benches/ covering ZCPacket::payload_bytes and tunnel_payload_bytes at 1280/4096-byte payload sizes, using iter_batched so ZCPacket construction stays in the setup phase and is excluded from the timed region. - Register the [[bench]] entry in easytier/Cargo.toml. - Document the bench and PACKET_BYTES_* env vars in benches/README.md. * perf: reduce packet buffer slicing churn Replace BytesMut::split_off with Buf::advance in ZCPacket bytes extraction paths (payload_bytes, tunnel_payload_bytes, convert_type, drop_foreign_header) and in TunZCPacketToBytes, and simplify the copy_from_slice in new_from_payload. When the buffer is in its unique (VEC) representation, split_off promotes it to the shared (ARC) representation, allocating a Shared control block and bumping the refcount on every call, and pins the buffer in shared mode. advance only mutates the in-place ptr/len/cap fields, avoiding that allocation/refcount churn on the TX hot path. The byte data itself is not copied by either path.
66 lines
2.0 KiB
Rust
66 lines
2.0 KiB
Rust
use std::hint::black_box;
|
|
use std::time::Duration;
|
|
|
|
use criterion::{BatchSize, Criterion, Throughput, criterion_group, criterion_main};
|
|
|
|
use easytier::tunnel::packet_def::ZCPacket;
|
|
|
|
const PAYLOAD_SIZES: &[usize] = &[1280, 4096];
|
|
|
|
fn env_parse<T: std::str::FromStr>(key: &str, default: T) -> T {
|
|
std::env::var(key)
|
|
.ok()
|
|
.and_then(|v| v.parse().ok())
|
|
.unwrap_or(default)
|
|
}
|
|
|
|
fn bench_payload_bytes(c: &mut Criterion) {
|
|
let mut group = c.benchmark_group("payload_bytes");
|
|
for &size in PAYLOAD_SIZES {
|
|
let data = vec![0u8; size];
|
|
group.throughput(Throughput::Bytes(size as u64));
|
|
group.bench_with_input(format!("{size}"), &data, |b, data| {
|
|
b.iter_batched(
|
|
|| ZCPacket::new_with_payload(black_box(data)),
|
|
|p| black_box(p).payload_bytes(),
|
|
BatchSize::SmallInput,
|
|
)
|
|
});
|
|
}
|
|
group.finish();
|
|
}
|
|
|
|
fn bench_tunnel_payload_bytes(c: &mut Criterion) {
|
|
let mut group = c.benchmark_group("tunnel_payload_bytes");
|
|
for &size in PAYLOAD_SIZES {
|
|
let data = vec![0u8; size];
|
|
group.throughput(Throughput::Bytes(size as u64));
|
|
group.bench_with_input(format!("{size}"), &data, |b, data| {
|
|
b.iter_batched(
|
|
|| ZCPacket::new_with_payload(black_box(data)),
|
|
|p| black_box(p).tunnel_payload_bytes(),
|
|
BatchSize::SmallInput,
|
|
)
|
|
});
|
|
}
|
|
group.finish();
|
|
}
|
|
|
|
fn criterion_config() -> Criterion {
|
|
let measurement_secs = env_parse("PACKET_BYTES_MEASUREMENT_SECS", 10u64);
|
|
let warmup_secs = env_parse("PACKET_BYTES_WARMUP_SECS", 3u64);
|
|
let sample_size = env_parse("PACKET_BYTES_SAMPLE_SIZE", 10usize).max(10);
|
|
|
|
Criterion::default()
|
|
.measurement_time(Duration::from_secs(measurement_secs))
|
|
.warm_up_time(Duration::from_secs(warmup_secs))
|
|
.sample_size(sample_size)
|
|
}
|
|
|
|
criterion_group! {
|
|
name = benches;
|
|
config = criterion_config();
|
|
targets = bench_payload_bytes, bench_tunnel_payload_bytes
|
|
}
|
|
criterion_main!(benches);
|