mirror of
https://github.com/clash-verge-rev/clash-verge-rev.git
synced 2026-04-13 13:30:31 +08:00
dbb4877be6
* feat: implement draft functionality with apply and discard methods, and add benchmarks and tests * Refactor Draft management and integrate Tokio for asynchronous operations - Introduced a new `IVerge` struct for configuration management. - Updated `Draft` struct to use `Arc<RwLock>` for better concurrency handling. - Added asynchronous editing capabilities to `Draft` using Tokio. - Replaced synchronous editing methods with asynchronous counterparts. - Updated benchmark tests to reflect changes in the `Draft` API. - Removed redundant draft utility module and integrated its functionality into the main `Draft` implementation. - Adjusted tests to validate new behavior and ensure correctness of the `Draft` management flow.
130 lines
3.6 KiB
Rust
130 lines
3.6 KiB
Rust
use criterion::{Criterion, criterion_group, criterion_main};
|
|
use std::hint::black_box;
|
|
use std::process;
|
|
use tokio::runtime::Runtime;
|
|
|
|
use draft::Draft;
|
|
|
|
#[derive(Clone, Debug)]
|
|
struct IVerge {
|
|
enable_auto_launch: Option<bool>,
|
|
enable_tun_mode: Option<bool>,
|
|
}
|
|
|
|
impl Default for IVerge {
|
|
fn default() -> Self {
|
|
Self {
|
|
enable_auto_launch: None,
|
|
enable_tun_mode: None,
|
|
}
|
|
}
|
|
}
|
|
|
|
fn make_draft() -> Draft<IVerge> {
|
|
let verge = IVerge {
|
|
enable_auto_launch: Some(true),
|
|
enable_tun_mode: Some(false),
|
|
..Default::default()
|
|
};
|
|
Draft::new(verge)
|
|
}
|
|
|
|
pub fn bench_draft(c: &mut Criterion) {
|
|
let rt = Runtime::new().unwrap_or_else(|e| {
|
|
eprintln!("Tokio runtime init failed: {e}");
|
|
process::exit(1);
|
|
});
|
|
|
|
let mut group = c.benchmark_group("draft");
|
|
group.sample_size(100);
|
|
group.warm_up_time(std::time::Duration::from_millis(300));
|
|
group.measurement_time(std::time::Duration::from_secs(1));
|
|
|
|
group.bench_function("data_mut", |b| {
|
|
b.iter(|| {
|
|
let draft = black_box(make_draft());
|
|
draft.edit_draft(|d| d.enable_tun_mode = Some(true));
|
|
black_box(&draft.latest_arc().enable_tun_mode);
|
|
});
|
|
});
|
|
|
|
group.bench_function("draft_mut_first", |b| {
|
|
b.iter(|| {
|
|
let draft = black_box(make_draft());
|
|
draft.edit_draft(|d| d.enable_auto_launch = Some(false));
|
|
let latest = draft.latest_arc();
|
|
black_box(&latest.enable_auto_launch);
|
|
});
|
|
});
|
|
|
|
group.bench_function("draft_mut_existing", |b| {
|
|
b.iter(|| {
|
|
let draft = black_box(make_draft());
|
|
{
|
|
draft.edit_draft(|d| {
|
|
d.enable_tun_mode = Some(true);
|
|
});
|
|
let latest1 = draft.latest_arc();
|
|
black_box(&latest1.enable_tun_mode);
|
|
}
|
|
draft.edit_draft(|d| {
|
|
d.enable_tun_mode = Some(false);
|
|
});
|
|
let latest2 = draft.latest_arc();
|
|
black_box(&latest2.enable_tun_mode);
|
|
});
|
|
});
|
|
|
|
group.bench_function("latest_arc", |b| {
|
|
b.iter(|| {
|
|
let draft = black_box(make_draft());
|
|
let latest = draft.latest_arc();
|
|
black_box(&latest.enable_auto_launch);
|
|
});
|
|
});
|
|
|
|
group.bench_function("apply", |b| {
|
|
b.iter(|| {
|
|
let draft = black_box(make_draft());
|
|
{
|
|
draft.edit_draft(|d| {
|
|
d.enable_auto_launch = Some(false);
|
|
});
|
|
}
|
|
draft.apply();
|
|
black_box(&draft);
|
|
});
|
|
});
|
|
|
|
group.bench_function("discard", |b| {
|
|
b.iter(|| {
|
|
let draft = black_box(make_draft());
|
|
{
|
|
draft.edit_draft(|d| {
|
|
d.enable_auto_launch = Some(false);
|
|
});
|
|
}
|
|
draft.discard();
|
|
black_box(&draft);
|
|
});
|
|
});
|
|
|
|
group.bench_function("with_data_modify_async", |b| {
|
|
b.to_async(&rt).iter(|| async {
|
|
let draft = black_box(make_draft());
|
|
let _: Result<(), anyhow::Error> = draft
|
|
.with_data_modify::<_, _, _>(|mut box_data| async move {
|
|
box_data.enable_auto_launch =
|
|
Some(!box_data.enable_auto_launch.unwrap_or(false));
|
|
Ok((box_data, ()))
|
|
})
|
|
.await;
|
|
});
|
|
});
|
|
|
|
group.finish();
|
|
}
|
|
|
|
criterion_group!(benches, bench_draft);
|
|
criterion_main!(benches);
|