lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

DataJdbcSource优化

Browse Source
This commit is contained in:
redkale
2023-07-18 20:01:43 +08:00
parent 2a5184f4fc
commit 7dcaa3eda0
1 changed files with 54 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

108
src/main/java/org/redkale/source/DataJdbcSource.java
View File

@@ -385,16 +385,16 @@ public class DataJdbcSource extends AbstractDataSqlSource {
throw se; throw se;
} }
//创建单表结构 //创建单表结构
Statement st = conn.createUpdateStatement(); Statement stmt = conn.createUpdateStatement();
if (tableSqls.length == 1) { if (tableSqls.length == 1) {
st.execute(tableSqls[0]); stmt.execute(tableSqls[0]);
} else { } else {
for (String tableSql : tableSqls) { for (String tableSql : tableSqls) {
st.addBatch(tableSql); stmt.addBatch(tableSql);
} }
st.executeBatch(); stmt.executeBatch();
} }
st.close(); stmt.close();
} else { //分库分表 } else { //分库分表
info.disTableLock().lock(); info.disTableLock().lock();
try { try {
@@ -409,77 +409,77 @@ public class DataJdbcSource extends AbstractDataSqlSource {
}); });
try { try {
//执行一遍创建分表操作 //执行一遍创建分表操作
Statement st = conn.createUpdateStatement(); Statement stmt = conn.createUpdateStatement();
for (String copySql : tableCopys) { for (String copySql : tableCopys) {
st.addBatch(copySql); stmt.addBatch(copySql);
} }
st.executeBatch(); stmt.executeBatch();
st.close(); stmt.close();
} catch (SQLException sqle) { //多进程并发时可能会出现重复建表 } catch (SQLException sqle) { //多进程并发时可能会出现重复建表
if (isTableNotExist(info, sqle.getSQLState())) { if (isTableNotExist(info, sqle.getSQLState())) {
if (newCatalogs.isEmpty()) { //分表的原始表不存在 if (newCatalogs.isEmpty()) { //分表的原始表不存在
String[] tableSqls = createTableSqls(info); String[] tableSqls = createTableSqls(info);
if (tableSqls != null) { if (tableSqls != null) {
//创建原始表 //创建原始表
Statement st = conn.createUpdateStatement(); Statement stmt = conn.createUpdateStatement();
if (tableSqls.length == 1) { if (tableSqls.length == 1) {
st.execute(tableSqls[0]); stmt.execute(tableSqls[0]);
} else { } else {
for (String tableSql : tableSqls) { for (String tableSql : tableSqls) {
st.addBatch(tableSql); stmt.addBatch(tableSql);
} }
st.executeBatch(); stmt.executeBatch();
} }
st.close(); stmt.close();
//再执行一遍创建分表操作 //再执行一遍创建分表操作
st = conn.createUpdateStatement(); stmt = conn.createUpdateStatement();
for (String copySql : tableCopys) { for (String copySql : tableCopys) {
st.addBatch(copySql); stmt.addBatch(copySql);
} }
st.executeBatch(); stmt.executeBatch();
st.close(); stmt.close();
} }
} else { //需要先建库 } else { //需要先建库
Statement st; Statement stmt;
try { try {
st = conn.createUpdateStatement(); stmt = conn.createUpdateStatement();
for (String newCatalog : newCatalogs) { for (String newCatalog : newCatalogs) {
st.addBatch(("postgresql".equals(dbtype()) ? "CREATE SCHEMA IF NOT EXISTS " : "CREATE DATABASE IF NOT EXISTS ") + newCatalog); stmt.addBatch(("postgresql".equals(dbtype()) ? "CREATE SCHEMA IF NOT EXISTS " : "CREATE DATABASE IF NOT EXISTS ") + newCatalog);
} }
st.executeBatch(); stmt.executeBatch();
st.close(); stmt.close();
} catch (SQLException sqle1) { } catch (SQLException sqle1) {
logger.log(Level.SEVERE, "create database " + tableCopys + " error", sqle1); logger.log(Level.SEVERE, "create database " + tableCopys + " error", sqle1);
} }
try { try {
//再执行一遍创建分表操作 //再执行一遍创建分表操作
st = conn.createUpdateStatement(); stmt = conn.createUpdateStatement();
for (String copySql : tableCopys) { for (String copySql : tableCopys) {
st.addBatch(copySql); stmt.addBatch(copySql);
} }
st.executeBatch(); stmt.executeBatch();
st.close(); stmt.close();
} catch (SQLException sqle2) { } catch (SQLException sqle2) {
if (isTableNotExist(info, sqle2.getSQLState())) { if (isTableNotExist(info, sqle2.getSQLState())) {
String[] tableSqls = createTableSqls(info); String[] tableSqls = createTableSqls(info);
if (tableSqls != null) { //创建原始表 if (tableSqls != null) { //创建原始表
st = conn.createUpdateStatement(); stmt = conn.createUpdateStatement();
if (tableSqls.length == 1) { if (tableSqls.length == 1) {
st.execute(tableSqls[0]); stmt.execute(tableSqls[0]);
} else { } else {
for (String tableSql : tableSqls) { for (String tableSql : tableSqls) {
st.addBatch(tableSql); stmt.addBatch(tableSql);
} }
st.executeBatch(); stmt.executeBatch();
} }
st.close(); stmt.close();
//再执行一遍创建分表操作 //再执行一遍创建分表操作
st = conn.createUpdateStatement(); stmt = conn.createUpdateStatement();
for (String copySql : tableCopys) { for (String copySql : tableCopys) {
st.addBatch(copySql); stmt.addBatch(copySql);
} }
st.executeBatch(); stmt.executeBatch();
st.close(); stmt.close();
} }
} else { } else {
logger.log(Level.SEVERE, "create table2 " + tableCopys + " error", sqle2); logger.log(Level.SEVERE, "create table2 " + tableCopys + " error", sqle2);
@@ -651,16 +651,16 @@ public class DataJdbcSource extends AbstractDataSqlSource {
if (info.getTableStrategy() == null) { if (info.getTableStrategy() == null) {
String[] tableSqls = createTableSqls(info); String[] tableSqls = createTableSqls(info);
if (tableSqls != null) { if (tableSqls != null) {
Statement st = conn.createUpdateStatement(); Statement stmt = conn.createUpdateStatement();
if (tableSqls.length == 1) { if (tableSqls.length == 1) {
st.execute(tableSqls[0]); stmt.execute(tableSqls[0]);
} else { } else {
for (String tableSql : tableSqls) { for (String tableSql : tableSqls) {
st.addBatch(tableSql); stmt.addBatch(tableSql);
} }
st.executeBatch(); stmt.executeBatch();
} }
st.close(); stmt.close();
return 0; return 0;
} }
//单表结构不存在 //单表结构不存在
@@ -1148,16 +1148,16 @@ public class DataJdbcSource extends AbstractDataSqlSource {
String[] tableSqls = createTableSqls(info); String[] tableSqls = createTableSqls(info);
if (tableSqls != null) { if (tableSqls != null) {
try { try {
Statement st = conn.createUpdateStatement(); Statement stmt = conn.createUpdateStatement();
if (tableSqls.length == 1) { if (tableSqls.length == 1) {
st.execute(tableSqls[0]); stmt.execute(tableSqls[0]);
} else { } else {
for (String tableSql : tableSqls) { for (String tableSql : tableSqls) {
st.addBatch(tableSql); stmt.addBatch(tableSql);
} }
st.executeBatch(); stmt.executeBatch();
} }
st.close(); stmt.close();
} catch (SQLException e2) { } catch (SQLException e2) {
} }
} }
@@ -1380,16 +1380,16 @@ public class DataJdbcSource extends AbstractDataSqlSource {
String[] tableSqls = createTableSqls(info); String[] tableSqls = createTableSqls(info);
if (tableSqls != null) { if (tableSqls != null) {
try { try {
Statement st = conn.createUpdateStatement(); Statement stmt = conn.createUpdateStatement();
if (tableSqls.length == 1) { if (tableSqls.length == 1) {
st.execute(tableSqls[0]); stmt.execute(tableSqls[0]);
} else { } else {
for (String tableSql : tableSqls) { for (String tableSql : tableSqls) {
st.addBatch(tableSql); stmt.addBatch(tableSql);
} }
st.executeBatch(); stmt.executeBatch();
} }
st.close(); stmt.close();
} catch (SQLException e2) { } catch (SQLException e2) {
} }
} }
@@ -2530,12 +2530,12 @@ public class DataJdbcSource extends AbstractDataSqlSource {
if (logger.isLoggable(Level.FINEST)) { if (logger.isLoggable(Level.FINEST)) {
logger.finest("direct query sql=" + sql); logger.finest("direct query sql=" + sql);
} }
final Statement statement = conn.createQueryStatement(); final Statement stmt = conn.createQueryStatement();
//final PreparedStatement prestmt = conn.prepareStatement(sql); //final PreparedStatement prestmt = conn.prepareStatement(sql);
final ResultSet set = statement.executeQuery(sql);// prestmt.executeQuery(); final ResultSet set = stmt.executeQuery(sql);// prestmt.executeQuery();
V rs = handler.apply(createDataResultSet(null, set)); V rs = handler.apply(createDataResultSet(null, set));
set.close(); set.close();
statement.close(); stmt.close();
slowLog(s, sql); slowLog(s, sql);
return rs; return rs;
} catch (Exception ex) { } catch (Exception ex) {