dromara/dbswitch
1
0
Fork 0
mirror of https://gitee.com/dromara/dbswitch.git synced 2025-10-14 22:00:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

代码简单调整

Browse Source
This commit is contained in:
inrgihc
2024-07-31 21:57:45 +08:00
parent 86cd6b7847
commit 3501225503
2 changed files with 8 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
dbswitch-common/src/main/java/com/gitee/dbswitch/common/type/CaseConvertEnum.java
View File

@@ -9,8 +9,10 @@
/////////////////////////////////////////////////////////////
package com.gitee.dbswitch.common.type;
import cn.hutool.core.util.StrUtil;
/**
* 处理大小写转换的枚举类
* 处理名称转换的枚举类
*
* @author tang
*/
@@ -30,33 +32,11 @@ public enum CaseConvertEnum {
/**
* 驼峰转下划线
*/
SNAKE(s -> {
StringBuilder snakeCaseString = new StringBuilder(s);
for (int i = 1; i < snakeCaseString.length(); i++) {
if (Character.isUpperCase(snakeCaseString.charAt(i))) {
snakeCaseString.insert(i, '_');
i++;
}
}
return snakeCaseString.toString().toLowerCase();
}),
SNAKE(StrUtil::toUnderlineCase),
/**
* 下划线转驼峰
*/
CAMEL(s -> {
String[] parts = s.split("_");
StringBuilder camelCaseString = new StringBuilder();
for (int i = 0; i < parts.length; i++) {
String part = parts[i];
if (i == 0) {
camelCaseString.append(part.toLowerCase()); // 第一个单词保持全小写
} else {
camelCaseString.append(Character.toUpperCase(part.charAt(0))); // 其他单词首字母大写
camelCaseString.append(part.substring(1).toLowerCase()); // 其他部分小写
}
}
return camelCaseString.toString();
});
CAMEL(StrUtil::toCamelCase);
private Converter function;

7
dbswitch-product/dbswitch-product-postgresql/src/main/java/com/gitee/dbswitch/product/postgresql/PostgresMetadataQueryProvider.java
View File

@@ -69,8 +69,7 @@ public class PostgresMetadataQueryProvider extends AbstractMetadataProvider {
@Override
public List<TableDescription> queryTableList(Connection connection, String schemaName) {
List<TableDescription> tableList = super.queryTableList(connection, schemaName);
Set<String> partitionedTable = getPartitionedTable(connection, schemaName,
SHOW_SUB_PARTITIONED_TABLE);
Set<String> partitionedTable = getPartitionedTable(connection, schemaName);
return tableList.stream().filter(t -> !partitionedTable.contains(t.getTableName()))
.collect(Collectors.toList());
}
@@ -277,8 +276,8 @@ public class PostgresMetadataQueryProvider extends AbstractMetadataProvider {
return results;
}
protected Set<String> getPartitionedTable(Connection connection, String schemaName, String sql) {
String query = String.format(sql, schemaName);
protected Set<String> getPartitionedTable(Connection connection, String schemaName) {
String query = String.format(SHOW_SUB_PARTITIONED_TABLE, schemaName);
Set<String> partitionedTable = new HashSet<>();
try (Statement st = connection.createStatement()) {
ResultSet resultSet = st.executeQuery(query);