YGGverse/pulsarss
1
0
Fork 0
mirror of https://github.com/YGGverse/pulsarss.git synced 2026-03-31 17:15:29 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update limit argument data type

Browse source
This commit is contained in:
yggverse 2025-02-12 20:12:27 +02:00
parent a22fa6449a
commit f3045ad5a4
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
src/argument.rs
View file

@ -8,8 +8,8 @@ pub struct Argument {
pub index: Option<String>, pub index: Option<String>,
/// Limit channel items (unlimited by default) /// Limit channel items (unlimited by default)
#[arg(short, long, default_value_t = 0)] #[arg(short, long)]
pub limit: usize, pub limit: Option<usize>,
/// Show output (`dw` by default) /// Show output (`dw` by default)
#[arg(short, long, default_value_t = String::from("dw"))] #[arg(short, long, default_value_t = String::from("dw"))]

4
src/main.rs
View file

@ -43,7 +43,7 @@ fn crawl(argument: &Argument, output: &Output) -> Result<(), Box<dyn Error>> {
.items() .items()
.iter() .iter()
{ {
if argument.limit > 0 && total >= argument.limit { if argument.limit.is_some_and(|limit| total >= limit) {
break; break;
} }
@ -108,7 +108,7 @@ fn crawl(argument: &Argument, output: &Output) -> Result<(), Box<dyn Error>> {
let mut files: Vec<_> = read_dir(&path)?.filter_map(Result::ok).collect(); let mut files: Vec<_> = read_dir(&path)?.filter_map(Result::ok).collect();
files.sort_by_key(|f| f.file_name()); files.sort_by_key(|f| f.file_name());
let mut data = Vec::with_capacity(argument.limit); let mut data = Vec::with_capacity(argument.limit.unwrap_or_default());
let mut index = File::create(&index_filename)?; let mut index = File::create(&index_filename)?;
let mut total = 0; let mut total = 0;