Ceeql.java

package org.mrcsparker.ceeql;

import com.codahale.metrics.MetricRegistry;
import com.zaxxer.hikari.HikariConfig;
import com.zaxxer.hikari.HikariDataSource;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.skife.jdbi.v2.*;
import org.skife.jdbi.v2.logging.Log4JLog;

import javax.sql.DataSource;
import java.util.Map;

public class Ceeql implements AutoCloseable {

    private final static Logger log = LogManager.getLogger(Ceeql.class);

    private final String driverName;
    private final String url;
    private final String username;
    private final String password;

    private boolean isConnected;
    private DBI dbi;

    private final MetricRegistry metricRegistry;

    public Ceeql(String driverName, String url, String username, String password, MetricRegistry metricRegistry) {
        this.metricRegistry = metricRegistry;
        this.driverName = driverName;
        this.url = url;
        this.username = username;
        this.password = password;
        this.isConnected = false;

        connectToDatabase();
    }

    public Ceeql(String driverName, String url, String username, String password) {
        this(driverName, url, username, password, null);
    }

    public boolean isConnected() {
        return isConnected;
    }

    public Boolean query(String query, Object ... args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            handle.getHandle().execute(query, args);
            return true;
        } catch (Exception e) {
            e.printStackTrace();
            return false;
        }
    }

    public Handle getDbiHandle() {
        return new CeeqlHandle(dbi).getHandle();
    }

    public String selectOne(String sql, Map<String, String> args) {
        return selectOneJson(sql, args);
    }

    public String selectOneJson(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlSelect(handle.getHandle(), sql, args).first();
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String selectOneCsv(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlSelect(handle.getHandle(), sql, args).first(CeeqlOutputType.CSV);
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String selectOneXml(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlSelect(handle.getHandle(), sql, args).first(CeeqlOutputType.XML);
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String select(String sql, Map<String, String> args) {
        return selectJson(sql, args);
    }

    public String selectJson(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlSelect(handle.getHandle(), sql, args).all();
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String selectCsv(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlSelect(handle.getHandle(), sql, args).all(CeeqlOutputType.CSV);
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String selectXml(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlSelect(handle.getHandle(), sql, args).all(CeeqlOutputType.XML);
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String insert(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlInsert(handle.getHandle(), sql, args).exec();
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String batch(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlBatch(handle.getHandle(), sql, args).exec();
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String update(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlUpdate(handle.getHandle(), sql, args).exec();
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    public String delete(String sql, Map<String, String> args) {
        try(CeeqlHandle handle = new CeeqlHandle(dbi)) {
            return new CeeqlDelete(handle.getHandle(), sql, args).exec();
        } catch (Exception e) {
            return CeeqlError.error(e);
        }
    }

    @Override
    public void close() {
        this.isConnected = false;
    }

    private String connectToDatabase() {
        try {
            log.debug("Connecting with driver " + driverName + " to " + url);

            Class clazz = Class.forName(driverName);
            log.debug(clazz);

            HikariConfig config = new HikariConfig();
            config.setJdbcUrl(url);
            config.setUsername(username);
            config.setPassword(password);
            config.addDataSourceProperty("dataSourceClassName", driverName);
            config.setPoolName("HikariPool" + "." + url + "." + username);

            if(metricRegistry != null) {
                config.setMetricRegistry(metricRegistry);
            }

            DataSource ds = new HikariDataSource(config);

            this.dbi = new DBI(ds);
            this.dbi.setSQLLog(new Log4JLog());

            this.isConnected = true;
            return CeeqlMessage.message("Connected");
        } catch (Exception e) {
            this.isConnected = false;
            return CeeqlError.errorType(e.getClass().getSimpleName(), e.getMessage());
        }
    }

    public String reconnect() {
        return connectToDatabase();
    }
}