docker_image_pusher/upload/
progress.rs

1//! Progress tracking for uploads
2
3use crate::output::OutputManager;
4use std::time::{Duration, Instant};
5
6#[derive(Clone)]
7pub struct ProgressTracker {
8    total_size: u64,
9    start_time: Instant,
10    last_update: Instant,
11    last_uploaded: u64,
12    output: OutputManager,
13    operation_name: String,
14}
15
16impl ProgressTracker {
17    pub fn new(total_size: u64, output: OutputManager, operation_name: String) -> Self {
18        Self {
19            total_size,
20            start_time: Instant::now(),
21            last_update: Instant::now(),
22            last_uploaded: 0,
23            output,
24            operation_name,
25        }
26    }
27
28    pub fn update(&mut self, uploaded: u64) {
29        let now = Instant::now();
30        let elapsed_since_last = now.duration_since(self.last_update);
31
32        // Update progress every 5 seconds or 10MB or every 5% of total
33        let size_threshold = std::cmp::min(10 * 1024 * 1024, self.total_size / 20); // 10MB or 5% of total
34
35        if elapsed_since_last >= Duration::from_secs(5)
36            || uploaded - self.last_uploaded >= size_threshold
37            || uploaded == self.total_size
38        {
39            self.output
40                .progress_with_metrics(uploaded, self.total_size, &self.operation_name);
41            self.last_update = now;
42            self.last_uploaded = uploaded;
43        }
44    }
45
46    pub fn finish(&self) {
47        self.output
48            .progress_with_metrics(self.total_size, self.total_size, &self.operation_name);
49
50        let total_elapsed = self.start_time.elapsed();
51        let avg_speed = if total_elapsed.as_secs() > 0 {
52            self.total_size / total_elapsed.as_secs()
53        } else {
54            self.total_size
55        };
56
57        self.output.success(&format!(
58            "{} completed in {} (avg speed: {})",
59            self.operation_name,
60            self.output.format_duration(total_elapsed),
61            self.output.format_size(avg_speed)
62        ));
63    }
64
65    pub fn set_phase(&mut self, phase: &str) {
66        self.operation_name = format!("{} - {}", self.operation_name, phase);
67    }
68
69    pub fn get_elapsed_time(&self) -> Duration {
70        self.start_time.elapsed()
71    }
72
73    pub fn get_estimated_remaining(&self, uploaded: u64) -> Option<Duration> {
74        let elapsed = self.start_time.elapsed();
75        if uploaded > 0 && elapsed.as_secs() > 0 {
76            let speed = uploaded / elapsed.as_secs();
77            if speed > 0 {
78                let remaining_bytes = self.total_size.saturating_sub(uploaded);
79                return Some(Duration::from_secs(remaining_bytes / speed));
80            }
81        }
82        None
83    }
84}
docker_image_pusher/upload/progress.rs

docker_image_pusher/upload/
progress.rs
