yimeng
/
ETL
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

获取数据库下的所有表名

ays
An Yong Shuai 2024-07-03 20:41:39 +08:00
parent 7a1dfe1f90
commit e4d2d5db6d
6 changed files with 40 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
.idea/httpRequests/http-client.cookies 100644
View File

0
.idea/httpRequests/http-requests-log.http 100644
View File

43
.idea/workspace.xml
View File

File diff suppressed because one or more lines are too long

17
etl-cleaning/src/main/java/com/etl/cleaning/controller/GetTableController.java
View File

@ -3,6 +3,8 @@ package com.etl.cleaning.controller;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.etl.cleaning.domian.pojo.DataPlace;
import com.etl.cleaning.domian.request.TableRequest;
import com.etl.cleaning.enums.EnumMsg;
import com.etl.cleaning.server.GetTableServerTest;
import com.etl.cleaning.service.PlaceService;
import com.etl.common.result.Result;
import org.springframework.web.bind.annotation.PostMapping;
@ -10,6 +12,9 @@ import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.validation.Valid;
import java.util.List;
/**
* 获取表名控制器
*/
@ -17,12 +22,12 @@ import org.springframework.web.bind.annotation.RestController;
@RequestMapping("/dataTable")
public class GetTableController {
private final PlaceService placeService;
public GetTableController(PlaceService placeService) {
this.placeService = placeService;
}
/**
* 获取所有表名
*
@ -30,12 +35,18 @@ public class GetTableController {
* @return
*/
@PostMapping("/getTableName")
public Result getTableName(@RequestBody TableRequest tableRequest) {
public Result getTableName(@RequestBody @Valid TableRequest tableRequest) {
LambdaUpdateWrapper<DataPlace> eqed = new LambdaUpdateWrapper<DataPlace>().eq(DataPlace::getName, tableRequest.getName()).or().eq(DataPlace::getTypeId, tableRequest.getTypeId());
DataPlace dataPlace = placeService.getOne(eqed);
if (dataPlace == null) {
throw new RuntimeException("您所选中的数据源不存在");
}
return Result.success();
String dataUrl = dataPlace.getDataUrl();
String dataName = dataPlace.getDataName();
String password = dataPlace.getPassword();
String username = dataPlace.getUsername();
String url = EnumMsg.URL_PRE.getMessage() + "://" + dataUrl + "/" + dataName;
List<String> tableNames = GetTableServerTest.getTableNames(url, dataName, username, password);
return Result.success(tableNames);
}
}

3
etl-cleaning/src/main/java/com/etl/cleaning/domian/request/TableRequest.java
View File

@ -2,6 +2,7 @@ package com.etl.cleaning.domian.request;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import javax.validation.constraints.NotBlank;
import javax.validation.constraints.NotNull;
@ -11,6 +12,7 @@ import javax.validation.constraints.NotNull;
*/
@Data
@AllArgsConstructor
@NoArgsConstructor
public class TableRequest {
/**
* 数据源类型ID
@ -25,5 +27,6 @@ public class TableRequest {
@NotBlank(message = "数据源描述不能为空")
private String name;
}

8
etl-cleaning/src/main/java/com/etl/cleaning/server/GetTableServerTest.java
View File

@ -1,6 +1,8 @@
package com.etl.cleaning.server;
import com.etl.cleaning.enums.EnumMsg;
import java.sql.*;
import java.util.ArrayList;
import java.util.List;
@ -9,14 +11,14 @@ import java.util.List;
* 获取数据库下的所有表名
*/
public class GetTableServerTest {
public static List<String> getTableNames(String url,String username,String password){
public static List<String> getTableNames(String url,String dataName,String username,String password){
ArrayList<String> tableNames = new ArrayList<>();
Connection connection = null;
try {
Class.forName("com.mysql.cj.jdbc.Driver");
Class.forName(EnumMsg.DRIVER_CLASS_NAME.getMessage());
connection = DriverManager.getConnection(url, username, password);
DatabaseMetaData metaData = connection.getMetaData();
ResultSet tables = metaData.getTables(null, null, "%", new String[]{"TABLE"});
ResultSet tables = metaData.getTables(dataName, null, "%", new String[]{"TABLE"});
while (tables.next()){
tableNames.add(tables.getString(3));
}