#[cfg(test)]
#[path = "./connection_test.rs"]
mod connection_test;
use crate::types::{RedisEmptyResult, RedisError, RedisResult};
pub(crate) struct Connection {
connection: Option<redis::Connection>,
}
fn open_connection(connection: &mut Connection, client: &redis::Client) -> RedisEmptyResult {
let output: RedisEmptyResult;
if !connection.is_connection_open() {
output = match client.get_connection() {
Ok(redis_connection) => {
connection.connection = Some(redis_connection);
Ok(())
}
Err(error) => Err(RedisError::RedisError(error)),
}
} else {
output = Ok(());
}
output
}
impl Connection {
pub(crate) fn is_connection_open(self: &mut Connection) -> bool {
let open;
match self.connection {
Some(ref mut redis_connection) => {
let result: redis::RedisResult<()> = redis::cmd("PING").query(redis_connection);
open = result.is_ok();
}
None => open = false,
}
open
}
pub(crate) fn get_redis_connection(
self: &mut Connection,
client: &redis::Client,
) -> RedisResult<&mut redis::Connection> {
match open_connection(self, client) {
Err(error) => Err(error),
_ => match self.connection {
Some(ref mut redis_connection) => Ok(redis_connection),
None => Err(RedisError::Description("Redis connection not available.")),
},
}
}
}
pub(crate) fn create() -> Connection {
Connection { connection: None }
}