butane_core 0.5.1

use super::connmethods::ConnectionMethodWrapper;
use super::*;
use crate::Result;

/// R2D2 support for Butane. Implements [`r2d2::ManageConnection`].
pub struct ConnectionManager {
    spec: ConnectionSpec,
}
impl ConnectionManager {
    pub fn new(spec: ConnectionSpec) -> Self {
        ConnectionManager { spec }
    }
}

impl r2d2::ManageConnection for ConnectionManager {
    type Connection = Connection;
    type Error = crate::Error;

    fn connect(&self) -> Result<Self::Connection> {
        crate::db::connect(&self.spec)
    }

    fn is_valid(&self, conn: &mut Self::Connection) -> Result<()> {
        conn.execute("SELECT 1")
    }

    fn has_broken(&self, conn: &mut Self::Connection) -> bool {
        conn.is_closed()
    }
}

impl ConnectionMethodWrapper for r2d2::PooledConnection<ConnectionManager> {
    type Wrapped = Connection;
    fn wrapped_connection_methods(&self) -> Result<&Connection> {
        Ok(self.deref())
    }
}

connection_method_wrapper!(r2d2::PooledConnection<ConnectionManager>);