topdog/supersonic
1
0
Fork 0
mirror of https://github.com/tencentmusic/supersonic.git synced 2025-12-14 13:47:09 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

[improvement][headless] When validating permissions from SQL, handle Identifier types compatibility (#1844)

Browse Source
This commit is contained in:
lexluo09
2024-10-24 22:26:00 +08:00
committed by GitHub
parent 679630bfd5
commit bd68a7212a
1 changed files with 34 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
headless/server/src/main/java/com/tencent/supersonic/headless/server/utils/QueryStructUtils.java
View File

@@ -12,6 +12,7 @@ import com.tencent.supersonic.common.pojo.enums.DatePeriodEnum;
import com.tencent.supersonic.common.pojo.enums.TypeEnums;
import com.tencent.supersonic.common.util.DateModeUtils;
import com.tencent.supersonic.common.util.SqlFilterUtils;
import com.tencent.supersonic.headless.api.pojo.Identify;
import com.tencent.supersonic.headless.api.pojo.ItemDateFilter;
import com.tencent.supersonic.headless.api.pojo.MetaFilter;
import com.tencent.supersonic.headless.api.pojo.SchemaItem;
@@ -19,6 +20,7 @@ import com.tencent.supersonic.headless.api.pojo.request.QuerySqlReq;
import com.tencent.supersonic.headless.api.pojo.request.QueryStructReq;
import com.tencent.supersonic.headless.api.pojo.response.DimensionResp;
import com.tencent.supersonic.headless.api.pojo.response.MetricResp;
import com.tencent.supersonic.headless.api.pojo.response.ModelResp;
import com.tencent.supersonic.headless.api.pojo.response.SemanticSchemaResp;
import com.tencent.supersonic.headless.server.service.SchemaService;
import lombok.extern.slf4j.Slf4j;
@@ -143,6 +145,37 @@ public class QueryStructUtils {
return Lists.newArrayList();
}
private Set<Long> getModelIdsByIdentifiesFromSql(QuerySqlReq querySqlReq,
SemanticSchemaResp semanticSchemaResp) {
Set<String> resNameSet = getResName(querySqlReq);
Set<Long> modelIds = new HashSet<>();
if (semanticSchemaResp == null) {
return modelIds;
}
if (CollectionUtils.isEmpty(semanticSchemaResp.getModelResps())) {
return modelIds;
}
for (ModelResp modelResp : semanticSchemaResp.getModelResps()) {
if (modelHasMatchingIdentifier(modelResp, resNameSet)) {
modelIds.add(modelResp.getId());
}
}
return modelIds;
}
private boolean modelHasMatchingIdentifier(ModelResp modelResp, Set<String> resNameSet) {
if (modelResp.getModelDetail() == null) {
return false;
}
List<Identify> identifiers = modelResp.getModelDetail().getIdentifiers();
if (CollectionUtils.isEmpty(identifiers)) {
return false;
}
return identifiers.stream().anyMatch(identifier -> resNameSet.contains(identifier.getName())
|| resNameSet.contains(identifier.getBizName()));
}
private List<DimensionResp> getDimensionsFromSql(QuerySqlReq querySqlReq,
SemanticSchemaResp semanticSchemaResp) {
Set<String> resNameSet = getResName(querySqlReq);
@@ -162,6 +195,7 @@ public class QueryStructUtils {
modelIds.addAll(
dimensions.stream().map(DimensionResp::getModelId).collect(Collectors.toList()));
modelIds.addAll(metrics.stream().map(MetricResp::getModelId).collect(Collectors.toList()));
modelIds.addAll(getModelIdsByIdentifiesFromSql(querySqlReq, semanticSchemaResp));
return modelIds;
}