diff --git a/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java b/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java index b0a2924..42dc5b8 100644 --- a/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java +++ b/muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataValueServiceImpl.java @@ -178,9 +178,12 @@ public class DataValueServiceImpl extends ServiceImpl batchResults = new ArrayList<>(); String query = "SELECT * FROM " + tableName + " LIMIT ? OFFSET ?"; - try (Connection conn = mysqlPool.getConn(); - PreparedStatement preparedStatement = conn.prepareStatement(query)) { - preparedStatement.setInt(1, BATCH_SIZE); - preparedStatement.setInt(2, offset); + try (Connection conn = mysqlPool.getConn()) { + if (conn == null) { + throw new SQLException("Failed to get database connection"); + } + try (PreparedStatement preparedStatement = conn.prepareStatement(query)) { + preparedStatement.setInt(1, BATCH_SIZE); + preparedStatement.setInt(2, offset); - try (ResultSet resultSet = preparedStatement.executeQuery()) { - ResultSetMetaData metaData = resultSet.getMetaData(); - int columnCount = metaData.getColumnCount(); + try (ResultSet resultSet = preparedStatement.executeQuery()) { + ResultSetMetaData metaData = resultSet.getMetaData(); + int columnCount = metaData.getColumnCount(); - while (resultSet.next()) { - for (int i = 1; i <= columnCount; i++) { - String columnTypeName = metaData.getColumnTypeName(i); + while (resultSet.next()) { + for (int i = 1; i <= columnCount; i++) { + String columnTypeName = metaData.getColumnTypeName(i); - DatabaseMetaData metaDataColumns = conn.getMetaData(); - ResultSet columns = metaDataColumns.getColumns(null, null, metaData.getTableName(i), metaData.getColumnName(i)); - String remarks = null; - while (columns.next()) { - remarks = columns.getString("REMARKS"); + DatabaseMetaData metaDataColumns = conn.getMetaData(); + ResultSet columns = metaDataColumns.getColumns(null, null, metaData.getTableName(i), metaData.getColumnName(i)); + String remarks = null; + while (columns.next()) { + remarks = columns.getString("REMARKS"); + } + + DataValue build = DataValue.builder() + .key(metaData.getColumnName(i)) + .label(remarks) + .value(resultSet.getObject(i, DataType.convertType(columnTypeName))) + .type(DataType.convertTypeString(columnTypeName)) + .build(); + batchResults.add(build); } - - DataValue build = DataValue.builder() - .key(metaData.getColumnName(i)) - .label(remarks) - .value(resultSet.getObject(i, DataType.convertType(columnTypeName))) - .type(DataType.convertTypeString(columnTypeName)) - .build(); - batchResults.add(build); } } }