Search in sources :

Example 1 with PartitionType

use of io.siddhi.query.api.execution.partition.PartitionType in project siddhi by wso2.

the class StreamPartitioner method createSingleInputStreamExecutors.

private void createSingleInputStreamExecutors(SingleInputStream inputStream, Partition partition, MetaStreamEvent metaEvent, List<VariableExpressionExecutor> executors, Map<String, Table> tableMap, SiddhiQueryContext siddhiQueryContext) {
    List<PartitionExecutor> executorList = new ArrayList<PartitionExecutor>();
    partitionExecutorLists.add(executorList);
    if (!inputStream.isInnerStream()) {
        for (PartitionType partitionType : partition.getPartitionTypeMap().values()) {
            if (partitionType instanceof ValuePartitionType) {
                if (partitionType.getStreamId().equals(inputStream.getStreamId())) {
                    executorList.add(new ValuePartitionExecutor(ExpressionParser.parseExpression(((ValuePartitionType) partitionType).getExpression(), metaEvent, SiddhiConstants.UNKNOWN_STATE, tableMap, executors, false, 0, ProcessingMode.BATCH, false, siddhiQueryContext)));
                }
            } else {
                for (RangePartitionType.RangePartitionProperty rangePartitionProperty : ((RangePartitionType) partitionType).getRangePartitionProperties()) {
                    if (partitionType.getStreamId().equals(inputStream.getStreamId())) {
                        executorList.add(new RangePartitionExecutor((ConditionExpressionExecutor) ExpressionParser.parseExpression(rangePartitionProperty.getCondition(), metaEvent, SiddhiConstants.UNKNOWN_STATE, tableMap, executors, false, 0, ProcessingMode.BATCH, false, siddhiQueryContext), rangePartitionProperty.getPartitionKey()));
                    }
                }
            }
        }
    }
}
Also used : ValuePartitionExecutor(io.siddhi.core.partition.executor.ValuePartitionExecutor) RangePartitionType(io.siddhi.query.api.execution.partition.RangePartitionType) PartitionExecutor(io.siddhi.core.partition.executor.PartitionExecutor) RangePartitionExecutor(io.siddhi.core.partition.executor.RangePartitionExecutor) ValuePartitionExecutor(io.siddhi.core.partition.executor.ValuePartitionExecutor) ArrayList(java.util.ArrayList) RangePartitionExecutor(io.siddhi.core.partition.executor.RangePartitionExecutor) ValuePartitionType(io.siddhi.query.api.execution.partition.ValuePartitionType) ConditionExpressionExecutor(io.siddhi.core.executor.condition.ConditionExpressionExecutor) PartitionType(io.siddhi.query.api.execution.partition.PartitionType) ValuePartitionType(io.siddhi.query.api.execution.partition.ValuePartitionType) RangePartitionType(io.siddhi.query.api.execution.partition.RangePartitionType)

Example 2 with PartitionType

use of io.siddhi.query.api.execution.partition.PartitionType in project siddhi by wso2.

the class SiddhiQLBaseVisitorImpl method visitPartition_with_stream.

/**
 * {@inheritDoc}
 * <p>The default implementation returns the result of calling
 * {@link #visitChildren} on {@code ctx}.</p>
 *
 * @param ctx
 */
@Override
public PartitionType visitPartition_with_stream(@NotNull SiddhiQLParser.Partition_with_streamContext ctx) {
    String streamId = (String) visit(ctx.stream_id());
    activeStreams.add(streamId);
    try {
        if (ctx.condition_ranges() != null) {
            PartitionType partitionType = new RangePartitionType(streamId, (RangePartitionType.RangePartitionProperty[]) visit(ctx.condition_ranges()));
            populateQueryContext(partitionType, ctx);
            return partitionType;
        } else if (ctx.attribute() != null) {
            PartitionType partitionType = new ValuePartitionType(streamId, (Expression) visit(ctx.attribute()));
            populateQueryContext(partitionType, ctx);
            return partitionType;
        } else {
            throw newSiddhiParserException(ctx);
        }
    } finally {
        activeStreams.clear();
    }
}
Also used : RangePartitionType(io.siddhi.query.api.execution.partition.RangePartitionType) Expression(io.siddhi.query.api.expression.Expression) ValuePartitionType(io.siddhi.query.api.execution.partition.ValuePartitionType) ValuePartitionType(io.siddhi.query.api.execution.partition.ValuePartitionType) PartitionType(io.siddhi.query.api.execution.partition.PartitionType) RangePartitionType(io.siddhi.query.api.execution.partition.RangePartitionType)

Aggregations

PartitionType (io.siddhi.query.api.execution.partition.PartitionType)2 RangePartitionType (io.siddhi.query.api.execution.partition.RangePartitionType)2 ValuePartitionType (io.siddhi.query.api.execution.partition.ValuePartitionType)2 ConditionExpressionExecutor (io.siddhi.core.executor.condition.ConditionExpressionExecutor)1 PartitionExecutor (io.siddhi.core.partition.executor.PartitionExecutor)1 RangePartitionExecutor (io.siddhi.core.partition.executor.RangePartitionExecutor)1 ValuePartitionExecutor (io.siddhi.core.partition.executor.ValuePartitionExecutor)1 Expression (io.siddhi.query.api.expression.Expression)1 ArrayList (java.util.ArrayList)1