forked from sqlancer/sqlancer
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathDataFusionSelect.java
More file actions
43 lines (36 loc) · 1.61 KB
/
DataFusionSelect.java
File metadata and controls
43 lines (36 loc) · 1.61 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
package sqlancer.datafusion.ast;
import java.util.List;
import java.util.Optional;
import java.util.stream.Collectors;
import sqlancer.common.ast.SelectBase;
import sqlancer.common.ast.newast.Select;
import sqlancer.datafusion.DataFusionSchema.DataFusionColumn;
import sqlancer.datafusion.DataFusionSchema.DataFusionTable;
import sqlancer.datafusion.DataFusionToStringVisitor;
public class DataFusionSelect extends SelectBase<DataFusionExpression> implements DataFusionExpression,
Select<DataFusionJoin, DataFusionExpression, DataFusionTable, DataFusionColumn> {
public Optional<String> fetchColumnsString = Optional.empty(); // When available, override `fetchColumns` in base
/*
* If set fetch columns with string It will override `fetchColumns` in base class when
* `DataFusionToStringVisitor.asString()` is called
*
* This method can be helpful to mutate select in oracle checks: SELECT [expr] ... -> SELECT SUM[expr]
*/
public void setFetchColumnsString(String selectExpr) {
this.fetchColumnsString = Optional.of(selectExpr);
}
@Override
public void setJoinClauses(List<DataFusionJoin> joinStatements) {
List<DataFusionExpression> expressions = joinStatements.stream().map(e -> (DataFusionExpression) e)
.collect(Collectors.toList());
setJoinList(expressions);
}
@Override
public List<DataFusionJoin> getJoinClauses() {
return getJoinList().stream().map(e -> (DataFusionJoin) e).collect(Collectors.toList());
}
@Override
public String asString() {
return DataFusionToStringVisitor.asString(this);
}
}