use of org.apache.carbondata.core.scan.expression.conditional.EqualToExpression in project carbondata by apache.
the class CarbonInputFormat_FT method testGetFilteredSplits.
@Test
public void testGetFilteredSplits() throws Exception {
CarbonInputFormat carbonInputFormat = new CarbonInputFormat();
JobConf jobConf = new JobConf(new Configuration());
Job job = Job.getInstance(jobConf);
FileInputFormat.addInputPath(job, new Path("/opt/carbonstore/db/table1"));
job.getConfiguration().set(CarbonInputFormat.INPUT_SEGMENT_NUMBERS, "1,2");
Expression expression = new EqualToExpression(new ColumnExpression("c1", DataType.STRING), new LiteralExpression("a", DataType.STRING));
CarbonInputFormat.setFilterPredicates(job.getConfiguration(), expression);
List splits = carbonInputFormat.getSplits(job);
Assert.assertTrue(splits != null);
Assert.assertTrue(!splits.isEmpty());
}
use of org.apache.carbondata.core.scan.expression.conditional.EqualToExpression in project carbondata by apache.
the class FilterExpressionProcessor method createPartitionFilterTree.
/**
* create partition filter by basing on pushed-down filter
* @param expressionTree
* @param partitionInfo
* @return
*/
private PartitionFilterIntf createPartitionFilterTree(Expression expressionTree, PartitionInfo partitionInfo) {
ExpressionType filterExpressionType = expressionTree.getFilterExpressionType();
String partitionColumnName = partitionInfo.getColumnSchemaList().get(0).getColumnName();
BinaryExpression currentExpression = null;
ColumnExpression left = null;
switch(filterExpressionType) {
case OR:
currentExpression = (BinaryExpression) expressionTree;
return new OrFilterImpl(createPartitionFilterTree(currentExpression.getLeft(), partitionInfo), createPartitionFilterTree(currentExpression.getRight(), partitionInfo));
case RANGE:
case AND:
currentExpression = (BinaryExpression) expressionTree;
return new AndFilterImpl(createPartitionFilterTree(currentExpression.getLeft(), partitionInfo), createPartitionFilterTree(currentExpression.getRight(), partitionInfo));
case EQUALS:
EqualToExpression equalTo = (EqualToExpression) expressionTree;
if (equalTo.getLeft() instanceof ColumnExpression && equalTo.getRight() instanceof LiteralExpression) {
left = (ColumnExpression) equalTo.getLeft();
if (partitionColumnName.equals(left.getCarbonColumn().getColName())) {
return new EqualToFilterImpl(equalTo, partitionInfo);
}
}
return new KeepAllPartitionFilterImpl();
case IN:
InExpression in = (InExpression) expressionTree;
if (in.getLeft() instanceof ColumnExpression && in.getRight() instanceof ListExpression) {
left = (ColumnExpression) in.getLeft();
if (partitionColumnName.equals(left.getCarbonColumn().getColName())) {
return new InFilterImpl(in, partitionInfo);
}
}
return new KeepAllPartitionFilterImpl();
case FALSE:
return new PruneAllPartitionFilterImpl();
case TRUE:
return new KeepAllPartitionFilterImpl();
case GREATERTHAN:
GreaterThanExpression greaterThan = (GreaterThanExpression) expressionTree;
if (greaterThan.getLeft() instanceof ColumnExpression && greaterThan.getRight() instanceof LiteralExpression) {
left = (ColumnExpression) greaterThan.getLeft();
if (partitionColumnName.equals(left.getCarbonColumn().getColName())) {
return new RangeFilterImpl((LiteralExpression) greaterThan.getRight(), true, false, partitionInfo);
}
}
return new KeepAllPartitionFilterImpl();
case GREATERTHAN_EQUALTO:
GreaterThanEqualToExpression greaterThanEqualTo = (GreaterThanEqualToExpression) expressionTree;
if (greaterThanEqualTo.getLeft() instanceof ColumnExpression && greaterThanEqualTo.getRight() instanceof LiteralExpression) {
left = (ColumnExpression) greaterThanEqualTo.getLeft();
if (partitionColumnName.equals(left.getCarbonColumn().getColName())) {
return new RangeFilterImpl((LiteralExpression) greaterThanEqualTo.getRight(), true, true, partitionInfo);
}
}
return new KeepAllPartitionFilterImpl();
case LESSTHAN:
LessThanExpression lessThan = (LessThanExpression) expressionTree;
if (lessThan.getLeft() instanceof ColumnExpression && lessThan.getRight() instanceof LiteralExpression) {
left = (ColumnExpression) lessThan.getLeft();
if (partitionColumnName.equals(left.getCarbonColumn().getColName())) {
return new RangeFilterImpl((LiteralExpression) lessThan.getRight(), false, false, partitionInfo);
}
}
return new KeepAllPartitionFilterImpl();
case LESSTHAN_EQUALTO:
LessThanEqualToExpression lessThanEqualTo = (LessThanEqualToExpression) expressionTree;
if (lessThanEqualTo.getLeft() instanceof ColumnExpression && lessThanEqualTo.getRight() instanceof LiteralExpression) {
left = (ColumnExpression) lessThanEqualTo.getLeft();
if (partitionColumnName.equals(left.getCarbonColumn().getColName())) {
return new RangeFilterImpl((LiteralExpression) lessThanEqualTo.getRight(), false, true, partitionInfo);
}
}
return new KeepAllPartitionFilterImpl();
case NOT_IN:
case NOT_EQUALS:
default:
return new KeepAllPartitionFilterImpl();
}
}
use of org.apache.carbondata.core.scan.expression.conditional.EqualToExpression in project carbondata by apache.
the class NoDictionaryTypeVisitor method populateFilterResolvedInfo.
/**
* Visitor Method will update the filter related details in visitableObj, For no dictionary
* type columns the filter members will resolved directly, no need to look up in dictionary
* since it will not be part of dictionary, directly the actual data can be converted as
* byte[] and can be set. this type of encoding is effective when the particular column
* is having very high cardinality.
*
* @param visitableObj
* @param metadata
* @throws FilterUnsupportedException,if exception occurs while evaluating
* filter models.
*/
public void populateFilterResolvedInfo(ColumnResolvedFilterInfo visitableObj, FilterResolverMetadata metadata) throws FilterUnsupportedException {
if (visitableObj instanceof DimColumnResolvedFilterInfo) {
DimColumnResolvedFilterInfo resolveDimension = (DimColumnResolvedFilterInfo) visitableObj;
ColumnFilterInfo resolvedFilterObject = null;
List<String> evaluateResultListFinal = new ArrayList<>(1);
try {
// handling for is null case scenarios
if (metadata.getExpression() instanceof EqualToExpression) {
EqualToExpression expression = (EqualToExpression) metadata.getExpression();
if (expression.isNull) {
evaluateResultListFinal = new ArrayList<>(1);
evaluateResultListFinal.add(CarbonCommonConstants.MEMBER_DEFAULT_VAL);
}
} else {
evaluateResultListFinal = metadata.getExpression().evaluate(null).getListAsString();
}
// displaying the report as per hive compatibility.
if (!metadata.isIncludeFilter() && !evaluateResultListFinal.contains(CarbonCommonConstants.MEMBER_DEFAULT_VAL)) {
evaluateResultListFinal.add(CarbonCommonConstants.MEMBER_DEFAULT_VAL);
}
} catch (FilterIllegalMemberException e) {
throw new FilterUnsupportedException(e);
}
resolvedFilterObject = FilterUtil.getNoDictionaryValKeyMemberForFilter(evaluateResultListFinal, metadata.isIncludeFilter(), metadata.getColumnExpression().getDataType());
resolveDimension.setFilterValues(resolvedFilterObject);
}
}
use of org.apache.carbondata.core.scan.expression.conditional.EqualToExpression in project carbondata by apache.
the class MeasureColumnVisitor method populateFilterResolvedInfo.
/**
* Visitor Method will update the filter related details in visitableObj, For no dictionary
* type columns the filter members will resolved directly, no need to look up in dictionary
* since it will not be part of dictionary, directly the actual data can be converted as
* byte[] and can be set. this type of encoding is effective when the particular column
* is having very high cardinality.
*
* @param visitableObj
* @param metadata
* @throws FilterUnsupportedException,if exception occurs while evaluating
* filter models.
*/
public void populateFilterResolvedInfo(ColumnResolvedFilterInfo visitableObj, FilterResolverMetadata metadata) throws FilterUnsupportedException {
if (visitableObj instanceof MeasureColumnResolvedFilterInfo) {
MeasureColumnResolvedFilterInfo resolveDimension = (MeasureColumnResolvedFilterInfo) visitableObj;
ColumnFilterInfo resolvedFilterObject = null;
List<String> evaluateResultListFinal = new ArrayList<>(1);
try {
// handling for is null case scenarios
if (metadata.getExpression() instanceof EqualToExpression) {
EqualToExpression expression = (EqualToExpression) metadata.getExpression();
if (expression.isNull) {
evaluateResultListFinal = new ArrayList<>(1);
evaluateResultListFinal.add(CarbonCommonConstants.MEMBER_DEFAULT_VAL);
}
} else {
evaluateResultListFinal = metadata.getExpression().evaluate(null).getListAsString();
}
// displaying the report as per hive compatibility.
if (!metadata.isIncludeFilter() && !evaluateResultListFinal.contains(CarbonCommonConstants.MEMBER_DEFAULT_VAL)) {
evaluateResultListFinal.add(CarbonCommonConstants.MEMBER_DEFAULT_VAL);
}
} catch (FilterIllegalMemberException e) {
throw new FilterUnsupportedException(e);
}
resolvedFilterObject = FilterUtil.getMeasureValKeyMemberForFilter(evaluateResultListFinal, metadata.isIncludeFilter(), metadata.getColumnExpression().getDataType(), resolveDimension.getMeasure());
resolveDimension.setFilterValues(resolvedFilterObject);
}
}
use of org.apache.carbondata.core.scan.expression.conditional.EqualToExpression in project carbondata by apache.
the class ObjectSerializationUtilTest method testConvertStringToObject.
@Test
public void testConvertStringToObject() throws Exception {
Expression expression = new EqualToExpression(new ColumnExpression("c1", DataTypes.STRING), new LiteralExpression("a", DataTypes.STRING));
String string = ObjectSerializationUtil.convertObjectToString(expression);
Assert.assertTrue(string != null);
Object object = ObjectSerializationUtil.convertStringToObject(string);
Assert.assertTrue(object != null);
Assert.assertTrue(object instanceof Expression);
}
Aggregations