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)(semantic) perfect dsl permission (#161)

Browse Source
This commit is contained in:
mainmain
2023-09-27 21:01:44 +08:00
committed by GitHub
parent 6047c787b3
commit e688422ec3
10 changed files with 233 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
common/src/main/java/com/tencent/supersonic/common/util/jsqlparser/FieldlValueReplaceVisitor.java
View File

@@ -2,10 +2,19 @@ package com.tencent.supersonic.common.util.jsqlparser;
import java.util.Map;
import java.util.Objects;
import net.sf.jsqlparser.expression.Expression;
import net.sf.jsqlparser.expression.ExpressionVisitorAdapter;
import net.sf.jsqlparser.expression.StringValue;
import net.sf.jsqlparser.expression.LongValue;
import net.sf.jsqlparser.expression.DoubleValue;
import net.sf.jsqlparser.expression.operators.relational.GreaterThanEquals;
import net.sf.jsqlparser.expression.operators.relational.GreaterThan;
import net.sf.jsqlparser.expression.operators.relational.MinorThan;
import net.sf.jsqlparser.expression.operators.relational.MinorThanEquals;
import net.sf.jsqlparser.expression.operators.relational.ComparisonOperator;
import net.sf.jsqlparser.expression.operators.relational.EqualsTo;
import net.sf.jsqlparser.schema.Column;
import org.apache.commons.lang3.StringUtils;
import org.springframework.util.CollectionUtils;
@@ -51,4 +60,64 @@ public class FieldlValueReplaceVisitor extends ExpressionVisitorAdapter {
}
}
}
public void visit(GreaterThan expr) {
replaceComparisonExpression(expr);
}
public void visit(GreaterThanEquals expr) {
replaceComparisonExpression(expr);
}
public void visit(MinorThanEquals expr) {
replaceComparisonExpression(expr);
}
public void visit(MinorThan expr) {
replaceComparisonExpression(expr);
}
public <T extends Expression> void replaceComparisonExpression(T expression) {
if ((expression instanceof GreaterThanEquals) || (expression instanceof GreaterThan)
|| (expression instanceof MinorThanEquals) || (expression instanceof MinorThan)) {
Expression leftExpression = ((ComparisonOperator) expression).getLeftExpression();
Expression rightExpression = ((ComparisonOperator) expression).getRightExpression();
if (!(leftExpression instanceof Column)) {
return;
}
if (CollectionUtils.isEmpty(filedNameToValueMap)) {
return;
}
if (Objects.isNull(rightExpression) || Objects.isNull(leftExpression)) {
return;
}
Column leftColumnName = (Column) leftExpression;
String columnName = leftColumnName.getColumnName();
if (StringUtils.isEmpty(columnName)) {
return;
}
Map<String, String> valueMap = filedNameToValueMap.get(columnName);
if (Objects.isNull(valueMap) || valueMap.isEmpty()) {
return;
}
for (String oriValue : valueMap.keySet()) {
String replaceValue = valueMap.get(oriValue);
if (StringUtils.isNotEmpty(replaceValue)) {
if (rightExpression instanceof LongValue) {
LongValue rightStringValue = (LongValue) rightExpression;
rightStringValue.setValue(Long.parseLong(replaceValue));
}
if (rightExpression instanceof DoubleValue) {
DoubleValue rightStringValue = (DoubleValue) rightExpression;
rightStringValue.setValue(Double.parseDouble(replaceValue));
}
if (rightExpression instanceof StringValue) {
StringValue rightStringValue = (StringValue) rightExpression;
rightStringValue.setValue(replaceValue);
}
}
}
}
}
}