dromara/dbswitch
1
0
Fork 0
mirror of https://gitee.com/dromara/dbswitch.git synced 2025-09-25 21:28:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

!69 代码优化

Browse Source 
* 代码优化
This commit is contained in:
inrgihc
2022-01-29 09:58:03 +00:00
parent f153b643aa
commit ea37936497
3 changed files with 11 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
dbswitch-dbwriter/src/main/java/com/gitee/dbswitch/dbwriter/AbstractDatabaseWriter.java
View File

@@ -13,6 +13,7 @@ import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.Statement;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.List;
@@ -113,15 +114,11 @@ public abstract class AbstractDatabaseWriter implements IDatabaseWriter {
try {
int[] affects = jdbcTemplate.batchUpdate(sqlInsert, recordValues, argTypes);
int affectCount = 0;
for (int i : affects) {
affectCount += i;
}
int affectCount = Arrays.stream(affects).sum();
recordValues.clear();
transactionManager.commit(status);
if (log.isDebugEnabled()) {
log.debug("{} insert write data affect count:{}", getDatabaseProductName(), affectCount);
log.debug("{} insert data affect count: {}", getDatabaseProductName(), affectCount);
}
return affectCount;
} catch (TransactionException e) {

9
dbswitch-dbwriter/src/main/java/com/gitee/dbswitch/dbwriter/mssql/SqlServerWriterImpl.java
View File

@@ -11,6 +11,7 @@ package com.gitee.dbswitch.dbwriter.mssql;
import com.gitee.dbswitch.dbwriter.AbstractDatabaseWriter;
import com.gitee.dbswitch.dbwriter.IDatabaseWriter;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import javax.sql.DataSource;
@@ -68,15 +69,11 @@ public class SqlServerWriterImpl extends AbstractDatabaseWriter implements IData
try {
int[] affects = jdbcTemplate.batchUpdate(sqlInsert, recordValues, argTypes);
int affectCount = 0;
for (int i : affects) {
affectCount += i;
}
int affectCount = Arrays.stream(affects).sum();
recordValues.clear();
transactionManager.commit(status);
if (log.isDebugEnabled()) {
log.debug("SQL Server insert data affect count: {}", affectCount);
log.debug("{} insert data affect count: {}", getDatabaseProductName(), affectCount);
}
return affectCount;
} catch (TransactionException e) {

8
dbswitch-dbwriter/src/main/java/com/gitee/dbswitch/dbwriter/mysql/MySqlWriterImpl.java
View File

@@ -11,6 +11,7 @@ package com.gitee.dbswitch.dbwriter.mysql;
import com.gitee.dbswitch.dbwriter.AbstractDatabaseWriter;
import com.gitee.dbswitch.dbwriter.IDatabaseWriter;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import javax.sql.DataSource;
@@ -73,9 +74,10 @@ public class MySqlWriterImpl extends AbstractDatabaseWriter implements IDatabase
Integer ret = transactionTemplate.execute((TransactionStatus transactionStatus) -> {
try {
int[] affects = jdbcTemplate.batchUpdate(sqlInsert, recordValues, argTypes);
int affectCount = 0;
for (int i : affects) {
affectCount += i;
int affectCount = Arrays.stream(affects).sum();
recordValues.clear();
if (log.isDebugEnabled()) {
log.debug("{} insert data affect count: {}", getDatabaseProductName(), affectCount);
}
return affectCount;
} catch (Throwable t) {