Add builder-like with_finish method to ProgressBarIter. (#548)

Co-authored-by: Matthew Chan <>

Author: matthewhchan

src/iter.rs

@@ -12,6 +12,7 @@ use std::time::Duration;
 use tokio::io::{ReadBuf, SeekFrom};
 
 use crate::progress_bar::ProgressBar;
+use crate::state::ProgressFinish;
 use crate::style::ProgressStyle;
 
 /// Wraps an iterator to display its progress.
@@ -103,6 +104,14 @@ impl<T> ProgressBarIter<T> {
         self.progress = self.progress.with_elapsed(elapsed);
         self
     }
+
+    /// Builder-like function for setting underlying progress bar's finish behavior.
+    ///
+    /// See [ProgressBar::with_finish].
+    pub fn with_finish(mut self, finish: ProgressFinish) -> Self {
+        self.progress = self.progress.with_finish(finish);
+        self
+    }
 }
 
 impl<S, T: Iterator<Item = S>> Iterator for ProgressBarIter<T> {