parent
05dbc6a354
commit
649e98b9cc
GPG Key ID:
284488987C31F630
@ -0,0 +1,68 @@
|
|||||||
|
use super::*;
|
||||||
|
use std::{
|
||||||
|
hash::Hash,
|
||||||
|
collections::{HashSet, HashMap},
|
||||||
|
mem,
|
||||||
|
};
|
||||||
|
|
||||||
|
pub trait DedupFullExt
|
||||||
|
{
|
||||||
|
fn dedup_full(&mut self);
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<T> DedupFullExt for Vec<T>
|
||||||
|
where T: Hash + Default + Eq
|
||||||
|
{
|
||||||
|
fn dedup_full(&mut self)
|
||||||
|
{
|
||||||
|
let mut set: HashMap<T, usize> = HashMap::new();
|
||||||
|
for (i,x) in (0..).zip(self.iter_mut())
|
||||||
|
{
|
||||||
|
if !set.contains_key(x) {
|
||||||
|
let us = mem::replace(x, Default::default());
|
||||||
|
set.insert(us, i);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// To preserve order:
|
||||||
|
let mut tmp = Vec::with_capacity(self.len());
|
||||||
|
for item in set.into_iter()
|
||||||
|
{
|
||||||
|
tmp.push(item);
|
||||||
|
}
|
||||||
|
tmp.sort_by(move |a, b| a.1.partial_cmp(&b.1).unwrap());
|
||||||
|
self.truncate(tmp.len());
|
||||||
|
for (d,s) in self.iter_mut().zip(tmp.into_iter())
|
||||||
|
{
|
||||||
|
*d = s.0;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod test
|
||||||
|
{
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn dedup_full()
|
||||||
|
{
|
||||||
|
let mut vec = vec![
|
||||||
|
"hello",
|
||||||
|
"how",
|
||||||
|
"are",
|
||||||
|
"you",
|
||||||
|
"hello",
|
||||||
|
"hello",
|
||||||
|
"today",
|
||||||
|
"today",
|
||||||
|
"you",
|
||||||
|
"how",
|
||||||
|
"hello",
|
||||||
|
];
|
||||||
|
|
||||||
|
vec.dedup_full();
|
||||||
|
|
||||||
|
assert_eq!(vec.len(), 5);
|
||||||
|
assert_eq!(&vec[..], &["hello","how", "are", "you", "today"]);
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,131 @@
|
|||||||
|
use super::*;
|
||||||
|
use std::{
|
||||||
|
fmt,
|
||||||
|
};
|
||||||
|
use lazy_static::lazy_static;
|
||||||
|
|
||||||
|
/// Logging mode
|
||||||
|
#[derive(Debug,Clone,PartialEq,Eq,Hash)]
|
||||||
|
pub enum Mode
|
||||||
|
{
|
||||||
|
/// Output nothing
|
||||||
|
None,
|
||||||
|
/// Only print errors
|
||||||
|
Error,
|
||||||
|
/// Only print warnings
|
||||||
|
Warn,
|
||||||
|
/// Verbose message
|
||||||
|
Verbose,
|
||||||
|
/// Debug messages
|
||||||
|
Debug,
|
||||||
|
}
|
||||||
|
|
||||||
|
#[derive(Debug, Clone, Copy, PartialEq,Eq,Hash)]
|
||||||
|
pub enum Level
|
||||||
|
{
|
||||||
|
Debug,
|
||||||
|
Info,
|
||||||
|
Warning,
|
||||||
|
Error,
|
||||||
|
Fatal,
|
||||||
|
}
|
||||||
|
|
||||||
|
const LEVELS_FOR: &[(Mode, &[Level]); 5] = &[
|
||||||
|
(Mode::None, &[]),
|
||||||
|
(Mode::Error, &[Level::Fatal, Level::Error]),
|
||||||
|
(Mode::Warn, &[Level::Fatal, Level::Error, Level::Warning]),
|
||||||
|
(Mode::Verbose, &[Level::Fatal, Level::Error, Level::Warning, Level::Info]),
|
||||||
|
(Mode::Debug, &[Level::Fatal, Level::Error, Level::Warning, Level::Info, Level::Debug]),
|
||||||
|
];
|
||||||
|
|
||||||
|
impl Mode
|
||||||
|
{
|
||||||
|
/// Can we print this level?
|
||||||
|
#[inline]
|
||||||
|
pub fn level(&self, lv: Level) -> Option<Level>
|
||||||
|
{
|
||||||
|
if lv.mode_ok(self) {
|
||||||
|
Some(lv)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Level
|
||||||
|
{
|
||||||
|
/// Can we print for `mode`?
|
||||||
|
pub fn mode_ok(&self, mode: &Mode) -> bool
|
||||||
|
{
|
||||||
|
use std::collections::{HashMap, HashSet};
|
||||||
|
lazy_static! {
|
||||||
|
static ref MAP: HashMap<&'static Mode, HashSet<&'static Level>> = {
|
||||||
|
let mut m = HashMap::new();
|
||||||
|
for (key, values) in LEVELS_FOR.iter() {
|
||||||
|
let mut set = HashSet::new();
|
||||||
|
for value in values.iter() {
|
||||||
|
set.insert(value);
|
||||||
|
}
|
||||||
|
m.insert(key, set);
|
||||||
|
}
|
||||||
|
m
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(values) = MAP.get(mode) {
|
||||||
|
if values.contains(self) {
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
unreachable!()
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl fmt::Display for Level
|
||||||
|
{
|
||||||
|
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result
|
||||||
|
{
|
||||||
|
match self {
|
||||||
|
Level::Debug => write!(f, "DEBUG"),
|
||||||
|
Level::Info => write!(f, "INFO"),
|
||||||
|
Level::Warning => write!(f, "WARN"),
|
||||||
|
Level::Error => write!(f, "ERROR"),
|
||||||
|
Level::Fatal => write!(f, "FATAL"),
|
||||||
|
_ => write!(f, "(unbound)"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
macro_rules! log {
|
||||||
|
($level:tt, $mode:expr => $format:expr, $($rest:expr),*) => {
|
||||||
|
{
|
||||||
|
|
||||||
|
if let Some(level) = $mode.level($crate::log::Level::$level) {
|
||||||
|
println!("{} [{}]: {}", $crate::log::timestamp(), level, format!($format, $($rest)*));
|
||||||
|
true
|
||||||
|
} else {
|
||||||
|
false
|
||||||
|
};
|
||||||
|
}
|
||||||
|
};
|
||||||
|
($mode:expr => $format:expr, $($rest:expr),*) => {
|
||||||
|
{
|
||||||
|
log!(Info, $mode => $format, $($rest)*);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
}
|
||||||
|
|
||||||
|
pub fn timestamp() -> String
|
||||||
|
{
|
||||||
|
use chrono::{
|
||||||
|
offset::Utc,
|
||||||
|
DateTime,
|
||||||
|
};
|
||||||
|
let time: DateTime<Utc> = std::time::SystemTime::now().into();
|
||||||
|
time.format("%Y.%m.%d %h:%M:%S").to_string()
|
||||||
|
}
|
||||||
|
|
||||||
|
//log!(Level::Debug, cmode => "Hello {}", "hi");
|
||||||
Loading…
Reference in new issue