// Copyright 2026 PingCAP, Inc.

use std::time::Duration;

use concurrency_manager::{ActionOnInvalidMaxTs, ConcurrencyManager};
use criterion::{Criterion, black_box, criterion_group, criterion_main};
use txn_types::TimeStamp;

fn benchmark_update_max_ts(c: &mut Criterion) {
    let latest_ts = TimeStamp::new(1000);
    let limit_valid_time = Duration::from_secs(20);
    let cm = ConcurrencyManager::new_with_config(
        latest_ts,
        limit_valid_time,
        ActionOnInvalidMaxTs::Error,
        None,
        Duration::ZERO,
        false,
    );

    cm.set_max_ts_limit(TimeStamp::new(4000));

    let new_ts = TimeStamp::new(3000);

    c.bench_function("update_max_ts", |b| {
        b.iter(|| {
            cm.update_max_ts(black_box(new_ts), || format!("benchmark-{}", new_ts))
                .unwrap();
        })
    });
}

criterion_group!(benches, benchmark_update_max_ts);
criterion_main!(benches);