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

!200 调整GP分布键判断逻辑

Browse Source 
* 调整GP分布键判断逻辑
This commit is contained in:
Donaghy
2024-08-03 09:00:29 +00:00
committed by inrgihc
parent 519b9b36fd
commit 320e761d4a
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
dbswitch-product/dbswitch-product-greenplum/src/main/java/com/gitee/dbswitch/product/greenplum/GreenplumMetadataQueryProvider.java
View File

@@ -32,22 +32,22 @@ public class GreenplumMetadataQueryProvider extends PostgresMetadataQueryProvide
@Override @Override
public void postAppendCreateTableSql(StringBuilder builder, String tblComment, List<String> primaryKeys, public void postAppendCreateTableSql(StringBuilder builder, String tblComment, List<String> primaryKeys,
SourceProperties tblProperties) { SourceProperties tblProperties) {
if (CollectionUtils.isEmpty(primaryKeys)) { List<String> distributed = determineDistributed(primaryKeys, tblProperties.getDistributedKeys());
if (null == distributed) {
return; return;
} }
// 分布键不为空，且需要满足是主键的子集 String dk = getPrimaryKeyAsString(distributed);
if (!CollectionUtils.isEmpty(tblProperties.getDistributedKeys()) && new HashSet<>(primaryKeys).containsAll( builder.append("\n DISTRIBUTED BY (").append(dk).append(")");
tblProperties.getDistributedKeys())) {
getPkOrDkAsString(builder, tblProperties.getDistributedKeys());
} else {
getPkOrDkAsString(builder, primaryKeys);
log.info("using primaryKey as distributed key");
}
} }
private void getPkOrDkAsString(StringBuilder builder, List<String> primaryKeys) { private List<String> determineDistributed(List<String> primaryKeys, List<String> distributedKeys) {
String pk = getPrimaryKeyAsString(primaryKeys); if (CollectionUtils.isEmpty(distributedKeys)) {
builder.append("\n DISTRIBUTED BY (").append(pk).append(")"); // 分布键为空,看是否有主键
return CollectionUtils.isEmpty(primaryKeys) ? null : primaryKeys;
}
// 分布键不为空,看是否是主键的子集,主键为空直接用分布键
return CollectionUtils.isEmpty(primaryKeys) || new HashSet<>(primaryKeys).containsAll(distributedKeys)
? distributedKeys : null;
} }
} }