public Builder mergeFrom(org.apache.drill.exec.proto.BitControl.PlanFragment other) { if (other == org.apache.drill.exec.proto.BitControl.PlanFragment.getDefaultInstance()) return this; if (other.hasHandle()) { mergeHandle(other.getHandle());
int fragmentIndex = 0; for (final PlanFragment planFragment : fragments) { final ExecProtos.FragmentHandle fragmentHandle = planFragment.getHandle(); stringBuilder.append("PlanFragment("); stringBuilder.append(++fragmentIndex);
void addFragmentStatusTracker(final PlanFragment fragment, final boolean isRoot) { final DrillbitEndpoint assignment = fragment.getAssignment(); NodeTracker tracker = nodeMap.get(assignment); if (tracker == null) { tracker = new NodeTracker(assignment); nodeMap.put(assignment, tracker); } tracker.addFragment(); addFragment(new FragmentData(fragment.getHandle(), assignment, isRoot)); }
public Builder mergeFrom(org.apache.drill.exec.proto.BitControl.PlanFragment other) { if (other == org.apache.drill.exec.proto.BitControl.PlanFragment.getDefaultInstance()) return this; if (other.hasHandle()) { mergeHandle(other.getHandle());
"frag:" + QueryIdHelper.getFragmentId(fragment.getHandle()), fragment.getMemInitial(), fragment.getMemMax());
jsonArray.add(objectMapper.readTree(fragment.getFragmentJson())); } catch (IOException e) { logger.error("Exception while trying to read PlanFragment JSON for %s", fragment.getHandle().getQueryId(), e); throw new RpcException(e);
final List<PlanFragment> planFragments = Lists.newArrayList(); for (PlanFragment myFragment : fragmentsList) { final FragmentHandle handle = myFragment.getHandle();
public void writeTo(com.dyuproject.protostuff.Output output, org.apache.drill.exec.proto.BitControl.PlanFragment message) throws java.io.IOException output.writeObject(1, message.getHandle(), org.apache.drill.exec.proto.SchemaExecProtos.FragmentHandle.WRITE, false);
/** * Submits root and non-root fragments fragments for running. * In case of success move query to the running state. */ public void submit() throws ExecutionSetupException { assert planFragments != null; assert rootPlanFragment != null; assert rootOperator != null; QueryId queryId = foreman.getQueryId(); assert queryId == rootPlanFragment.getHandle().getQueryId(); QueryManager queryManager = foreman.getQueryManager(); drillbitContext.getWorkBus().addFragmentStatusListener(queryId, queryManager.getFragmentStatusListener()); drillbitContext.getClusterCoordinator().addDrillbitStatusListener(queryManager.getDrillbitStatusListener()); logger.debug("Submitting fragments to run."); // set up the root fragment first so we'll have incoming buffers available. setupRootFragment(rootPlanFragment, rootOperator); setupNonRootFragments(planFragments); logger.debug("Fragments running."); }
@Override public BufferAllocator getNewChildAllocator(final String operatorName, final int operatorId, final long initialReservation, final long maximumReservation) throws OutOfMemoryException { return allocator.newChildAllocator( "op:" + QueryIdHelper.getFragmentId(fragment.getHandle()) + ":" + operatorId + ":" + operatorName, initialReservation, maximumReservation ); }
/** * The FragmentHandle for this Fragment * @return FragmentHandle */ @Override public FragmentHandle getHandle() { return fragment.getHandle(); }
public AbstractFragmentManager(final PlanFragment fragment, final FragmentExecutor executor, final FragmentStatusReporter statusReporter, final FragmentRoot rootOperator) { this.fragmentHandle = fragment.getHandle(); this.fragmentContext = executor.getContext(); this.buffers = new IncomingBuffers(fragment, fragmentContext); this.fragmentContext.setBuffers(buffers); this.fragmentExecutor = executor; }
public void writeTo(com.dyuproject.protostuff.Output output, org.apache.drill.exec.proto.BitControl.PlanFragment message) throws java.io.IOException output.writeObject(1, message.getHandle(), org.apache.drill.exec.proto.SchemaExecProtos.FragmentHandle.WRITE, false);