Moved cache and messages. Added docs.
This commit is contained in:
258
src/lib.rs
258
src/lib.rs
@ -1,217 +1,123 @@
|
||||
mod counter;
|
||||
mod cache;
|
||||
mod messages;
|
||||
|
||||
use rand::distributions::{Alphanumeric, DistString};
|
||||
use cache::Cache;
|
||||
use messages::{ReceiveMsg, SendMsg, SessionRequest};
|
||||
use std::{
|
||||
collections::HashMap,
|
||||
fmt,
|
||||
sync::mpsc::{channel, Receiver, Sender},
|
||||
sync::mpsc::{channel, Sender},
|
||||
thread::spawn,
|
||||
};
|
||||
use uuid::Uuid;
|
||||
|
||||
pub enum Session {
|
||||
Ok,
|
||||
New(String),
|
||||
}
|
||||
|
||||
struct ValidateSession {
|
||||
id: Option<String>,
|
||||
tx: Sender<Session>,
|
||||
}
|
||||
|
||||
impl ValidateSession {
|
||||
fn new(id: Option<String>, tx: Sender<Session>) -> Self {
|
||||
Self { id: id, tx: tx }
|
||||
}
|
||||
}
|
||||
|
||||
enum SendMsg {
|
||||
ValidateSess(ValidateSession),
|
||||
}
|
||||
|
||||
struct Cache {
|
||||
data: HashMap<String, DataType>,
|
||||
rx: Receiver<SendMsg>,
|
||||
}
|
||||
|
||||
impl Cache {
|
||||
fn new(recv: Receiver<SendMsg>) -> Self {
|
||||
Self {
|
||||
rx: recv,
|
||||
data: HashMap::new(),
|
||||
}
|
||||
}
|
||||
|
||||
fn gen_id(&self) -> String {
|
||||
Alphanumeric.sample_string(&mut rand::thread_rng(), 16)
|
||||
}
|
||||
|
||||
fn listen(&mut self) {
|
||||
loop {
|
||||
match self.rx.recv().unwrap() {
|
||||
SendMsg::ValidateSess(vsess) => {
|
||||
vsess.tx.send(self.validate_session(vsess.id)).unwrap()
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn validate_session(&mut self, sess: Option<String>) -> Session {
|
||||
let session: Session;
|
||||
if sess.is_some_and(|sess| true) {// self.data.contains(&sess)) {
|
||||
session = Session::Ok;
|
||||
} else {
|
||||
let id = self.gen_id();
|
||||
// `self.data.push(id.clone());
|
||||
session = Session::New(id);
|
||||
}
|
||||
session
|
||||
}
|
||||
}
|
||||
|
||||
/// Application connection to the database
|
||||
#[derive(Clone)]
|
||||
pub struct MoreThanText {
|
||||
id: Option<Uuid>,
|
||||
tx: Sender<SendMsg>,
|
||||
}
|
||||
|
||||
impl MoreThanText {
|
||||
/// Create a MoreThanText database.
|
||||
///
|
||||
/// Example:
|
||||
///
|
||||
/// ```
|
||||
/// use morethantext::MoreThanText;
|
||||
///
|
||||
/// MoreThanText::new();
|
||||
/// ```
|
||||
pub fn new() -> Self {
|
||||
let (tx, rx) = channel();
|
||||
spawn(move || {
|
||||
let mut cache = Cache::new(rx);
|
||||
cache.listen();
|
||||
});
|
||||
Self { tx: tx }
|
||||
Self { id: None, tx: tx }
|
||||
}
|
||||
|
||||
pub fn get_session(&self, id: Option<String>) -> Session {
|
||||
/// Opens an existing or new session with the database.
|
||||
///
|
||||
/// If the string is None, incorrect, or expired,
|
||||
/// a new session will be created.
|
||||
///
|
||||
/// Example:
|
||||
///
|
||||
/// ```
|
||||
/// use morethantext::MoreThanText;
|
||||
///
|
||||
/// let mut mtt = MoreThanText::new();
|
||||
/// mtt.open_session(None);
|
||||
/// mtt.open_session(Some("7b1ff340-7dfa-4f29-b144-601384e54423".to_string()));
|
||||
/// ```
|
||||
pub fn open_session(&mut self, id: Option<String>) {
|
||||
let (tx, rx) = channel();
|
||||
self.tx
|
||||
.send(SendMsg::ValidateSess(ValidateSession::new(id, tx)))
|
||||
.unwrap();
|
||||
rx.recv().unwrap()
|
||||
let request = SessionRequest {
|
||||
id: id,
|
||||
tx: tx,
|
||||
};
|
||||
self.tx.send(SendMsg::OpenSession(request)).unwrap();
|
||||
match rx.recv().unwrap() {
|
||||
ReceiveMsg::Session(sid) => self.id = Some(sid),
|
||||
}
|
||||
}
|
||||
|
||||
/// Get the session id
|
||||
///
|
||||
/// Example:
|
||||
///
|
||||
/// ```
|
||||
/// use morethantext::MoreThanText;
|
||||
///
|
||||
/// let mut mtt = MoreThanText::new();
|
||||
/// mtt.get_id();
|
||||
/// ```
|
||||
pub fn get_id(&self) -> String {
|
||||
match self.id {
|
||||
Some(sid) => sid.to_string(),
|
||||
None => "none".to_string(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod client {
|
||||
mod mtt_client {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn session_ids_are_unique() {
|
||||
let conn = MoreThanText::new();
|
||||
let mut ids: Vec<String> = Vec::new();
|
||||
fn uniques_ids() {
|
||||
let mut mtt = MoreThanText::new();
|
||||
let mut ids: Vec<Uuid> = Vec::new();
|
||||
for _ in 1..10 {
|
||||
match conn.get_session(None) {
|
||||
Session::New(id) => {
|
||||
if ids.contains(&id) {
|
||||
assert!(false, "{} is a duplicate id", id);
|
||||
}
|
||||
ids.push(id)
|
||||
}
|
||||
Session::Ok => assert!(false, "Should have returned a new id."),
|
||||
mtt.open_session(None);
|
||||
let id = mtt.id.clone().unwrap();
|
||||
if ids.contains(&id) {
|
||||
assert!(false, "{} is a duplicate id", id);
|
||||
}
|
||||
ids.push(id);
|
||||
}
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn existing_ids_return_ok() {
|
||||
let conn = MoreThanText::new();
|
||||
let sid: String;
|
||||
match conn.get_session(None) {
|
||||
Session::New(id) => sid = id,
|
||||
Session::Ok => unreachable!(),
|
||||
}
|
||||
match conn.get_session(Some(sid.clone())) {
|
||||
Session::New(_) => assert!(false, "should not create a new session"),
|
||||
Session::Ok => {}
|
||||
}
|
||||
fn existing_ids_are_reused() {
|
||||
let mut mtt = MoreThanText::new();
|
||||
mtt.open_session(None);
|
||||
let holder = mtt.id.clone().unwrap().to_string();
|
||||
mtt.open_session(Some(holder.clone()));
|
||||
assert_eq!(mtt.id.clone().unwrap().to_string(), holder);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn bad_ids_get_new_session() {
|
||||
let conn = MoreThanText::new();
|
||||
let sid = "bad id";
|
||||
match conn.get_session(Some(sid.to_string())) {
|
||||
Session::New(id) => assert_ne!(sid, id, "do not reuse original id"),
|
||||
Session::Ok => assert!(false, "shouuld generate a new id"),
|
||||
}
|
||||
fn bad_ids_generate_new_ones() {
|
||||
let mut mtt = MoreThanText::new();
|
||||
mtt.open_session(Some("bad test string".to_string()));
|
||||
assert!(mtt.id.is_some());
|
||||
}
|
||||
}
|
||||
|
||||
enum Field {
|
||||
StaticString(String),
|
||||
}
|
||||
|
||||
impl fmt::Display for Field {
|
||||
fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
|
||||
match self {
|
||||
Field::StaticString(data) => write!(f, "{}", data),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
struct Record {
|
||||
data: HashMap<String, Field>,
|
||||
}
|
||||
|
||||
impl Record {
|
||||
fn new(data: HashMap<String, Field>) -> Self {
|
||||
Self {
|
||||
data: data,
|
||||
}
|
||||
}
|
||||
|
||||
fn get(&self, fieldname: &str) -> &Field {
|
||||
match self.data.get(fieldname) {
|
||||
Some(field) => field,
|
||||
None => unreachable!(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod records {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn create_record() {
|
||||
let input = HashMap::from([
|
||||
("one".to_string(), Field::StaticString("1".to_string())),
|
||||
("two".to_string(), Field::StaticString("2".to_string())),
|
||||
("three".to_string(), Field::StaticString("3".to_string())),
|
||||
]);
|
||||
let rec = Record::new(input);
|
||||
assert_eq!(rec.get("one").to_string(), "1");
|
||||
assert_eq!(rec.get("two").to_string(), "2");
|
||||
assert_eq!(rec.get("three").to_string(), "3");
|
||||
fn incorrect_ids_generate_new_ones() {
|
||||
let mut mtt = MoreThanText::new();
|
||||
let holder = Uuid::new_v4();
|
||||
mtt.open_session(Some(holder.clone().to_string()));
|
||||
assert_ne!(mtt.id, Some(holder));
|
||||
}
|
||||
}
|
||||
|
||||
struct Column;
|
||||
|
||||
struct Table {
|
||||
columns: HashMap<String, Column>,
|
||||
}
|
||||
|
||||
impl Table {
|
||||
fn new() -> Self {
|
||||
Self {
|
||||
columns: HashMap::new(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tables {
|
||||
use super::*;
|
||||
|
||||
#[test]
|
||||
fn create_table() {
|
||||
let tbl = Table::new();
|
||||
}
|
||||
}
|
||||
|
||||
enum DataType {
|
||||
Table(Table),
|
||||
Record(Record),
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user