myriation/atuin
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix new stats unique count (#772)

Browse source
This commit is contained in:
Conrad Ludgate 2023-03-09 09:47:22 +00:00 committed by GitHub
parent 991461f423
commit 1638cb57cb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
src/command/client/stats.rs
View file

@ -1,4 +1,4 @@
use std::collections::HashMap; use std::collections::{HashMap, HashSet};
use chrono::{prelude::*, Duration}; use chrono::{prelude::*, Duration};
use clap::Parser; use clap::Parser;
@ -24,18 +24,19 @@ pub struct Cmd {
} }
fn compute_stats(history: &[History], count: usize) -> Result<()> { fn compute_stats(history: &[History], count: usize) -> Result<()> {
let mut commands = HashMap::<&str, usize>::new(); let mut commands = HashSet::<&str>::with_capacity(history.len());
let mut prefixes = HashMap::<&str, usize>::with_capacity(history.len());
for i in history { for i in history {
let command = i.command.split_ascii_whitespace().next(); commands.insert(i.command.as_str());
if command.is_none() { let Some(command) = i.command.split_ascii_whitespace().next() else {
continue; continue
} };
*commands.entry(command.unwrap()).or_default() += 1; *prefixes.entry(command).or_default() += 1;
} }
let unique = commands.len(); let unique = commands.len();
let mut top = commands.into_iter().collect::<Vec<_>>(); let mut top = prefixes.into_iter().collect::<Vec<_>>();
top.sort_unstable_by_key(|x| std::cmp::Reverse(x.1)); top.sort_unstable_by_key(|x| std::cmp::Reverse(x.1));
top.truncate(count); top.truncate(count);
if top.is_empty() { if top.is_empty() {