com.ecfront.easybi.dbutils.inner.dbutilsext.QueryRunnerExt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ez-dbutils Show documentation
DBUtils的封装类库
There is a newer version: 2.8.0
package com.ecfront.easybi.dbutils.inner.dbutilsext;

import com.ecfront.easybi.dbutils.inner.ConnectionWrap;
import com.ecfront.easybi.dbutils.inner.dialect.DialectType;
import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.ResultSetHandler;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

public class QueryRunnerExt extends QueryRunner {

    public int[] batch(ConnectionWrap cw, String sql, Object[][] params) throws SQLException {
        if (cw.type != DialectType.SPARK_SQL) {
            return super.batch(cw.conn, sql, params);
        } else {
            return batchSimple(cw.conn, sql);
        }
    }


    public  T query(ConnectionWrap cw, String sql, ResultSetHandler rsh, Object... params) throws SQLException {
        if (cw.type != DialectType.SPARK_SQL) {
            return super.query(cw.conn, sql, rsh, params);
        } else {
            return querySimple(cw.conn, sql, rsh);
        }
    }


    public  T query(ConnectionWrap cw, String sql, ResultSetHandler rsh) throws SQLException {
        if (cw.type != DialectType.SPARK_SQL) {
            return super.query(cw.conn, sql, rsh);
        } else {
            return querySimple(cw.conn, sql, rsh);
        }
    }

    public int update(ConnectionWrap cw, String sql) throws SQLException {
        if (cw.type != DialectType.SPARK_SQL) {
            return super.update(cw.conn, sql);
        } else {
            return updateSimple(cw.conn, sql);
        }
    }

    public int update(ConnectionWrap cw, String sql, Object[] param) throws SQLException {
        if (cw.type != DialectType.SPARK_SQL) {
            return super.update(cw.conn, sql, param);
        } else {
            return updateSimple(cw.conn, sql);
        }
    }


    protected int[] batchSimple(Connection conn, String sql) throws SQLException {
        throw new SQLException("Method not  supported");
    }

    protected  T querySimple(Connection conn, String sql, ResultSetHandler rsh) throws SQLException {
        if (conn == null) {
            throw new SQLException("Null connection");
        }
        if (sql == null) {
            throw new SQLException("Null SQL statement");
        }
        if (rsh == null) {
            throw new SQLException("Null ResultSetHandler");
        }
        Statement st = null;
        ResultSet rs = null;
        T result = null;
        try {
            logger.debug("Executing:" + sql);
            st = conn.createStatement();
            rs = this.wrap(st.executeQuery(sql));
            result = rsh.handle(rs);
            logger.debug("Executed:" + sql);
        } catch (SQLException e) {
            this.rethrow(e, sql);
        } finally {
            try {
                close(rs);
            } finally {
                close(st);
            }
        }
        return result;
    }


    protected int updateSimple(Connection conn, String sql) throws SQLException {
        if (conn == null) {
            throw new SQLException("Null connection");
        }
        if (sql == null) {
            throw new SQLException("Null SQL statement");
        }

        Statement st = null;
        int rows = 0;
        try {
            logger.debug("Executing:"+sql);
            st = conn.createStatement();
            rows = st.executeUpdate(sql);
            logger.debug("Executed:" + sql);
        } catch (SQLException e) {
            this.rethrow(e, sql);
        } finally {
            close(st);
        }
        return rows;
    }

    private static final Logger logger = LoggerFactory.getLogger(QueryRunnerExt.class);
}