欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

用于JDBC操作数据库的公共类 博客分类: Database JDBCSQLDBDatabase数据库 

程序员文章站 2024-03-15 08:00:41
...
   哈哈,再发一个用于JDBC操作数据库的公共类,这也是我在很久以前写的,其中的RWProperties类在我的“Java基础”分类中有一篇《用于读写Properties属性配制文件的通用类》的文章中就有,希望能对大家有些帮助,以后我还会发更多的原创文章上来,也希望大家能提出一些宝贵的建议和想法,让我们大家在这里共同成长!

修改日志(20111010)
   在写完该工具之后的日子里对该工具进行过很多迭代式的修改,这里今年修改的最新的版本,更加通用、更加灵活,同时支持ICO、批量操作和手动提交事务,并去除了对RWProperties.java的依赖。

/* 
 * @(#)CommonSql.java  2011-9-5 
 *  
 * Copyright 2011 Bianjing,All rights reserved. 
 */  

import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.List;

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

/**
 * 用于JDBC操作数据库的共通类
 * 
 * @author Bianjing
 * @version 1.0.0 2011-9-5
 */
public class CommonSql {
    /** 数据源 */
    private DataSource dataSource;

    /** 数据库连接对象 */
    private Connection connection;

    /** 数据库操作对象 */
    private PreparedStatement ps;

    /** 数据库操作对象 */
    private Statement statement;

    /** 返回的数据结果集对象 */
    private ResultSet rs;

    /** 是否自动提交事务,默认为true,如果该值为false则需要手动提交事务 */
    private boolean autoCommit = true;

    /** 数据库连接是否已经打开 */
    private boolean openConnection;

    /** JNDI名称 */
    private String jndiName;

    /** 数据库驱动 */
    private String driver;

    /** 数据库访问地址 */
    private String url;

    /** 用户名 */
    private String user;

    /** 密码 */
    private String pwd;

    public CommonSql() {
        
    }

    public CommonSql(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    public CommonSql(String jndiName) {
        this.jndiName = jndiName;
    }

    public CommonSql(String driver, String url, String user, String pwd) {
        this.driver = driver;
        this.url = url;
        this.user = user;
        this.pwd = pwd;
    }

    /**
     * 打开数据库连接并创建数据库连接对象<br/>
     * 支持通过ICO注入数据源、数据库驱动、数据库驱动、JNDI名称、数据库访问地址和用户名、密码
     * 
     * @return boolean true:连接成功,false:连接失败
     */
    public boolean openConnection() {
        /**
         * 通过数据源来获取数据库连接对象
         */
        if (dataSource != null) {
            try {
                connection = dataSource.getConnection();
                // 数据库连接已经打开
                openConnection = true;
            } catch (SQLException e) {
                closeAll();
                // 所有的"System.out.println"都可以替换为"logger.error"
                System.out.println("从数据源获取数据库连接失败!");
                throw new RuntimeException(e);
            }

            return openConnection;
        }
        /**
         * 通过JNDI来获取数据库连接对象
         */
        if (jndiName != null) {
            try {
                Context initContext = new InitialContext();
                dataSource = (DataSource) initContext.lookup(jndiName);
                connection = dataSource.getConnection();
                // 数据库连接已经打开
                openConnection = true;
            } catch (Exception e) {
                closeAll();
                System.out.println("从JNDI获取数据库连接失败!");
                throw new RuntimeException(e);
            }

            return openConnection;
        }
        /**
         * 通过数据库驱动、数据库访问地址、用户名、密码来获取数据库连接对象
         */
        try {
            Class.forName(driver);
            connection = DriverManager.getConnection(url, user, pwd);
            // 数据库连接已经打开
            openConnection = true;
        } catch (Exception e) {
            closeAll();
            System.out.println("数据库连接失败!");
            throw new RuntimeException(e);
        }

        return openConnection;
    }

    /**
     * 执行数据库的更新操作
     * 
     * @param sql
     *            要执行的SQL语句
     * @return boolean true:执行成功,false:执行失败
     */
    public boolean execUpdate(String sql, Object... args) {
        boolean isPassed = false;
        // 判断连接数据库是否成功
        if (openConnection) {
            try {
                ps = connection.prepareStatement(sql);
                // 设置参数
                if (args != null && args.length > 0) {
                    for (int i = 0; i < args.length; i++) {
                        ps.setObject(i + 1, args[i]);
                    }
                }
                ps.executeUpdate();

                isPassed = true;
            } catch (SQLException e) {
                try {
                    if (autoCommit) {
                        connection.rollback();
                    }
                } catch (SQLException e1) {
                    throw new RuntimeException(e1);
                }
                System.out.println("SQL:" + sql);
                throw new RuntimeException(e);
            } finally {
                if (autoCommit) {
                    closeAll();
                }
            }
        } else {
            System.out.println("数据库连接对象没有打开!");
        }

        return isPassed;
    }
    
    /**
     * 执行数据库的更新操作
     * 
     * @param sql
     *            要执行的SQL语句
     * @return boolean true:执行成功,false:执行失败
     */
    public boolean execUpdate(String sql, List<?> args) {
        return execUpdate(sql, args.toArray());
    }

    /**
     * 执行批量更新数据库操作
     * 
     * @param sql
     *            要执行的SQL语句的字符串数组
     * @return boolean true:执行成功,false:执行失败
     */
    public boolean execUpdate(Object[] sql) {
        boolean flag = false;
        // 判断连接数据库是否成功
        if (openConnection) {
            try {
                statement = connection.createStatement();
                for (int i = 0; i < sql.length; i++) {
                    statement.addBatch((String) sql[i]);
                }
                statement.executeBatch();

                flag = true;
            } catch (SQLException e) {
                try {
                    if (autoCommit) {
                        connection.rollback();
                    }
                } catch (SQLException e1) {
                    throw new RuntimeException(e1);
                }
                for (int i = 0; i < sql.length; i++) {
                    System.out.println("SQL " + (i + 1) + ":" + sql[i]);
                }
                throw new RuntimeException(e);
            } finally {
                if (autoCommit) {
                    closeAll();
                }
            }
        } else {
            System.out.println("数据库连接对象没有打开!");
        }

        return flag;
    }

    /**
     * 执行批量更新数据库操作
     * 
     * @param sql
     *            要执行的SQL语句的集合
     * @return boolean true:执行成功,false:执行失败
     */
    public boolean execUpdate(List<?> sql) {
        return execUpdate(sql.toArray());
    }

    /**
     * 执行数据库查询操作
     * 
     * @param sql
     *            要执行的SQL语句
     * @param args
     *            查询参数列表
     * @return ResultSet 返回查询的结果集对象
     */
    public ResultSet execQuery(String sql, Object... args) {
        rs = null;
        // 判断连接数据库是否成功
        if (openConnection) {
            try {
                ps = connection.prepareStatement(sql);
                // 设置参数
                if (args != null && args.length > 0) {
                    for (int i = 0; i < args.length; i++) {
                        ps.setObject(i + 1, args[i]);
                    }
                }
                
                rs = ps.executeQuery();
            } catch (SQLException e) {
                if (autoCommit) {
                    closeAll();
                }
                System.out.println("SQL:" + sql);
                throw new RuntimeException(e);
            }
        } else {
            System.out.println("数据库连接对象没有打开!");
        }

        return rs;
    }

    /**
     * 执行数据库查询操作
     * 
     * @param sql
     *            要执行的SQL语句
     * @param args
     *            查询参数列表
     * @return ResultSet 返回查询的结果集对象
     */
    public ResultSet execQuery(String sql, List<?> args) {
        return execQuery(sql, args.toArray());
    }

    /**
     * 根据标准SQL查询数据库,返回一个int值
     * 
     * @param sql
     *            要执行的SQL语句
     * @param args
     *            查询参数列表
     * @return int值,如果出错则返回-1
     */
    public int findForInt(String sql, Object... args) {
        ResultSet rs = execQuery(sql, args);
        int count = -1;
        try {
            if (rs != null && rs.next()) {
                count = rs.getInt(1);
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } finally {
            if (autoCommit) {
                closeAll();
            }
        }

        return count;
    }
    
    /**
     * 根据标准SQL查询数据库,返回一个int值
     * 
     * @param sql
     *            要执行的SQL语句
     * @param args
     *            查询参数列表
     * @return int值,如果出错则返回-1
     */
    public int findForInt(String sql, List<?> args) {
        return findForInt(sql, args.toArray());
    }

    /**
     * 关闭所有数据库连接对象
     */
    public void closeAll() {
        if (rs != null || ps != null || statement != null || connection != null) {
            try {
                if (rs != null) {
                    rs.close();
                }
            } catch (SQLException e) {
                throw new RuntimeException(e);
            } finally {
                rs = null;
                if (ps != null || statement != null || connection != null) {
                    try {
                        if (ps != null && !ps.isClosed()) {
                            ps.close();
                        }
                    } catch (SQLException e) {
                        throw new RuntimeException(e);
                    } finally {
                        ps = null;
                        if (statement != null || connection != null) {
                            try {
                                if (statement != null && !statement.isClosed()) {
                                    statement.close();
                                }
                            } catch (SQLException e) {
                                throw new RuntimeException(e);
                            } finally {
                                statement = null;
                                try {
                                    if (connection != null
                                            && !connection.isClosed()) {
                                        connection.close();
                                    }
                                } catch (SQLException e) {
                                    throw new RuntimeException(e);
                                } finally {
                                    connection = null;
                                }
                            }
                        }
                    }
                }
            }
        }
    }

    /**
     * 提交事务并关闭数据库连接
     */
    public void commit() {
        try {
            if (!autoCommit) {
                connection.commit();
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } finally {
            autoCommit = true;
            closeAll();
        }
    }

    /**
     * 回滚事务并关闭数据库连接
     */
    public void rollback() {
        try {
            if (!autoCommit) {
                connection.rollback();
            }
        } catch (SQLException e) {
            throw new RuntimeException(e);
        } finally {
            autoCommit = true;
            closeAll();
        }
    }

    public DataSource getDataSource() {
        return dataSource;
    }

    public void setDataSource(DataSource dataSource) {
        this.dataSource = dataSource;
    }

    public String getDriver() {
        return driver;
    }

    public void setDriver(String driver) {
        this.driver = driver;
    }

    public String getUrl() {
        return url;
    }

    public void setUrl(String url) {
        this.url = url;
    }

    public boolean getAutoCommit() {
        return autoCommit;
    }

    public void setAutoCommit(boolean autoCommit) {
        try {
            connection.setAutoCommit(autoCommit);
        } catch (SQLException e) {
            closeAll();
            throw new RuntimeException(e);
        }
        this.autoCommit = autoCommit;
    }

    public boolean getOpenConnection() {
        return openConnection;
    }

    public String getJndiName() {
        return jndiName;
    }

    public void setJndiName(String jndiName) {
        this.jndiName = jndiName;
    }

    public String getUser() {
        return user;
    }

    public void setUser(String user) {
        this.user = user;
    }

    public String getPwd() {
        return pwd;
    }

    public void setPwd(String pwd) {
        this.pwd = pwd;
    }

    public Connection getConnection() {
        return connection;
    }

    /**
     * 测试数据库连接是否成功
     * 
     * @param args
     * @throws SQLException
     */
    public static void main(String[] args) throws SQLException {
        String driver = "com.microsoft.sqlserver.jdbc.SQLServerDriver";
        String url = "jdbc:sqlserver://127.0.0.1:1433;databaseName=demo";
        String user = "sa";
        String pwd = "sa";

        CommonSql commonSql = new CommonSql(driver, url, user, pwd);
        if (commonSql.openConnection()) {
            System.out.println("数据库连接成功!");
            DatabaseMetaData dbMetaData = commonSql.getConnection()
                    .getMetaData();
            System.out
                    .print("当前连接的数据库是:" + dbMetaData.getDatabaseProductName());
            System.out.println(" " + dbMetaData.getDatabaseProductVersion());
        } else {
            System.out.println("数据库连接失败!");
        }

        commonSql.closeAll();
    }
}