Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/*
* Copyright (c) 2015, Haiyang Li.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.landawn.abacus.util;
import static com.landawn.abacus.core.AbacusConfiguration.DataSourceConfiguration.DRIVER;
import static com.landawn.abacus.core.AbacusConfiguration.DataSourceConfiguration.PASSWORD;
import static com.landawn.abacus.core.AbacusConfiguration.DataSourceConfiguration.URL;
import static com.landawn.abacus.core.AbacusConfiguration.DataSourceConfiguration.USER;
import static com.landawn.abacus.util.IOUtil.DEFAULT_QUEUE_SIZE_FOR_ROW_PARSER;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.io.Reader;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.math.BigDecimal;
import java.sql.CallableStatement;
import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.sql.SQLFeatureNotSupportedException;
import java.sql.SQLType;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Stack;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.Executor;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;
import javax.xml.parsers.DocumentBuilder;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.xml.sax.SAXException;
import com.landawn.abacus.DataSet;
import com.landawn.abacus.DataSource;
import com.landawn.abacus.DataSourceManager;
import com.landawn.abacus.DataSourceSelector;
import com.landawn.abacus.DirtyMarker;
import com.landawn.abacus.IsolationLevel;
import com.landawn.abacus.SliceSelector;
import com.landawn.abacus.Transaction;
import com.landawn.abacus.Transaction.Status;
import com.landawn.abacus.core.AbacusConfiguration;
import com.landawn.abacus.core.AbacusConfiguration.DataSourceConfiguration;
import com.landawn.abacus.core.AbacusConfiguration.DataSourceManagerConfiguration;
import com.landawn.abacus.core.RowDataSet;
import com.landawn.abacus.core.sql.dataSource.NonSliceSelector;
import com.landawn.abacus.core.sql.dataSource.SQLDataSource;
import com.landawn.abacus.core.sql.dataSource.SQLDataSourceManager;
import com.landawn.abacus.core.sql.dataSource.SimpleSourceSelector;
import com.landawn.abacus.exception.AbacusException;
import com.landawn.abacus.exception.NonUniqueResultException;
import com.landawn.abacus.exception.ParseException;
import com.landawn.abacus.exception.UncheckedIOException;
import com.landawn.abacus.exception.UncheckedSQLException;
import com.landawn.abacus.logging.Logger;
import com.landawn.abacus.logging.LoggerFactory;
import com.landawn.abacus.parser.ParserUtil;
import com.landawn.abacus.parser.ParserUtil.EntityInfo;
import com.landawn.abacus.type.Type;
import com.landawn.abacus.util.SQLExecutor.JdbcSettings;
import com.landawn.abacus.util.Tuple.Tuple2;
import com.landawn.abacus.util.Tuple.Tuple3;
import com.landawn.abacus.util.Tuple.Tuple4;
import com.landawn.abacus.util.Tuple.Tuple5;
import com.landawn.abacus.util.stream.Stream;
/**
*
* @since 0.8
*
* @author Haiyang Li
*
* @see http://docs.oracle.com/javase/7/docs/api/java/sql/Connection.html
* @see http://docs.oracle.com/javase/7/docs/api/java/sql/Statement.html
* @see http://docs.oracle.com/javase/7/docs/api/java/sql/PreparedStatement.html
* @see http://docs.oracle.com/javase/7/docs/api/java/sql/ResultSet.html
*/
public final class JdbcUtil {
private static final Logger logger = LoggerFactory.getLogger(JdbcUtil.class);
// ...
private static final String CURRENT_DIR_PATH = "./";
private static final Try.BiConsumer super PreparedStatement, ? super Object[], SQLException> DEFAULT_STMT_SETTER = new Try.BiConsumer() {
@Override
public void accept(PreparedStatement stmt, Object[] parameters) throws SQLException {
for (int i = 0, len = parameters.length; i < len; i++) {
stmt.setObject(i + 1, parameters[i]);
}
}
};
private static final Set sqlStateForTableNotExists = new HashSet<>();
static {
sqlStateForTableNotExists.add("42S02"); // for MySQL.
sqlStateForTableNotExists.add("42P01"); // for PostgreSQL.
sqlStateForTableNotExists.add("42501"); // for HSQLDB.
}
private JdbcUtil() {
// singleton
}
public static DBVersion getDBVersion(final Connection conn) throws UncheckedSQLException {
try {
String dbProudctName = conn.getMetaData().getDatabaseProductName();
String dbProudctVersion = conn.getMetaData().getDatabaseProductVersion();
DBVersion dbVersion = DBVersion.OTHERS;
String upperCaseProductName = dbProudctName.toUpperCase();
if (upperCaseProductName.contains("H2")) {
dbVersion = DBVersion.H2;
} else if (upperCaseProductName.contains("HSQL")) {
dbVersion = DBVersion.HSQLDB;
} else if (upperCaseProductName.contains("MYSQL")) {
if (dbProudctVersion.startsWith("5.5")) {
dbVersion = DBVersion.MYSQL_5_5;
} else if (dbProudctVersion.startsWith("5.6")) {
dbVersion = DBVersion.MYSQL_5_6;
} else if (dbProudctVersion.startsWith("5.7")) {
dbVersion = DBVersion.MYSQL_5_7;
} else if (dbProudctVersion.startsWith("5.8")) {
dbVersion = DBVersion.MYSQL_5_8;
} else if (dbProudctVersion.startsWith("5.9")) {
dbVersion = DBVersion.MYSQL_5_9;
} else if (dbProudctVersion.startsWith("6")) {
dbVersion = DBVersion.MYSQL_6;
} else if (dbProudctVersion.startsWith("7")) {
dbVersion = DBVersion.MYSQL_7;
} else {
dbVersion = DBVersion.MYSQL_OTHERS;
}
} else if (upperCaseProductName.contains("POSTGRESQL")) {
if (dbProudctVersion.startsWith("9.2")) {
dbVersion = DBVersion.POSTGRESQL_9_2;
} else if (dbProudctVersion.startsWith("9.3")) {
dbVersion = DBVersion.POSTGRESQL_9_3;
} else if (dbProudctVersion.startsWith("9.4")) {
dbVersion = DBVersion.POSTGRESQL_9_4;
} else if (dbProudctVersion.startsWith("9.5")) {
dbVersion = DBVersion.POSTGRESQL_9_5;
} else if (dbProudctVersion.startsWith("10")) {
dbVersion = DBVersion.POSTGRESQL_10;
} else {
dbVersion = DBVersion.POSTGRESQL_OTHERS;
}
} else if (upperCaseProductName.contains("ORACLE")) {
dbVersion = DBVersion.ORACLE;
} else if (upperCaseProductName.contains("DB2")) {
dbVersion = DBVersion.DB2;
} else if (upperCaseProductName.contains("SQL SERVER")) {
dbVersion = DBVersion.SQL_SERVER;
}
return dbVersion;
} catch (SQLException e) {
throw new UncheckedSQLException(e);
}
}
/**
*
* @param dataSourceXmlFile
* @return DataSourceManager
*
* @see DataSource.xsd
*/
public static DataSourceManager createDataSourceManager(final String dataSourceXmlFile) throws UncheckedIOException, UncheckedSQLException {
InputStream is = null;
try {
is = new FileInputStream(Configuration.findFile(dataSourceXmlFile));
return createDataSourceManager(is, dataSourceXmlFile);
} catch (IOException e) {
throw new UncheckedIOException(e);
} finally {
IOUtil.close(is);
}
}
/**
*
* @param dataSourceXmlInputStream
* @return DataSourceManager
*
* @see DataSource.xsd
*/
public static DataSourceManager createDataSourceManager(final InputStream dataSourceXmlInputStream) throws UncheckedIOException, UncheckedSQLException {
return createDataSourceManager(dataSourceXmlInputStream, CURRENT_DIR_PATH);
}
private static DataSourceManager createDataSourceManager(final InputStream dataSourceXmlInputStream, final String dataSourceXmlFile)
throws UncheckedIOException, UncheckedSQLException {
DocumentBuilder domParser = XMLUtil.createDOMParser();
Document doc = null;
try {
doc = domParser.parse(dataSourceXmlInputStream);
Element rootElement = doc.getDocumentElement();
final Map props = new HashMap<>();
List propertiesElementList = XMLUtil.getElementsByTagName(rootElement, AbacusConfiguration.PROPERTIES);
if (N.notNullOrEmpty(propertiesElementList)) {
for (Element propertiesElement : propertiesElementList) {
File resourcePropertiesFile = Configuration.findFileByFile(new File(dataSourceXmlFile),
propertiesElement.getAttribute(AbacusConfiguration.RESOURCE));
java.util.Properties properties = new java.util.Properties();
InputStream is = null;
try {
is = new FileInputStream(resourcePropertiesFile);
if (resourcePropertiesFile.getName().endsWith(".xml")) {
properties.loadFromXML(is);
} else {
properties.load(is);
}
} finally {
IOUtil.close(is);
}
for (Object key : properties.keySet()) {
props.put((String) key, (String) properties.get(key));
}
}
}
String nodeName = rootElement.getNodeName();
if (nodeName.equals(DataSourceManagerConfiguration.DATA_SOURCE_MANAGER)) {
DataSourceManagerConfiguration config = new DataSourceManagerConfiguration(rootElement, props);
return new SQLDataSourceManager(config);
} else if (nodeName.equals(DataSourceConfiguration.DATA_SOURCE)) {
DataSourceConfiguration config = new DataSourceConfiguration(rootElement, props);
return new SimpleDataSourceManager(new SQLDataSource(config));
} else {
throw new AbacusException("Unknown xml format with root element: " + nodeName);
}
} catch (SAXException e) {
throw new ParseException(e);
} catch (IOException e) {
throw new UncheckedIOException(e);
}
}
public static DataSource createDataSource(final String dataSourceFile) throws UncheckedIOException, UncheckedSQLException {
InputStream is = null;
try {
is = new FileInputStream(Configuration.findFile(dataSourceFile));
return createDataSource(is, dataSourceFile);
} catch (IOException e) {
throw new UncheckedIOException(e);
} finally {
IOUtil.close(is);
}
}
public static DataSource createDataSource(final InputStream dataSourceInputStream) throws UncheckedIOException, UncheckedSQLException {
return createDataSource(dataSourceInputStream, CURRENT_DIR_PATH);
}
private static DataSource createDataSource(final InputStream dataSourceInputStream, final String dataSourceFile)
throws UncheckedIOException, UncheckedSQLException {
final String dataSourceString = IOUtil.readString(dataSourceInputStream);
if (CURRENT_DIR_PATH.equals(dataSourceFile) || dataSourceFile.endsWith(".xml")) {
try {
return createDataSourceManager(new ByteArrayInputStream(dataSourceString.getBytes())).getPrimaryDataSource();
} catch (ParseException e) {
// ignore.
} catch (UncheckedIOException e) {
// ignore.
}
}
final Map newProps = new HashMap<>();
final java.util.Properties properties = new java.util.Properties();
try {
properties.load(new ByteArrayInputStream(dataSourceString.getBytes()));
Object value = null;
for (Object key : properties.keySet()) {
value = properties.get(key);
newProps.put(key.toString().trim(), value.toString().trim());
}
} catch (IOException e) {
throw new UncheckedIOException(e);
}
return new SQLDataSource(newProps);
}
public static DataSource createDataSource(final String url, final String user, final String password) throws UncheckedSQLException {
return createDataSource(getDriverClasssByUrl(url), url, user, password);
}
public static DataSource createDataSource(final String driver, final String url, final String user, final String password) throws UncheckedSQLException {
final Class extends Driver> driverClass = ClassUtil.forClass(driver);
return createDataSource(driverClass, url, user, password);
}
public static DataSource createDataSource(final Class extends Driver> driverClass, final String url, final String user, final String password)
throws UncheckedSQLException {
final Map props = new HashMap<>();
props.put(DRIVER, driverClass.getCanonicalName());
props.put(URL, url);
props.put(USER, user);
props.put(PASSWORD, password);
return createDataSource(props);
}
/**
*
* @param props refer to Connection.xsd for the supported properties.
* @return
*/
public static DataSource createDataSource(final Map props) throws UncheckedSQLException {
final String driver = (String) props.get(DRIVER);
if (N.isNullOrEmpty(driver)) {
final String url = (String) props.get(URL);
if (N.isNullOrEmpty(url)) {
throw new IllegalArgumentException("Url is not specified");
}
final Map tmp = new HashMap<>(props);
tmp.put(DRIVER, getDriverClasssByUrl(url).getCanonicalName());
return new SQLDataSource(tmp);
} else {
return new SQLDataSource(props);
}
}
public static DataSource wrap(final javax.sql.DataSource sqlDataSource) {
return sqlDataSource instanceof DataSource ? ((DataSource) sqlDataSource) : new SimpleDataSource(sqlDataSource);
}
public static Connection createConnection(final String url, final String user, final String password) throws UncheckedSQLException {
return createConnection(getDriverClasssByUrl(url), url, user, password);
}
private static Class extends Driver> getDriverClasssByUrl(final String url) {
Class extends Driver> driverClass = null;
// jdbc:mysql://localhost:3306/abacustest
if (url.indexOf("mysql") > 0 || StringUtil.indexOfIgnoreCase(url, "mysql") > 0) {
driverClass = ClassUtil.forClass("com.mysql.jdbc.Driver");
// jdbc:postgresql://localhost:5432/abacustest
} else if (url.indexOf("postgresql") > 0 || StringUtil.indexOfIgnoreCase(url, "postgresql") > 0) {
driverClass = ClassUtil.forClass("org.postgresql.Driver");
// jdbc:h2:hsql://:/
} else if (url.indexOf("h2") > 0 || StringUtil.indexOfIgnoreCase(url, "h2") > 0) {
driverClass = ClassUtil.forClass("org.h2.Driver");
// jdbc:hsqldb:hsql://localhost/abacustest
} else if (url.indexOf("hsqldb") > 0 || StringUtil.indexOfIgnoreCase(url, "hsqldb") > 0) {
driverClass = ClassUtil.forClass("org.hsqldb.jdbc.JDBCDriver");
// jdbc.url=jdbc:oracle:thin:@localhost:1521:abacustest
} else if (url.indexOf("oracle") > 0 || StringUtil.indexOfIgnoreCase(url, "oracle") > 0) {
driverClass = ClassUtil.forClass("oracle.jdbc.driver.OracleDriver");
// jdbc.url=jdbc:sqlserver://localhost:1433;Database=abacustest
} else if (url.indexOf("sqlserver") > 0 || StringUtil.indexOfIgnoreCase(url, "sqlserver") > 0) {
driverClass = ClassUtil.forClass("com.microsoft.sqlserver.jdbc.SQLServerDrive");
// jdbc:db2://localhost:50000/abacustest
} else if (url.indexOf("db2") > 0 || StringUtil.indexOfIgnoreCase(url, "db2") > 0) {
driverClass = ClassUtil.forClass("com.ibm.db2.jcc.DB2Driver");
} else {
throw new IllegalArgumentException(
"Can not identity the driver class by url: " + url + ". Only mysql, postgresql, hsqldb, sqlserver, oracle and db2 are supported currently");
}
return driverClass;
}
public static Connection createConnection(final String driverClass, final String url, final String user, final String password)
throws UncheckedSQLException {
Class extends Driver> cls = ClassUtil.forClass(driverClass);
return createConnection(cls, url, user, password);
}
public static Connection createConnection(final Class extends Driver> driverClass, final String url, final String user, final String password)
throws UncheckedSQLException {
try {
DriverManager.registerDriver(N.newInstance(driverClass));
return DriverManager.getConnection(url, user, password);
} catch (SQLException e) {
throw new UncheckedSQLException("Failed to close create connection", e);
}
}
public static void close(final ResultSet rs) throws UncheckedSQLException {
if (rs != null) {
try {
rs.close();
} catch (SQLException e) {
throw new UncheckedSQLException(e);
}
}
}
public static void close(final ResultSet rs, final boolean closeStatement) throws UncheckedSQLException {
close(rs, closeStatement, false);
}
/**
*
* @param rs
* @param closeStatement
* @param closeConnection
* @throws IllegalArgumentException if {@code closeStatement = false} while {@code closeConnection = true}.
* @throws UncheckedSQLException
*/
public static void close(final ResultSet rs, final boolean closeStatement, final boolean closeConnection)
throws IllegalArgumentException, UncheckedSQLException {
if (closeConnection && closeStatement == false) {
throw new IllegalArgumentException("'closeStatement' can't be false while 'closeConnection' is true");
}
if (rs == null) {
return;
}
Connection conn = null;
Statement stmt = null;
try {
if (closeStatement || closeConnection) {
stmt = rs.getStatement();
}
if (closeConnection && stmt != null) {
conn = stmt.getConnection();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
close(rs, stmt, conn);
}
}
public static void close(final Statement stmt) throws UncheckedSQLException {
if (stmt != null) {
try {
if (stmt instanceof PreparedStatement) {
try {
((PreparedStatement) stmt).clearParameters();
} catch (SQLException e) {
logger.error("Failed to clear parameters", e);
}
}
stmt.close();
} catch (SQLException e) {
throw new UncheckedSQLException(e);
}
}
}
public static void close(final Connection conn) throws UncheckedSQLException {
if (conn != null) {
try {
conn.close();
} catch (SQLException e) {
throw new UncheckedSQLException(e);
}
}
}
public static void close(final ResultSet rs, final Statement stmt) throws UncheckedSQLException {
try {
if (rs != null) {
rs.close();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
try {
if (stmt != null) {
if (stmt instanceof PreparedStatement) {
try {
((PreparedStatement) stmt).clearParameters();
} catch (SQLException e) {
logger.error("Failed to clear parameters", e);
}
}
stmt.close();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
}
}
}
public static void close(final Statement stmt, final Connection conn) throws UncheckedSQLException {
try {
if (stmt != null) {
if (stmt instanceof PreparedStatement) {
try {
((PreparedStatement) stmt).clearParameters();
} catch (SQLException e) {
logger.error("Failed to clear parameters", e);
}
}
stmt.close();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
try {
if (conn != null) {
conn.close();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
}
}
}
public static void close(final ResultSet rs, final Statement stmt, final Connection conn) throws UncheckedSQLException {
try {
if (rs != null) {
rs.close();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
try {
if (stmt != null) {
if (stmt instanceof PreparedStatement) {
try {
((PreparedStatement) stmt).clearParameters();
} catch (SQLException e) {
logger.error("Failed to clear parameters", e);
}
}
stmt.close();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
try {
if (conn != null) {
conn.close();
}
} catch (SQLException e) {
throw new UncheckedSQLException(e);
}
}
}
}
/**
* Unconditionally close an ResultSet.
*
* Equivalent to {@link ResultSet#close()}, except any exceptions will be ignored.
* This is typically used in finally blocks.
*
* @param rs
*/
public static void closeQuietly(final ResultSet rs) {
closeQuietly(rs, null, null);
}
public static void closeQuietly(final ResultSet rs, final boolean closeStatement) throws UncheckedSQLException {
closeQuietly(rs, closeStatement, false);
}
/**
*
* @param rs
* @param closeStatement
* @param closeConnection
* @throws IllegalArgumentException if {@code closeStatement = false} while {@code closeConnection = true}.
*/
public static void closeQuietly(final ResultSet rs, final boolean closeStatement, final boolean closeConnection) throws IllegalArgumentException {
if (closeConnection && closeStatement == false) {
throw new IllegalArgumentException("'closeStatement' can't be false while 'closeConnection' is true");
}
if (rs == null) {
return;
}
Connection conn = null;
Statement stmt = null;
try {
if (closeStatement || closeConnection) {
stmt = rs.getStatement();
}
if (closeConnection && stmt != null) {
conn = stmt.getConnection();
}
} catch (SQLException e) {
logger.error("Failed to get Statement or Connection by ResultSet", e);
} finally {
closeQuietly(rs, stmt, conn);
}
}
/**
* Unconditionally close an Statement.
*
* Equivalent to {@link Statement#close()}, except any exceptions will be ignored.
* This is typically used in finally blocks.
*
* @param stmt
*/
public static void closeQuietly(final Statement stmt) {
closeQuietly(null, stmt, null);
}
/**
* Unconditionally close an Connection.
*
* Equivalent to {@link Connection#close()}, except any exceptions will be ignored.
* This is typically used in finally blocks.
*
* @param conn
*/
public static void closeQuietly(final Connection conn) {
closeQuietly(null, null, conn);
}
/**
* Unconditionally close the ResultSet, Statement.
*
* Equivalent to {@link ResultSet#close()}, {@link Statement#close()}, except any exceptions will be ignored.
* This is typically used in finally blocks.
*
* @param rs
* @param stmt
*/
public static void closeQuietly(final ResultSet rs, final Statement stmt) {
closeQuietly(rs, stmt, null);
}
/**
* Unconditionally close the Statement, Connection.
*
* Equivalent to {@link Statement#close()}, {@link Connection#close()}, except any exceptions will be ignored.
* This is typically used in finally blocks.
*
* @param stmt
* @param conn
*/
public static void closeQuietly(final Statement stmt, final Connection conn) {
closeQuietly(null, stmt, conn);
}
/**
* Unconditionally close the ResultSet, Statement, Connection.
*
* Equivalent to {@link ResultSet#close()}, {@link Statement#close()}, {@link Connection#close()}, except any exceptions will be ignored.
* This is typically used in finally blocks.
*
* @param rs
* @param stmt
* @param conn
*/
public static void closeQuietly(final ResultSet rs, final Statement stmt, final Connection conn) {
if (rs != null) {
try {
rs.close();
} catch (Exception e) {
logger.error("Failed to close ResultSet", e);
}
}
if (stmt != null) {
if (stmt instanceof PreparedStatement) {
try {
((PreparedStatement) stmt).clearParameters();
} catch (Exception e) {
logger.error("Failed to clear parameters", e);
}
}
try {
stmt.close();
} catch (Exception e) {
logger.error("Failed to close Statement", e);
}
}
if (conn != null) {
try {
conn.close();
} catch (Exception e) {
logger.error("Failed to close Connection", e);
}
}
}
/**
*
* @param rs
* @param n the count of row to move ahead.
* @throws SQLException
*/
public static void skip(final ResultSet rs, int n) throws SQLException {
skip(rs, (long) n);
}
/**
*
* @param rs
* @param n the count of row to move ahead.
* @throws SQLException
* @see {@link ResultSet#absolute(int)}
*/
public static void skip(final ResultSet rs, long n) throws SQLException {
// N.checkArgument(offset >= 0, "'offset' can't be negative: %s", offset);
//
// if (offset == 0) {
// return;
// }
if (n <= 0) {
return;
} else if (n == 1) {
rs.next();
} else if (n <= Integer.MAX_VALUE) {
try {
if (n > Integer.MAX_VALUE - rs.getRow()) {
while (n-- > 0L && rs.next()) {
}
} else {
rs.absolute((int) n + rs.getRow());
}
} catch (SQLException e) {
while (n-- > 0L && rs.next()) {
}
}
} else {
while (n-- > 0L && rs.next()) {
}
}
}
/**
*
* @param conn
* @param tableName
* @return
* @throws SQLException
*/
public static List getColumnNameList(final Connection conn, final String tableName) throws SQLException {
final String query = "SELECT * FROM " + tableName + " WHERE 1 > 2";
PreparedStatement stmt = null;
ResultSet rs = null;
try {
stmt = prepareStatement(conn, query);
rs = stmt.executeQuery();
final ResultSetMetaData metaData = rs.getMetaData();
final int columnCount = metaData.getColumnCount();
final List columnNameList = new ArrayList<>(columnCount);
for (int i = 1, n = columnCount + 1; i < n; i++) {
columnNameList.add(metaData.getColumnName(i));
}
return columnNameList;
} finally {
closeQuietly(rs, stmt);
}
}
/**
*
* @param rs
* @return
* @throws SQLException
*/
public static List getColumnLabelList(ResultSet rs) throws SQLException {
final ResultSetMetaData metaData = rs.getMetaData();
final int columnCount = metaData.getColumnCount();
final List labelList = new ArrayList<>(columnCount);
for (int i = 1, n = columnCount + 1; i < n; i++) {
labelList.add(metaData.getColumnLabel(i));
}
return labelList;
}
public static int getColumnCount(ResultSet rs) throws SQLException {
return rs.getMetaData().getColumnCount();
}
/**
* Here is the general code pattern to work with {@code SimpleTransaction}.
* The transaction will be shared in the same thread for the same {@code DataSource} or {@code Connection}.
*
*
*
* public void doSomethingA() {
* ...
* final SimpleTransaction tranA = JdbcUtil.beginTransacion(dataSource1, isolation);
* final Connection conn = tranA.connection();
* try {
* // do your work with the conn...
* ...
* doSomethingB(); // Share the same transaction 'tranA' because they're in the same thread and start transaction with same DataSource 'dataSource1'.
* ...
* doSomethingC(); // won't share the same transaction 'tranA' although they're in the same thread but start transaction with different DataSources.
* ...
* tranA.commit();
* } finally {
* tranA.rollbackIfNotCommitted();
* }
* }
*
* public void doSomethingB() {
* ...
* final SimpleTransaction tranB = JdbcUtil.beginTransacion(dataSource1, isolation);
* final Connection conn = tranB.connection();
* try {
* // do your work with the conn...
* ...
* tranB.commit();
* } finally {
* tranB.rollbackIfNotCommitted();
* }
* }
*
* public void doSomethingC() {
* ...
* final SimpleTransaction tranC = JdbcUtil.beginTransacion(dataSource2, isolation);
* final Connection conn = tranC.connection();
* try {
* // do your work with the conn...
* ...
* tranC.commit();
* } finally {
* tranC.rollbackIfNotCommitted();
* }
* }
*
*
*
* It's incorrect to use flag to identity the transaction should be committed or rolled back.
* Don't write below code:
*
*
* public void doSomethingA() {
* ...
* final SimpleTransaction tranA = JdbcUtil.beginTransacion(dataSource1, isolation);
* final Connection conn = tranA.connection();
* boolean flagToCommit = false;
* try {
* // do your work with the conn...
* ...
* flagToCommit = true;
* } finally {
* if (flagToCommit) {
* tranA.commit();
* } else {
* tranA.rollbackIfNotCommitted();
* }
* }
* }
*
*
*
* @param dataSource
* @param isolationLevel
* @return
* @throws SQLException
*/
public static SimpleTransaction beginTransaction(javax.sql.DataSource dataSource, IsolationLevel isolationLevel) throws UncheckedSQLException {
N.checkArgNotNull(dataSource);
N.checkArgNotNull(isolationLevel);
final String ttid = SimpleTransaction.getTransactionThreadId(dataSource);
SimpleTransaction tran = SimpleTransaction.threadTransacionMap.get(ttid);
if (tran == null) {
Connection conn = null;
boolean isOk = false;
try {
conn = dataSource.getConnection();
tran = new SimpleTransaction(ttid, conn, isolationLevel, true);
tran.incrementAndGet(isolationLevel);
isOk = true;
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
if (isOk == false) {
closeQuietly(conn);
}
}
logger.info("Create a new transaction(id={})", tran.id());
SimpleTransaction.threadTransacionMap.put(ttid, tran);
} else {
logger.info("Reusing the existing transaction(id={})", tran.id());
tran.incrementAndGet(isolationLevel);
}
logger.debug("Current active transaction: {}", SimpleTransaction.threadTransacionMap.values());
return tran;
}
/**
* Here is the general code pattern to work with {@code SimpleTransaction}.
* The transaction will be shared in the same thread for the same {@code DataSource} or {@code Connection}.
*
*
*
* public void doSomethingA() {
* ...
* final SimpleTransaction tranA = JdbcUtil.beginTransacion(conn1, isolation);
* final Connection conn = tranA.connection();
* try {
* // do your work with the conn...
* ...
* doSomethingB(); // Share the same transaction 'tranA' because they're in the same thread and start transaction with same Connection 'conn1'.
* ...
* doSomethingC(); // won't share the same transaction 'tranA' although they're in the same thread but start transaction with different Connections.
* ...
* tranA.commit();
* } finally {
* tranA.rollbackIfNotCommitted();
* }
* }
*
* public void doSomethingB() {
* ...
* final SimpleTransaction tranB = JdbcUtil.beginTransacion(conn1, isolation);
* final Connection conn = tranB.connection();
* try {
* // do your work with the conn...
* ...
* tranB.commit();
* } finally {
* tranB.rollbackIfNotCommitted();
* }
* }
*
* public void doSomethingC() {
* ...
* final SimpleTransaction tranC = JdbcUtil.beginTransacion(conn2, isolation);
* final Connection conn = tranC.connection();
* try {
* // do your work with the conn...
* ...
* tranC.commit();
* } finally {
* tranC.rollbackIfNotCommitted();
* }
* }
*
*
*
* It's incorrect to use flag to identity the transaction should be committed or rolled back.
* Don't write below code:
*
*
* public void doSomethingA() {
* ...
* final SimpleTransaction tranA = JdbcUtil.beginTransacion(conn1, isolation);
* final Connection conn = tranA.connection();
* boolean flagToCommit = false;
* try {
* // do your work with the conn...
* ...
* flagToCommit = true;
* } finally {
* if (flagToCommit) {
* tranA.commit();
* } else {
* tranA.rollbackIfNotCommitted();
* }
* }
* }
*
*
*
* Never write below code because it will definitely cause {@code Connection} leak:
*
*
* @param conn the specified {@code conn} won't be close after this query is executed.
* @param sql
* @return
* @throws SQLException
*/
public static PreparedQuery prepareQuery(final Connection conn, final String sql) throws SQLException {
return new PreparedQuery(prepareStatement(conn, sql));
}
/**
* Never write below code because it will definitely cause {@code Connection} leak:
*
*
* @param conn the specified {@code conn} won't be close after this query is executed.
* @param sql
* @param autoGeneratedKeys
* @return
* @throws SQLException
*/
public static PreparedQuery prepareQuery(final Connection conn, final String sql, final boolean autoGeneratedKeys) throws SQLException {
return new PreparedQuery(conn.prepareStatement(sql, autoGeneratedKeys ? Statement.RETURN_GENERATED_KEYS : Statement.NO_GENERATED_KEYS));
}
/**
*
* @param ds
* @param stmtCreator the created {@code PreparedStatement} will be closed after any execution methods in {@code PreparedQuery/PreparedCallableQuery} is called.
* An execution method is a method which doesn't return the instance of {@code PreparedQuery/PreparedCallableQuery}.
* @return
* @throws SQLException
*/
@SuppressWarnings("resource")
public static PreparedQuery prepareQuery(final javax.sql.DataSource ds, final Try.Function stmtCreator)
throws UncheckedSQLException {
PreparedQuery result = null;
Connection conn = null;
try {
conn = ds.getConnection();
result = new PreparedQuery(stmtCreator.apply(conn)).onClose(conn);
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
if (result == null) {
closeQuietly(conn);
}
}
return result;
}
/**
* Never write below code because it will definitely cause {@code Connection} leak:
*
*
* @param conn the specified {@code conn} won't be close after this query is executed.
* @param stmtCreator the created {@code PreparedStatement} will be closed after any execution methods in {@code PreparedQuery/PreparedCallableQuery} is called.
* An execution method is a method which doesn't return the instance of {@code PreparedQuery/PreparedCallableQuery}.
* @return
* @throws SQLException
*/
public static PreparedQuery prepareQuery(final Connection conn, final Try.Function stmtCreator)
throws SQLException {
return new PreparedQuery(stmtCreator.apply(conn));
}
/**
*
* @param stmtCreator the created {@code PreparedStatement} will be closed after any execution methods in {@code PreparedQuery/PreparedCallableQuery} is called.
* An execution method is a method which doesn't return the instance of {@code PreparedQuery/PreparedCallableQuery}.
* @return
* @throws SQLException
*/
public static PreparedQuery prepareQuery(final Try.Supplier stmtCreator) throws SQLException {
return new PreparedQuery(stmtCreator.get());
}
public static PreparedCallableQuery prepareCallableQuery(final javax.sql.DataSource ds, final String sql) throws UncheckedSQLException {
PreparedCallableQuery result = null;
Connection conn = null;
try {
conn = ds.getConnection();
result = prepareCallableQuery(conn, sql).onClose(conn);
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
if (result == null) {
closeQuietly(conn);
}
}
return result;
}
/**
* Never write below code because it will definitely cause {@code Connection} leak:
*
*
* @param conn the specified {@code conn} won't be close after this query is executed.
* @param sql
* @return
* @throws SQLException
*/
public static PreparedCallableQuery prepareCallableQuery(final Connection conn, final String sql) throws SQLException {
return new PreparedCallableQuery(prepareCall(conn, sql));
}
/**
* @param ds
* @param stmtCreator the created {@code CallableStatement} will be closed after any execution methods in {@code PreparedQuery/PreparedCallableQuery} is called.
* An execution method is a method which doesn't return the instance of {@code PreparedQuery/PreparedCallableQuery}.
* @return
* @throws SQLException
*/
@SuppressWarnings("resource")
public static PreparedCallableQuery prepareCallableQuery(final javax.sql.DataSource ds,
final Try.Function stmtCreator) throws UncheckedSQLException {
PreparedCallableQuery result = null;
Connection conn = null;
try {
conn = ds.getConnection();
result = new PreparedCallableQuery(stmtCreator.apply(conn)).onClose(conn);
} catch (SQLException e) {
throw new UncheckedSQLException(e);
} finally {
if (result == null) {
closeQuietly(conn);
}
}
return result;
}
/**
* Never write below code because it will definitely cause {@code Connection} leak:
*