replaced only_the_best option with max_mirrors_to_output

Author: westandskif

CHANGELOG.md

@@ -1,3 +1,8 @@
+# 0.22.0 (2025-08-31)
+
+- added option to limit number of mirrors to output [#78 by catap](https://github.com/westandskif/rate-mirrors/pull/78)
+
+
 # 0.21.0 (2025-08-29)
 
 - changed duration format in debug output [#76 by damentz](https://github.com/westandskif/rate-mirrors/pull/76)

Cargo.lock

@@ -1,6 +1,6 @@
 [package]
 name = "rate_mirrors"
-version = "0.21.0"
+version = "0.22.0"
 authors = ["Nikita Almakov <nikita.almakov@gmail.com>"]
 edition = "2024"
 description = "Everyday-use client-side map-aware mirror ranking tool (Arch Linux; Manjaro; custom ones)"

Cargo.toml

@@ -251,9 +251,9 @@ pub struct Config {
     )]
     pub top_mirrors_number_to_retest: usize,
 
-    /// Only one, the best mirror
-    #[arg(env = "RATE_MIRRORS_ONLY_THE_BEST", long)]
-    pub only_the_best: bool,
+    /// Max number of mirrors to output
+    #[arg(env = "RATE_MIRRORS_MAX_MIRRORS_TO_OUTPUT", long)]
+    pub max_mirrors_to_output: Option<usize>,
 
     /// Filename to save the output to in case of success
     #[arg(env = "RATE_MIRRORS_SAVE", long = "save", verbatim_doc_comment)]

src/config.rs

@@ -9,7 +9,7 @@ mod target_configs;
 mod targets;
 
 use crate::config::{AppError, Config, FetchMirrors};
-use crate::speed_test::{test_speed_by_countries, SpeedTestResults};
+use crate::speed_test::{test_speed_by_countries, SpeedTestResult, SpeedTestResults};
 use chrono::prelude::*;
 use clap::Parser;
 use config::LogFormatter;
@@ -93,10 +93,10 @@ impl<'a, T: LogFormatter> OutputSink<'a, T> {
 
 fn main() -> Result<(), AppError> {
     let config = Arc::new(Config::parse());
-    let only_the_best = config.only_the_best;
     if !config.allow_root && Uid::effective().is_root() {
         return Err(AppError::Root);
     }
+    let max_mirrors_to_output = config.max_mirrors_to_output.clone();
 
     let ref formatter = Arc::clone(&config).target;
     let mut output = OutputSink::new(
@@ -158,12 +158,13 @@ fn main() -> Result<(), AppError> {
 
         output.display_comment(format!("FINISHED AT: {}", Local::now()));
 
-        if only_the_best {
-                output.display_mirror(&results.first().unwrap().item);
-        } else {
-            for result in results.into_iter() {
-                output.display_mirror(&result.item);
-            }
+        let it: Box<dyn Iterator<Item = SpeedTestResult>> = match max_mirrors_to_output {
+            Some(n) => Box::new(results.into_iter().take(n)),
+            None => Box::new(results.into_iter()),
+        };
+
+        for result in it {
+            output.display_mirror(&result.item);
         }
     }