diff options
author | Ben Becker <benjamin.becker@gmail.com> | 2013-11-15 19:07:37 -0800 |
---|---|---|
committer | Jacques Nadeau <jacques@apache.org> | 2013-11-15 19:17:15 -0800 |
commit | dd39a5b799a357e4372ffd243a0b887351010f5d (patch) | |
tree | c558b6dd2bb9af757bc67795265163215cd13650 /exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/MergingReceiverCreator.java | |
parent | b07682084da96469e310028e67b365d005f99bdb (diff) |
DRILL-229: N-WAY merging receiver
Diffstat (limited to 'exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/MergingReceiverCreator.java')
-rw-r--r-- | exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/MergingReceiverCreator.java | 49 |
1 files changed, 49 insertions, 0 deletions
diff --git a/exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/MergingReceiverCreator.java b/exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/MergingReceiverCreator.java new file mode 100644 index 000000000..4058a7949 --- /dev/null +++ b/exec/java-exec/src/main/java/org/apache/drill/exec/physical/impl/MergingReceiverCreator.java @@ -0,0 +1,49 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one + * or more contributor license agreements. See the NOTICE file + * distributed with this work for additional information + * regarding copyright ownership. The ASF licenses this file + * to you under the Apache License, Version 2.0 (the + * "License"); you may not use this file except in compliance + * with the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.drill.exec.physical.impl; + +import org.apache.drill.common.exceptions.ExecutionSetupException; +import org.apache.drill.exec.ops.FragmentContext; +import org.apache.drill.exec.physical.config.MergingReceiverPOP; +import org.apache.drill.exec.physical.impl.mergereceiver.MergingRecordBatch; +import org.apache.drill.exec.record.RecordBatch; +import org.apache.drill.exec.work.batch.IncomingBuffers; +import org.apache.drill.exec.work.batch.RawBatchBuffer; + +import java.util.List; + +public class MergingReceiverCreator implements BatchCreator<MergingReceiverPOP> { + static final org.slf4j.Logger logger = org.slf4j.LoggerFactory.getLogger(MergingReceiverCreator.class); + + @Override + public RecordBatch getBatch(FragmentContext context, + MergingReceiverPOP receiver, + List<RecordBatch> children) + throws ExecutionSetupException { + + assert children == null || children.isEmpty(); + IncomingBuffers bufHolder = context.getBuffers(); + + assert bufHolder != null : "IncomingBuffers must be defined for any place a receiver is declared."; + RawBatchBuffer[] buffers = bufHolder.getBuffers(receiver.getOppositeMajorFragmentId()); + + return new MergingRecordBatch(context, receiver, buffers); + } + + +} |