dynamo-mocker 1.1.0

Mock LLM scheduler and KV manager for testing
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

// SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
// SPDX-License-Identifier: Apache-2.0

use indicatif::{ProgressBar, ProgressStyle};

pub(super) struct ReplayProgress {
    bar: ProgressBar,
}

impl ReplayProgress {
    pub(super) fn new(total_requests: usize, label: &'static str) -> Self {
        let bar = ProgressBar::new(total_requests as u64);
        bar.set_style(
            ProgressStyle::with_template(
                "[{elapsed_precise}] [{wide_bar:.cyan/blue}] {pos}/{len} ({eta}) {msg}",
            )
            .expect("progress bar template must be valid")
            .progress_chars("#>-"),
        );
        bar.set_message(label);
        Self { bar }
    }

    pub(super) fn inc_completed(&self) {
        self.bar.inc(1);
    }

    pub(super) fn finish(&self) {
        self.bar.finish_and_clear();
    }
}

impl Drop for ReplayProgress {
    fn drop(&mut self) {
        if !self.bar.is_finished() {
            self.bar.finish_and_clear();
        }
    }
}