1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36
use std::io; use crate::{ConnectionLike, RedisError}; macro_rules! impl_manage_connection { ($client:ty, $connection:ty) => { impl r2d2::ManageConnection for $client { type Connection = $connection; type Error = RedisError; fn connect(&self) -> Result<Self::Connection, Self::Error> { self.get_connection() } fn is_valid(&self, conn: &mut Self::Connection) -> Result<(), Self::Error> { if conn.check_connection() { Ok(()) } else { Err(RedisError::from(io::Error::from(io::ErrorKind::BrokenPipe))) } } fn has_broken(&self, conn: &mut Self::Connection) -> bool { !conn.is_open() } } }; } impl_manage_connection!(crate::Client, crate::Connection); #[cfg(feature = "cluster")] impl_manage_connection!( crate::cluster::ClusterClient, crate::cluster::ClusterConnection );