package sqlancer.tidb.ast; import java.util.List; import java.util.stream.Collectors; import sqlancer.common.ast.SelectBase; import sqlancer.common.ast.newast.Select; import sqlancer.tidb.TiDBSchema.TiDBColumn; import sqlancer.tidb.TiDBSchema.TiDBTable; import sqlancer.tidb.visitor.TiDBVisitor; public class TiDBSelect extends SelectBase implements TiDBExpression, Select { private TiDBExpression hint; public void setHint(TiDBExpression hint) { this.hint = hint; } public TiDBExpression getHint() { return hint; } @Override public void setJoinClauses(List joinStatements) { List expressions = joinStatements.stream().map(e -> (TiDBExpression) e) .collect(Collectors.toList()); setJoinList(expressions); } @Override public List getJoinClauses() { return getJoinList().stream().map(e -> (TiDBJoin) e).collect(Collectors.toList()); } @Override public String asString() { return TiDBVisitor.asString(this); } }