mysql_database.rs

#![allow(clippy::expect_fun_call)]
use diesel::connection::SimpleConnection;
use diesel::dsl::sql;
use diesel::sql_types::Bool;
use diesel::*;

pub struct Database {
    url: String,
}

impl Database {
    pub fn new(url: &str) -> Self {
        Database { url: url.into() }
    }

    pub fn create(self) -> Self {
        let (database, mysql_url) = self.split_url();
        let mut conn = MysqlConnection::establish(&mysql_url).unwrap();
        diesel::sql_query(format!("CREATE DATABASE `{}`", database))
            .execute(&mut conn)
            .unwrap();
        self
    }

    pub fn exists(&self) -> bool {
        MysqlConnection::establish(&self.url).is_ok()
    }

    pub fn table_exists(&self, table: &str) -> bool {
        select(sql::<Bool>(&format!(
            "EXISTS \
                (SELECT 1 \
                 FROM information_schema.tables \
                 WHERE table_name = '{}'
                 AND table_schema = DATABASE())",
            table
        )))
        .get_result(&mut self.conn())
        .unwrap()
    }

    pub fn conn(&self) -> MysqlConnection {
        MysqlConnection::establish(&self.url)
            .expect(&format!("Failed to open connection to {}", &self.url))
    }

    pub fn execute(&self, command: &str) {
        self.conn()
            .batch_execute(command)
            .expect(&format!("Error executing command {}", command));
    }

    fn split_url(&self) -> (String, String) {
        let mut split: Vec<&str> = self.url.split('/').collect();
        let database = split.pop().unwrap();
        let mysql_url = format!("{}/{}", split.join("/"), "information_schema");
        (database.into(), mysql_url)
    }
}

impl Drop for Database {
    fn drop(&mut self) {
        let (database, mysql_url) = self.split_url();
        let mut conn = try_drop!(
            MysqlConnection::establish(&mysql_url),
            "Couldn't connect to database"
        );
        try_drop!(
            diesel::sql_query(format!("DROP DATABASE IF EXISTS `{}`", database)).execute(&mut conn),
            "Couldn't drop database"
        );
    }
}