2112A-ONE
/
muyu-source
6
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

编写测试连接的代码

master
冷调 2024-08-23 19:56:04 +08:00
parent fd2d3d2602
commit ad75e66218
10 changed files with 247 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
.idea/inspectionProfiles/Project_Default.xml
View File

@ -2,7 +2,13 @@
<profile version="1.0">
<option name="myName" value="Project Default" />
<inspection_tool class="AliAccessStaticViaInstance" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AliArrayNamingShouldHaveBracket" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AliControlFlowStatementWithoutBraces" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AliDeprecation" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AliEqualsAvoidNull" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AliLongLiteralsEndingWithLowercaseL" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AliMissingOverrideAnnotation" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AliWrapperTypeEquality" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AlibabaAbstractClassShouldStartWithAbstractNaming" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AlibabaAbstractMethodOrInterfaceMethodMustUseJavadoc" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AlibabaAvoidApacheBeanUtilsCopy" enabled="true" level="WARNING" enabled_by_default="true" />
@ -57,5 +63,6 @@
<inspection_tool class="AlibabaUnsupportedExceptionWithModifyAsList" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AlibabaUseQuietReferenceNotation" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="AlibabaUseRightCaseForDateFormat" enabled="true" level="WARNING" enabled_by_default="true" />
<inspection_tool class="MapOrSetKeyShouldOverrideHashCodeEquals" enabled="true" level="WARNING" enabled_by_default="true" />
</profile>
</component>

77
muyu-source-client/src/main/java/com/muyu/config/SourceClientConfig.java
View File

@ -1,9 +1,12 @@
package com.muyu.config;
import cn.hutool.core.math.Arrangement;
import com.muyu.common.core.domain.Result;
import com.muyu.connection.DataSourceConfig;
import com.muyu.source.domain.Children;
import com.muyu.source.domain.DataSource;
import com.muyu.source.domain.DataType;
import com.muyu.source.domain.model.DataModel;
import com.muyu.source.remote.RemoteChildrenService;
import com.muyu.source.remote.RemoteDataSourceService;
import com.muyu.source.remote.RemoteDataTypeService;
@ -11,6 +14,9 @@ import jakarta.annotation.Resource;
import org.springframework.context.annotation.ComponentScan;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
/**
* @author Lenovo
@ -22,48 +28,57 @@ import java.sql.*;
*/
@ComponentScan
public class SourceClientConfig {
@Resource
private RemoteDataTypeService remoteDataTypeService;
@Resource
private RemoteDataSourceService remoteDataSourceService;
@Resource
private RemoteChildrenService remoteChildrenService;
public void getDataModel(Long id){
//根据ID查询出数据源的数据
Result<DataSource> dataSourceResult =remoteDataSourceService.getDataSource(id);
DataSource dataSource = dataSourceResult.getData();
//根据ID查询出数据库类型的数据
Result<DataType> dataTypeResult = remoteDataTypeService.getDataType(dataSource.getDataType());
DataType dataType = dataTypeResult.getData();
//根据ID查询出数据源的数据库结构的数据
Result<Children> childrenResult =remoteChildrenService.getChildren(dataSource.getId());
public List<List<DataModel>> getDataModel(Long id) {
List<List<DataModel>> list = new ArrayList<>();
//根据ID查询出表结构的数据
Result<Children> childrenResult = remoteChildrenService.getChildren(id);
Children children = childrenResult.getData();
try {
Class.forName(dataType.getDriverManager());
} catch (ClassNotFoundException e) {
throw new RuntimeException(e);
//根据ID查询出数据源的数据
Result<DataSource> dataSourceResult = remoteDataSourceService.getDataSource(children.getAssetId());
DataSource dataSource = dataSourceResult.getData();
String sql = "";
if ("MySql".equals(dataSource.getDataType())) {
sql = "SELECT * FROM " + children.getName();
}
try {
//获取连接对象 拼接数据库地址
Connection connection = DriverManager.getConnection(dataType.getPrefix()+dataSource.getIp()+dataSource.getPort()
+"//"+dataSource.getDatabaseName()+"?"+dataSource.getConnectionParam(),dataSource.getUserName(),dataSource.getPassword());
//创建执行对象
Statement statement = connection.createStatement();
//拼接sql语句
ResultSet resultSet = statement.executeQuery("SELECT * FROM" + children.getName());
//处理返回结果
while (resultSet.next()){
System.out.println(resultSet.getString(1));
//调用连接池获取连接
Connection connection = DataSourceConfig.getDataSource(dataSource.getId());
PreparedStatement preparedStatement = connection.prepareStatement(sql);
ResultSet resultSet = preparedStatement.executeQuery();
ResultSetMetaData resultSetMetaData = resultSet.getMetaData();
while (resultSet.next()) {
List<DataModel> dataModels = new ArrayList<>();
for (int i = 1; i <= resultSetMetaData.getColumnCount(); i++) {
//获取列名 字段名 key
String columnName = resultSetMetaData.getColumnName(i);
//获取字段的值 value
Object value = resultSet.getObject(columnName);
//获取列的类型 字段的类型 type
String columnTypeName = resultSetMetaData.getColumnTypeName(i);
//获取列的标题 label
String columnLabel = resultSetMetaData.getColumnLabel(i);
// java类型
String columnClassName = resultSetMetaData.getColumnClassName(i);
DataModel dataModel = DataModel.builder()
.key(columnName)
.label(columnLabel)
.type(columnTypeName)
.value(value)
.build();
dataModels.add(dataModel);
}
list.add(dataModels);
}
//关闭资源
resultSet.close();
statement.close();
connection.close();
//归还连接
DataSourceConfig.returnConnection(connection);
} catch (SQLException e) {
throw new RuntimeException(e);
}
return list;
}
}

45
muyu-source-client/src/main/java/com/muyu/config/SourceClientRunner.java 100644
View File

@ -0,0 +1,45 @@
package com.muyu.config;
import com.muyu.common.core.domain.Result;
import com.muyu.connection.DataSourceConfig;
import com.muyu.source.domain.DataSource;
import com.muyu.source.domain.DataType;
import com.muyu.source.remote.RemoteChildrenService;
import com.muyu.source.remote.RemoteDataSourceService;
import com.muyu.source.remote.RemoteDataTypeService;
import jakarta.annotation.Resource;
import lombok.extern.log4j.Log4j2;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.ApplicationArguments;
import org.springframework.boot.ApplicationRunner;
import org.springframework.context.annotation.Configuration;
import java.util.List;
/**
* @author Lenovo
* @ Tool：IntelliJ IDEA
* @ Author：CHX
* @ Date：2024-08-23-9:45
* @ Version：1.0
* @ Description：初始化加载
*/
@Log4j2
@Configuration
public class SourceClientRunner implements ApplicationRunner {
@Resource
private RemoteDataSourceService remoteDataSourceService;
@Resource
private RemoteDataTypeService remoteDataTypeService;
@Override
public void run(ApplicationArguments args) throws Exception {
Result<List<DataSource>> sourceResult = remoteDataSourceService.getDataSourceList();
log.info(sourceResult);
List<DataSource> sourceList = sourceResult.getData();
sourceList.stream().forEach(source -> {
Result<DataType> dataTypeResult = remoteDataTypeService.getDataType(source.getDataType());
DataType dataType = dataTypeResult.getData();
DataSourceConfig.init(source, dataType);
});
}
}

39
muyu-source-common/src/main/java/com/muyu/source/domain/model/DataModel.java 100644
View File

@ -0,0 +1,39 @@
package com.muyu.source.domain.model;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import lombok.experimental.SuperBuilder;
/**
* @author Lenovo
* @ Tool：IntelliJ IDEA
* @ Author：CHX
* @ Date：2024-08-23-9:41
* @ Version：1.0
* @ Description：数据转换对象
*/
@Data
@SuperBuilder
@AllArgsConstructor
@NoArgsConstructor
public class DataModel {
/**
* 键
*/
private String key;
/**
* 标签
*/
private String label;
/**
* 类型
*/
private String type;
/**
* 值
*/
private Object value;
}

5
muyu-source-remote/src/main/java/com/muyu/source/remote/RemoteDataSourceService.java
View File

@ -8,6 +8,8 @@ import org.springframework.cloud.openfeign.FeignClient;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import java.util.List;
/**
* @author Lenovo
* @ Tool：IntelliJ IDEA
@ -23,4 +25,7 @@ public interface RemoteDataSourceService {
@GetMapping("/getDataSourceById/{id}")
Result<DataSource> getDataSource(@PathVariable Long id);
@GetMapping("/getDataSourceList")
Result<List<DataSource>> getDataSourceList();
}

7
muyu-source-remote/src/main/java/com/muyu/source/remote/factory/RemoteDataSourceFactory.java
View File

@ -5,6 +5,8 @@ import com.muyu.source.domain.DataSource;
import com.muyu.source.remote.RemoteDataSourceService;
import org.springframework.cloud.openfeign.FallbackFactory;
import java.util.List;
/**
* @author Lenovo
* @ Tool：IntelliJ IDEA
@ -21,6 +23,11 @@ public class RemoteDataSourceFactory implements FallbackFactory<RemoteDataSource
public Result<DataSource> getDataSource(Long id) {
return Result.error(cause.getMessage());
}
@Override
public Result<List<DataSource>> getDataSourceList() {
return Result.error(cause.getMessage());
}
};
}
}

32
muyu-source-server/src/main/java/com/muyu/source/controller/DataSourceController.java
View File

@ -1,5 +1,7 @@
package com.muyu.source.controller;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.dtflys.forest.springboot.annotation.ForestScannerRegister;
import com.muyu.common.core.domain.Result;
import com.muyu.common.core.utils.poi.ExcelUtil;
@ -45,6 +47,36 @@ public class DataSourceController extends BaseController {
@Autowired
private DataSourceService dataSourceService;
/**
* 测试连接
* @param dataSource 数据源
* @return 是否连接成功
*/
@PostMapping("/test")
public Result testConnection(@RequestBody DataSource dataSource){
Boolean b =dataSourceService.testConnection(dataSource);
if(b){
dataSourceService.update(new LambdaUpdateWrapper<>() {{
set(DataSource::getIsInit, "Y");
eq(DataSource::getId, dataSource.getId());
}});
return Result.success(b,"连接成功");
}
return Result.error("连接失败");
}
/**
* 获取数据源列表
* @return 数据源列表
*/
@GetMapping("/getDataSourceList")
public Result<List<DataSource>> getDataSourceList(){
List<DataSource> dataSourceList =dataSourceService.list();
return success(dataSourceList);
}
/**
* 查询数据源列表
*/

2
muyu-source-server/src/main/java/com/muyu/source/service/DataSourceService.java
View File

@ -39,5 +39,5 @@ public interface DataSourceService extends IService<DataSource> {
Boolean checkIdUnique(DataSource dataSource);
Boolean testConnection(DataSource dataSource);
}

14
muyu-source-server/src/main/java/com/muyu/source/service/DataTypeService.java 100644
View File

@ -0,0 +1,14 @@
package com.muyu.source.service;
import com.baomidou.mybatisplus.extension.service.IService;
import com.muyu.source.domain.DataType;
/**
* @ Tool：IntelliJ IDEA
* @ Author：CHX
* @ Date：2024-08-23-18:59
* @ Version：1.0
* @ Description：数据库类型业务层
*/
public interface DataTypeService extends IService<DataType> {
}

51
muyu-source-server/src/main/java/com/muyu/source/service/Impl/DataSourceServiceImpl.java
View File

@ -5,11 +5,18 @@ import com.baomidou.mybatisplus.core.toolkit.Assert;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.muyu.common.core.utils.StringUtils;
import com.muyu.source.domain.DataSource;
import com.muyu.source.domain.DataType;
import com.muyu.source.mapper.DataSourceMapper;
import com.muyu.source.service.DataSourceService;
import com.muyu.source.service.DataTypeService;
import jakarta.annotation.Resource;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import java.util.List;
import java.util.concurrent.ExecutorService;
import static java.util.concurrent.Executors.newFixedThreadPool;
/**
* @author Lenovo
@ -22,6 +29,10 @@ import java.util.List;
@Service
public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSource> implements DataSourceService {
@Resource
private DataTypeService dataTypeService;
private final ExecutorService executor = newFixedThreadPool(10);
/**
* 精确查询数据源
@ -79,5 +90,45 @@ public class DataSourceServiceImpl extends ServiceImpl<DataSourceMapper, DataSou
return this.count(queryWrapper) > 0;
}
/**
* 测试连接
* @param dataSource 数据源
* @return 结果
*/
@Override
public Boolean testConnection(DataSource dataSource) {
DataType dataType = dataTypeService.getOne(new LambdaQueryWrapper<>() {{
eq(DataType::getType, dataSource.getDataType());
}});
//
String jdbcUrl="";
boolean flag=false;
try{
//判空
if(dataType.getDriverManager()!=null && dataType.getPrefix()!=null){
if("MySql".equals(dataType.getType())){
//mysql拼接连接路径
jdbcUrl=dataType.getDriverManager()+dataType.getPrefix()+dataSource.getIp()+":"+dataSource.getPort()+"/"+dataSource.getDatabaseName()+"?"+dataSource.getConnectionParam();
}
flag=testConnection(dataType.getDriverManager(),jdbcUrl,dataSource.getUserName(),dataSource.getPassword());
}
return flag;
}catch (Exception e){
e.printStackTrace();
}
return flag;
}
private boolean testConnection(String driverManager, String jdbcUrl, String userName, String password) {
if(StringUtils.isNotEmpty(driverManager) && StringUtils.isNotEmpty(jdbcUrl) && StringUtils.isNotEmpty(userName) && StringUtils.isNotEmpty(password)){
try{
Class.forName(driverManager);
return true;
}catch (Exception e){
e.printStackTrace();
}
}
return false;
}
}