Added random id generator.
This commit is contained in:
parent
c70c006abd
commit
de93ee1f2c
@ -1,9 +1,78 @@
|
|||||||
use super::{ErrorCode, FromCache, MTTError, Store, ToCache, ENTRY};
|
use super::{ErrorCode, FromCache, MTTError, Store, ToCache, ENTRY};
|
||||||
use async_std::{channel::Receiver, path::PathBuf};
|
use async_std::{channel::Receiver, path::PathBuf};
|
||||||
use std::collections::HashMap;
|
use rand::{distributions::Alphanumeric, thread_rng, Rng};
|
||||||
|
use std::{collections::{HashMap, VecDeque}, iter::Iterator};
|
||||||
|
|
||||||
|
struct IDGenerator {
|
||||||
|
ids: Option<VecDeque<String>>,
|
||||||
|
}
|
||||||
|
|
||||||
|
impl IDGenerator {
|
||||||
|
fn new() -> Self {
|
||||||
|
Self {
|
||||||
|
ids: None,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
fn with_ids<T, D>(ids: T) -> Self where T: Into<Vec<D>>, D: Into<String> {
|
||||||
|
let id_list = ids.into();
|
||||||
|
let mut data = VecDeque::new();
|
||||||
|
for id in id_list {
|
||||||
|
data.push_back(id.into());
|
||||||
|
}
|
||||||
|
Self {
|
||||||
|
ids: Some(data),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl Iterator for IDGenerator {
|
||||||
|
type Item = String;
|
||||||
|
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
match &self.ids {
|
||||||
|
Some(id_list) => {
|
||||||
|
let mut ids = id_list.clone();
|
||||||
|
let output = match ids.pop_front() {
|
||||||
|
Some(id) => Some(id.to_string()),
|
||||||
|
None => None,
|
||||||
|
};
|
||||||
|
self.ids = Some(ids);
|
||||||
|
output
|
||||||
|
}
|
||||||
|
None => Some(thread_rng().sample_iter(&Alphanumeric).take(64).collect()),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod genid {
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn unique_ids() {
|
||||||
|
let mut gen = IDGenerator::new();
|
||||||
|
let mut output: Vec<String> = Vec::new();
|
||||||
|
for _ in 0..10 {
|
||||||
|
let id = gen.next().unwrap();
|
||||||
|
assert!(!output.contains(&id), "{} found in {:?}", id, output);
|
||||||
|
output.push(id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn controlled_ids() {
|
||||||
|
let ids = ["one", "two", "three"];
|
||||||
|
let mut gen = IDGenerator::with_ids(ids.clone());
|
||||||
|
for id in ids {
|
||||||
|
assert_eq!(id, gen.next().unwrap());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub struct Cache {
|
pub struct Cache {
|
||||||
data: HashMap<String, Store>,
|
data: HashMap<String, Store>,
|
||||||
|
ids: IDGenerator,
|
||||||
}
|
}
|
||||||
|
|
||||||
impl Cache {
|
impl Cache {
|
||||||
@ -13,7 +82,10 @@ impl Cache {
|
|||||||
{
|
{
|
||||||
let mut data = HashMap::new();
|
let mut data = HashMap::new();
|
||||||
data.insert(ENTRY.to_string(), Store::new());
|
data.insert(ENTRY.to_string(), Store::new());
|
||||||
Self { data: data }
|
Self {
|
||||||
|
data: data,
|
||||||
|
ids: IDGenerator::new(),
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub async fn listen(&mut self, listener: Receiver<ToCache>) {
|
pub async fn listen(&mut self, listener: Receiver<ToCache>) {
|
||||||
|
Loading…
Reference in New Issue
Block a user