database/doris
2
0
Fork 0
Code Issues Pull Requests Actions 3 Packages Projects Releases Wiki Activity

[improvement](jdbc catalog) Optimize Loop Performance by Caching isNebula Method Result (#24260)

Browse Source
This commit is contained in:
zy-kkk
2023-09-13 21:40:28 +08:00
committed by GitHub
parent 07dd6830e8
commit dbfacdc4af
1 changed files with 21 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
fe/be-java-extensions/jdbc-scanner/src/main/java/org/apache/doris/jdbc/JdbcExecutor.java
View File

@ -148,11 +148,15 @@ public class JdbcExecutor {
int columnCount = resultSetMetaData.getColumnCount();
resultColumnTypeNames = new ArrayList<>(columnCount);
block = new ArrayList<>(columnCount);
for (int i = 0; i < columnCount; ++i) {
if (!isNebula()) {
resultColumnTypeNames.add(resultSetMetaData.getColumnClassName(i + 1));
if (isNebula()) {
for (int i = 0; i < columnCount; ++i) {
block.add((Object[]) Array.newInstance(Object.class, batchSizeNum));
}
} else {
for (int i = 0; i < columnCount; ++i) {
resultColumnTypeNames.add(resultSetMetaData.getColumnClassName(i + 1));
block.add((Object[]) Array.newInstance(Object.class, batchSizeNum));
}
block.add((Object[]) Array.newInstance(Object.class, batchSizeNum));
}
return columnCount;
} catch (SQLException e) {
@ -367,16 +371,22 @@ public class JdbcExecutor {
try {
int columnCount = resultSetMetaData.getColumnCount();
curBlockRows = 0;
do {
for (int i = 0; i < columnCount; ++i) {
if (isNebula()) {
if (isNebula()) {
do {
for (int i = 0; i < columnCount; ++i) {
block.get(i)[curBlockRows] = UdfUtils.convertObject((ValueWrapper) resultSet.getObject(i + 1));
} else {
}
curBlockRows++;
} while (curBlockRows < batchSize && resultSet.next());
} else {
do {
for (int i = 0; i < columnCount; ++i) {
block.get(i)[curBlockRows] = resultSet.getObject(i + 1);
}
}
curBlockRows++;
} while (curBlockRows < batchSize && resultSet.next());
curBlockRows++;
} while (curBlockRows < batchSize && resultSet.next());
}
} catch (SQLException e) {
throw new UdfRuntimeException("get next block failed: ", e);
}