dromara/dbswitch
1
0
Fork 0
mirror of https://gitee.com/dromara/dbswitch.git synced 2025-10-17 07:03:54 +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; package com.gitee.dbswitch.common.type;
import cn.hutool.core.util.StrUtil;
/** /**
* 处理大小写转换的枚举类 * 处理名称转换的枚举类
* *
* @author tang * @author tang
*/ */
@@ -30,33 +32,11 @@ public enum CaseConvertEnum {
/** /**
* 驼峰转下划线 * 驼峰转下划线
*/ */
SNAKE(s -> { SNAKE(StrUtil::toUnderlineCase),
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();
}),
/** /**
* 下划线转驼峰 * 下划线转驼峰
*/ */
CAMEL(s -> { CAMEL(StrUtil::toCamelCase);
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();
});
private Converter function; 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 @Override
public List<TableDescription> queryTableList(Connection connection, String schemaName) { public List<TableDescription> queryTableList(Connection connection, String schemaName) {
List<TableDescription> tableList = super.queryTableList(connection, schemaName); List<TableDescription> tableList = super.queryTableList(connection, schemaName);
Set<String> partitionedTable = getPartitionedTable(connection, schemaName, Set<String> partitionedTable = getPartitionedTable(connection, schemaName);
SHOW_SUB_PARTITIONED_TABLE);
return tableList.stream().filter(t -> !partitionedTable.contains(t.getTableName())) return tableList.stream().filter(t -> !partitionedTable.contains(t.getTableName()))
.collect(Collectors.toList()); .collect(Collectors.toList());
} }
@@ -277,8 +276,8 @@ public class PostgresMetadataQueryProvider extends AbstractMetadataProvider {
return results; return results;
} }
protected Set<String> getPartitionedTable(Connection connection, String schemaName, String sql) { protected Set<String> getPartitionedTable(Connection connection, String schemaName) {
String query = String.format(sql, schemaName); String query = String.format(SHOW_SUB_PARTITIONED_TABLE, schemaName);
Set<String> partitionedTable = new HashSet<>(); Set<String> partitionedTable = new HashSet<>();
try (Statement st = connection.createStatement()) { try (Statement st = connection.createStatement()) {
ResultSet resultSet = st.executeQuery(query); ResultSet resultSet = st.executeQuery(query);