修改查询语句
parent
6d0413a76c
commit
749f86cec7
|
|
@ -1,13 +1,13 @@
|
|||
package com.muyu.cloud.etl.mysql;
|
||||
|
||||
|
||||
import com.muyu.Hikari.HikariPool;
|
||||
import com.muyu.cloud.etl.service.SourceService;
|
||||
import com.muyu.common.core.utils.SpringUtils;
|
||||
import com.muyu.data.base.BaseDataAbsSource;
|
||||
import com.muyu.domain.DataValue;
|
||||
import com.muyu.domain.Source;
|
||||
import com.muyu.domain.enums.DataType;
|
||||
import com.zaxxer.hikari.HikariConfig;
|
||||
import com.zaxxer.hikari.HikariDataSource;
|
||||
import lombok.extern.log4j.Log4j2;
|
||||
import org.springframework.stereotype.Component;
|
||||
|
|
@ -257,43 +257,139 @@ public class MySqlDataSource extends BaseDataAbsSource {
|
|||
//第一种方法
|
||||
|
||||
|
||||
// @Override
|
||||
// public DataValue[][] getRows() {
|
||||
// MySqlQuery query = getQuery();
|
||||
// Integer one = Math.toIntExact(query.getOne());
|
||||
// Integer two = query.getTwo();
|
||||
// String sql = query.getSql();
|
||||
// Long dataSourceId = query.getDataSourceId();
|
||||
// ConcurrentHashMap<Integer, DataValue> map = new ConcurrentHashMap<>();
|
||||
// Source dataSources = sourceService.getById(dataSourceId);
|
||||
//
|
||||
// HikariDataSource hikariDataSource = HikariPool.getHikariDataSource(dataSources);
|
||||
//
|
||||
// Connection conn = null;
|
||||
// PreparedStatement preparedStatement = null;
|
||||
// ResultSet resultSet = null;
|
||||
//
|
||||
// try {
|
||||
// conn = hikariDataSource.getConnection();
|
||||
// preparedStatement = conn.prepareStatement(sql);
|
||||
// resultSet = preparedStatement.executeQuery();
|
||||
// ResultSetMetaData metaData = resultSet.getMetaData();
|
||||
// int columnCount = metaData.getColumnCount();
|
||||
// int c = 1;
|
||||
//
|
||||
// DataValue[][] dataValues = new DataValue[one][two];
|
||||
//
|
||||
// while (resultSet.next()) {
|
||||
// for (int i = 1; i <= columnCount; i++) {
|
||||
// if (resultSet.isFirst()) {
|
||||
// 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");
|
||||
// }
|
||||
// DataValue build = DataValue.builder()
|
||||
// .key(metaData.getColumnName(i))
|
||||
// .label(remarks)
|
||||
// .value(resultSet.getObject(i, DataType.convertType(columnTypeName)))
|
||||
// .type(DataType.findBySqlType(columnTypeName))
|
||||
// .build();
|
||||
// map.put(i, build);
|
||||
// if (c <= one && i <= columnCount) {
|
||||
// dataValues[c - 1][i - 1] = build;
|
||||
// }
|
||||
// } else {
|
||||
// DataValue build = DataValue.builder()
|
||||
// .key(metaData.getColumnName(i))
|
||||
// .label(map.get(i).getLabel())
|
||||
// .value(resultSet.getObject(i, map.get(i).getType().getTargetType()))
|
||||
// .type(map.get(i).getType())
|
||||
// .build();
|
||||
// if (c <= one && i <= columnCount) {
|
||||
// dataValues[c - 1][i - 1] = build;
|
||||
// }
|
||||
// }
|
||||
// }
|
||||
// if (c <= one) {
|
||||
// c++;
|
||||
// }
|
||||
// }
|
||||
// log.info("返回数据{}",dataValues.length);
|
||||
// return dataValues;
|
||||
// } catch (SQLException e) {
|
||||
// throw new RuntimeException(e);
|
||||
// } finally {
|
||||
// close(conn, preparedStatement, resultSet);
|
||||
// }
|
||||
// }
|
||||
//
|
||||
// private static void close(Connection conn, PreparedStatement preparedStatement, ResultSet resultSet) {
|
||||
// try {
|
||||
// if (resultSet != null) {
|
||||
// resultSet.close();
|
||||
// }
|
||||
// if (preparedStatement != null) {
|
||||
// preparedStatement.close();
|
||||
// }
|
||||
// if (conn != null) {
|
||||
// conn.close();
|
||||
// }
|
||||
// } catch (SQLException e) {
|
||||
//// log.log(Level.SEVERE, "Error closing resources", e);
|
||||
// }
|
||||
// }
|
||||
|
||||
|
||||
//壮法
|
||||
@Override
|
||||
public DataValue[][] getRows() {
|
||||
public DataValue[][] getRows () {
|
||||
|
||||
MySqlQuery query = getQuery();
|
||||
Integer one = Math.toIntExact(query.getOne());
|
||||
Integer two = query.getTwo();
|
||||
String sql = query.getSql();
|
||||
Long dataSourceId = query.getDataSourceId();
|
||||
ConcurrentHashMap<Integer, DataValue> map = new ConcurrentHashMap<>();
|
||||
|
||||
Source dataSources = sourceService.getById(dataSourceId);
|
||||
|
||||
HikariDataSource hikariDataSource = HikariPool.getHikariDataSource(dataSources);
|
||||
HikariConfig hikariConfig = new HikariConfig();
|
||||
hikariConfig.setPoolName("HikariCP 连接池");
|
||||
hikariConfig.setDriverClassName(dataSources.getDriverName());
|
||||
hikariConfig.setJdbcUrl(dataSources.getUrl()); // 修正这里
|
||||
hikariConfig.setUsername(dataSources.getUsername());
|
||||
hikariConfig.setPassword(dataSources.getPassword());
|
||||
hikariConfig.setMinimumIdle(2);
|
||||
hikariConfig.setMaximumPoolSize(10);
|
||||
|
||||
Connection conn = null;
|
||||
PreparedStatement preparedStatement = null;
|
||||
ResultSet resultSet = null;
|
||||
DataValue[][] dataValues = new DataValue[one][two];
|
||||
|
||||
HikariDataSource hikariDataSource = new HikariDataSource(hikariConfig);
|
||||
try {
|
||||
conn = hikariDataSource.getConnection();
|
||||
preparedStatement = conn.prepareStatement(sql);
|
||||
resultSet = preparedStatement.executeQuery();
|
||||
Connection conn = hikariDataSource.getConnection();
|
||||
PreparedStatement preparedStatement = conn.prepareStatement(sql);
|
||||
ResultSet resultSet = preparedStatement.executeQuery();
|
||||
ResultSetMetaData metaData = resultSet.getMetaData();
|
||||
int columnCount = metaData.getColumnCount();
|
||||
int c = 1;
|
||||
|
||||
DataValue[][] dataValues = new DataValue[one][two];
|
||||
|
||||
while (resultSet.next()) {
|
||||
int c = 0;
|
||||
while (resultSet.next()){
|
||||
for (int i = 1; i <= columnCount; i++) {
|
||||
log.info("喜欢次数为===={}",i);
|
||||
if (resultSet.isFirst()) {
|
||||
if (resultSet.isFirst()){
|
||||
String columnTypeName = metaData.getColumnTypeName(i);
|
||||
DatabaseMetaData metaDataColumns = conn.getMetaData();
|
||||
ResultSet columns = metaDataColumns.getColumns(null, null, metaData.getTableName(i), metaData.getColumnName(i));
|
||||
String remarks = null;
|
||||
String remarks =null;
|
||||
|
||||
while (columns.next()) {
|
||||
while (columns.next()){
|
||||
remarks = columns.getString("REMARKS");
|
||||
|
||||
log.info("字段备注:"+remarks);
|
||||
}
|
||||
DataValue build = DataValue.builder()
|
||||
.key(metaData.getColumnName(i))
|
||||
|
|
@ -301,49 +397,34 @@ public class MySqlDataSource extends BaseDataAbsSource {
|
|||
.value(resultSet.getObject(i, DataType.convertType(columnTypeName)))
|
||||
.type(DataType.findBySqlType(columnTypeName))
|
||||
.build();
|
||||
map.put(i, build);
|
||||
if (c <= one && i <= columnCount) {
|
||||
dataValues[c - 1][i - 1] = build;
|
||||
}
|
||||
} else {
|
||||
map.put(i,build);
|
||||
dataValues[c][i-1]=build;
|
||||
}else {
|
||||
DataValue build = DataValue.builder()
|
||||
.key(metaData.getColumnName(i))
|
||||
.label(map.get(i).getLabel())
|
||||
.value(resultSet.getObject(i, map.get(i).getType().getTargetType()))
|
||||
.type(map.get(i).getType())
|
||||
.build();
|
||||
if (c <= one && i <= columnCount) {
|
||||
dataValues[c - 1][i - 1] = build;
|
||||
}
|
||||
dataValues[c][i-1]=build;
|
||||
}
|
||||
}
|
||||
if (c <= one) {
|
||||
c++;
|
||||
}
|
||||
c++;
|
||||
}
|
||||
log.info("返回数据{}",dataValues.length);
|
||||
return dataValues;
|
||||
|
||||
} catch (SQLException e) {
|
||||
throw new RuntimeException(e);
|
||||
} finally {
|
||||
close(conn, preparedStatement, resultSet);
|
||||
}finally {
|
||||
close(hikariDataSource); // 关闭数据源
|
||||
}
|
||||
|
||||
return dataValues;
|
||||
}
|
||||
private static void close(HikariDataSource dataSource) {
|
||||
if (dataSource != null) {
|
||||
dataSource.close();
|
||||
}
|
||||
}
|
||||
|
||||
private static void close(Connection conn, PreparedStatement preparedStatement, ResultSet resultSet) {
|
||||
try {
|
||||
if (resultSet != null) {
|
||||
resultSet.close();
|
||||
}
|
||||
if (preparedStatement != null) {
|
||||
preparedStatement.close();
|
||||
}
|
||||
if (conn != null) {
|
||||
conn.close();
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
// log.log(Level.SEVERE, "Error closing resources", e);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in New Issue