br.com.codecode.dryutil.factory.JDBC Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dry-util Show documentation
Dont Repeat Yourself! - Utilities
The newest version!
package br.com.codecode.dryutil.factory;

import java.io.IOException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Properties;

import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;
import javax.sql.DataSource;

import br.com.codecode.dryutil.LoadProperties;
import br.com.codecode.dryutil.factory.exception.ConnectionException;

public final class JDBC {

    private Connection connection;

    private Context ctx = null;

    private String dbHost;

    private String dbLogin;

    private String dbName;

    private String dbPassword;

    private String dbPort;

    private DataSource ds = null;

    private String jndiName;

    private Properties properties;

    private String url = "jdbc:mysql://%s:%s/%s";

    private JDBC() {

	doLoadProperties();

	try {

	    doLoadJNDIConnection();

	    System.out.println("Connected with JNDI !");

	} catch (ConnectionException e) {

	    System.err.println("Failed to load JNDI, trying to load Driver Manager \n " + e);

	    try {

		doLoadDriverManager();

		System.out.println("Connected with Driver Manager !");

	    } catch (ConnectionException ex) {

		throw new RuntimeException("Failed to load Driver Manager \n" + ex);
	    }

	}
    }

    public void doCloseConnection() {

	try {
	    if (!connection.isClosed()) {
		connection.close();
		System.out.println("Connection was closed");
	    } else {
		System.out.println("Connection already closed");
	    }
	} catch (SQLException ex) {
	    System.err.println(ex);
	}
    }

    private void doLoadDriverManager() throws ConnectionException {

	System.out.println("[doLoadDriverManager]");

	url = String.format(url, dbHost, dbPort, dbName);

	try {

	    connection = (Connection) DriverManager.getConnection(url, dbLogin, dbPassword);

	} catch (SQLException ex) {

	    throw new ConnectionException("Could not load Driver Manager, check the properties of the connection. : " , ex);
	}

    }

    private void doLoadJNDIConnection() throws ConnectionException {

	System.out.println("[doLoadJNDIConnection]");

	try {
	    ctx = new InitialContext();

	    ds = (DataSource) ctx.lookup(jndiName);

	    connection = ds.getConnection();

	} catch (NamingException | SQLException ex) {

	    throw new ConnectionException("Could not load Driver Manager, check the properties of the connection. :", ex);

	}

    }

    private void doLoadProperties() {

	try {

	    properties = LoadProperties.readProperties(ClassLoader.class.getResourceAsStream("/META-INF/properties.properties"));

	    dbHost = properties.getProperty("dbHost");

	    dbPort = properties.getProperty("dbPort");

	    dbName = properties.getProperty("dbName");

	    dbLogin = properties.getProperty("dbLogin");

	    dbPassword = properties.getProperty("dbPassword");

	    jndiName = properties.getProperty("jndiName");

	} catch (IOException e) {

	    throw new RuntimeException("File properties Not Found ", e);
	}

    }

    public Connection getConnection() {

	return connection;
    }

}