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.progress_with_metrics(uploaded, self.total_size, &self.operation_name);
40            self.last_update = now;
41            self.last_uploaded = uploaded;
42        }
43    }
44
45    pub fn finish(&self) {
46        self.output.progress_with_metrics(self.total_size, self.total_size, &self.operation_name);
47        
48        let total_elapsed = self.start_time.elapsed();
49        let avg_speed = if total_elapsed.as_secs() > 0 {
50            self.total_size / total_elapsed.as_secs()
51        } else {
52            self.total_size
53        };
54
55        self.output.success(&format!(
56            "{} completed in {} (avg speed: {})",
57            self.operation_name,
58            self.output.format_duration(total_elapsed),
59            self.output.format_size(avg_speed)
60        ));
61    }
62
63    pub fn set_phase(&mut self, phase: &str) {
64        self.operation_name = format!("{} - {}", self.operation_name, phase);
65    }
66
67    pub fn get_elapsed_time(&self) -> Duration {
68        self.start_time.elapsed()
69    }
70
71    pub fn get_estimated_remaining(&self, uploaded: u64) -> Option<Duration> {
72        let elapsed = self.start_time.elapsed();
73        if uploaded > 0 && elapsed.as_secs() > 0 {
74            let speed = uploaded / elapsed.as_secs();
75            if speed > 0 {
76                let remaining_bytes = self.total_size.saturating_sub(uploaded);
77                return Some(Duration::from_secs(remaining_bytes / speed));
78            }
79        }
80        None
81    }
82}
docker_image_pusher/upload/progress.rs

docker_image_pusher/upload/
progress.rs
