jdbctask.java

DBExplorer 强烈推荐的一个JAVA项目 数据导出功能 支持的格式HTML PDF XLS等 支持的数据库有MYSQL ORACLE MSSQLSERVER等

package cn.com.qimingx.utils.sql;

import java.sql.*;
import java.util.*;
import java.util.logging.*;


/**
 * ִ�� JDBC ����ij�����
 * @author Unmi
 */

public abstract class JDBCTask {

	private static Logger log = Logger.getLogger(JDBCTask.class.getName());
	static{
		log.setLevel(Level.WARNING);
	}

    /**
     * Autocommit flag. Default value is false
     */
    private boolean autocommit = false;

    /**
     * DB driver.
     */
    private String driver = null;

    /**
     * DB url.
     */
    private String url = null;

    /**
     * User name.
     */
    private String userId = null;

    /**
     * Password
     */
    private String password = null;

    /**
     * Class name of the JDBC driver; required.
     * @param driver The driver to set
     */
    public void setDriver(String driver) {
        this.driver = driver.trim();
    }

    /**
     * Sets the database connection URL; required.
     * @param url The url to set
     */
    public void setUrl(String url) {
        this.url = url;
    }

    /**
     * Sets the password; required.
     * @param password The password to set
     */
    public void setPassword(String password) {
        this.password = password;
    }

    /**
     * Auto commit flag for database connection;
     * optional, default false.
     * @param autocommit The autocommit to set
     */
    public void setAutocommit(boolean autocommit) {
        this.autocommit = autocommit;
    }


    /**
     * Creates a new Connection as using the driver, url, userid and password
     * specified.
     *
     * The calling method is responsible for closing the connection.
     *
     * @return Connection the newly created connection.
     * @throws Exception if the UserId/Password/Url is not set or there
     * is no suitable driver or the driver fails to load.
     */
    protected Connection getConnection() throws Exception {
        if (userId == null) {
            throw new Exception("UserId attribute must be set!");
        }
        if (password == null) {
            throw new Exception("Password attribute must be set!");
        }
        if (url == null) {
            throw new Exception("Url attribute must be set!");
        }
        try {

            log.info("connecting to " + getUrl());
            Properties info = new Properties();
            info.put("user", getUserId());
            info.put("password", getPassword());
            Connection conn = getDriver().connect(getUrl(), info);

            if (conn == null) {
                // Driver doesn't understand the URL
                throw new SQLException("No suitable Driver for " + url);
            }

            conn.setAutoCommit(autocommit);
            return conn;
        } catch (SQLException e) {
            throw new Exception(e);
        }

    }

    /**
     * Gets an instance of the required driver.
     * @return Driver
     * @throws Exception
     */
    private Driver getDriver() throws Exception {
        if (driver == null) {
            throw new Exception("Driver attribute must be set!");
        }
        Driver driverInstance = null;
        try {
        	driverInstance= (Driver)Class.forName(driver).newInstance();
        } catch (ClassNotFoundException e) {
            throw new Exception(
                    "Class Not Found: JDBC driver " + driver + " could not be loaded",
                    e);
        } catch (IllegalAccessException e) {
            throw new Exception(
                    "Illegal Access: JDBC driver " + driver + " could not be loaded",
                    e);
        } catch (InstantiationException e) {
            throw new Exception(
                    "Instantiation Exception: JDBC driver " + driver + " could not be loaded",
                    e);
        }
        return driverInstance;
    }

    /**
     * Gets the autocommit.
     * @return Returns a boolean
     */
    public boolean isAutocommit() {
        return autocommit;
    }

    /**
     * Gets the url.
     * @return Returns a String
     */
    public String getUrl() {
        return url;
    }

    /**
     * Gets the userId.
     * @return Returns a String
     */
    public String getUserId() {
        return userId;
    }

    /**
     * Set the user name for the connection; required.
     * @param userId The userId to set
     */
    public void setUserid(String userId) {
        this.userId = userId;
    }

    /**
     * Gets the password.
     * @return Returns a String
     */
    public String getPassword() {
        return password;
    }
}