56 lines
1.6 KiB
Rust
56 lines
1.6 KiB
Rust
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<redis::Connection, &'static str> {
|
|
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<Option<Note>, &'static str> {
|
|
let mut conn = get_connection()?;
|
|
let value: Option<String> = 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(())
|
|
}
|