use redis; use redis::Commands; use crate::note::now; use crate::note::Note; lazy_static! { static ref REDIS_CLIENT: String = std::env::var("REDIS") .unwrap_or("redis://127.0.0.1/".to_string()) .parse() .unwrap(); } fn get_connection() -> Result { let client = redis::Client::open(REDIS_CLIENT.to_string()).map_err(|_| "Unable to connect to redis")?; client .get_connection() .map_err(|_| "Unable to connect to redis") } pub fn set(id: &String, note: &Note) -> Result<(), &'static str> { let serialized = serde_json::to_string(¬e.clone()).unwrap(); let mut conn = get_connection()?; conn.set(id, serialized) .map_err(|_| "Unable to set note in redis")?; match note.expiration { Some(e) => { let seconds = e - now(); conn.expire(id, seconds as usize) .map_err(|_| "Unable to set expiration on notion")? } None => {} }; Ok(()) } pub fn get(id: &String) -> Result, &'static str> { let mut conn = get_connection()?; let value: Option = conn.get(id).map_err(|_| "Could not load note in redis")?; match value { None => return Ok(None), Some(s) => { let deserialize: Note = serde_json::from_str(&s).unwrap(); return Ok(Some(deserialize)); } } } pub fn del(id: &String) -> Result<(), &'static str> { let mut conn = get_connection()?; conn.del(id).map_err(|_| "Unable to delete note in redis")?; Ok(()) }