@Override public Void visitReceiver(Receiver receiver, List<Collector> collectors) throws RuntimeException { List<MinorFragmentEndpoint> endpoints = receiver.getProvidingEndpoints(); List<IncomingMinorFragment> list = new ArrayList<>(endpoints.size()); for (MinorFragmentEndpoint ep : endpoints) { list.add(IncomingMinorFragment.newBuilder().setEndpoint(ep.getEndpoint()).setMinorFragment(ep.getId()).build()); } collectors.add(Collector.newBuilder() .setIsSpooling(receiver.isSpooling()) .setOppositeMajorFragmentId(receiver.getOppositeMajorFragmentId()) .setSupportsOutOfOrder(receiver.supportsOutOfOrderExchange()) .addAllIncomingMinorFragment(list) .build()); return null; }
@java.lang.Override public int hashCode() { if (memoizedHashCode != 0) { return memoizedHashCode; } int hash = 41; hash = (19 * hash) + getDescriptorForType().hashCode(); if (hasMinorFragment()) { hash = (37 * hash) + MINOR_FRAGMENT_FIELD_NUMBER; hash = (53 * hash) + getMinorFragment(); } if (hasEndpoint()) { hash = (37 * hash) + ENDPOINT_FIELD_NUMBER; hash = (53 * hash) + getEndpoint().hashCode(); } hash = (29 * hash) + getUnknownFields().hashCode(); memoizedHashCode = hash; return hash; }
public com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment buildPartial() { com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment result = new com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment(this); int from_bitField0_ = bitField0_; int to_bitField0_ = 0; if (((from_bitField0_ & 0x00000001) == 0x00000001)) { to_bitField0_ |= 0x00000001; } result.minorFragment_ = minorFragment_; if (((from_bitField0_ & 0x00000002) == 0x00000002)) { to_bitField0_ |= 0x00000002; } if (endpointBuilder_ == null) { result.endpoint_ = endpoint_; } else { result.endpoint_ = endpointBuilder_.build(); } result.bitField0_ = to_bitField0_; onBuilt(); return result; }
int index = 0; for (IncomingMinorFragment fragment: collector.getIncomingMinorFragmentList()) { fragmentMap.put(fragment.getMinorFragment(), index); completionMessages[index] = new CompletionMessageSender(fragment.getEndpoint(), fragment.getMinorFragment()); index++; try (AutoCloseables.RollbackCloseable rollbackCloseable = new AutoCloseables.RollbackCloseable()){ for (IncomingMinorFragment fragment : fragments) { final String name = String.format("nway-recv-%s-%s:%d:%d", spooling ? "spool" : "mem", fragment.getEndpoint().getAddress(), collector.getOppositeMajorFragmentId(), fragment.getMinorFragment()); final SharedResource resource = resourceGroup.createResource(name, spooling ? SharedResourceType.NWAY_RECV_SPOOL_BUFFER : SharedResourceType.NWAY_RECV_MEM_BUFFER); final RawBatchBuffer buffer; if (spooling) { buffer = new SpoolingRawBatchBuffer(resource, config, workQueue, handle, spillService, allocator, bufferCapacity, collector.getOppositeMajorFragmentId(), fragment.getMinorFragment()); } else { buffer = new UnlimitedRawBatchBuffer(resource, config, handle, allocator, bufferCapacity, collector.getOppositeMajorFragmentId()); buffers[fragment.getMinorFragment()] = buffer;
MinorFragmentEndpoint ep2 = mock(MinorFragmentEndpoint.class); when(ep2.getEndpoint()).thenReturn(CoordinationProtos.NodeEndpoint.newBuilder().setAddress("localhost").setFabricPort(12345).build()); list.add(CoordExecRPC.IncomingMinorFragment.newBuilder().setEndpoint(ep1.getEndpoint()).setMinorFragment(ep1.getId()).build()); list.add(CoordExecRPC.IncomingMinorFragment.newBuilder().setEndpoint(ep2.getEndpoint()).setMinorFragment(ep2.getId()).build()); CoordExecRPC.Collector collector = CoordExecRPC.Collector.newBuilder() .setIsSpooling(true)
public void mergeFrom(io.protostuff.Input input, com.dremio.exec.proto.CoordExecRPC.Collector.Builder builder) throws java.io.IOException { for(int number = input.readFieldNumber(this);; number = input.readFieldNumber(this)) { switch(number) { case 0: return; case 1: builder.setOppositeMajorFragmentId(input.readInt32()); break; case 2: builder.addIncomingMinorFragment(input.mergeObject(com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.newBuilder(), com.dremio.exec.proto.SchemaCoordExecRPC.IncomingMinorFragment.MERGE)); break; case 3: builder.setSupportsOutOfOrder(input.readBool()); break; case 4: builder.setIsSpooling(input.readBool()); break; default: input.handleUnknownField(number, this); } } } public boolean isInitialized(com.dremio.exec.proto.CoordExecRPC.Collector.Builder builder)
@java.lang.Override public boolean equals(final java.lang.Object obj) { if (obj == this) { return true; } if (!(obj instanceof com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment)) { return super.equals(obj); } com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment other = (com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment) obj; boolean result = true; result = result && (hasMinorFragment() == other.hasMinorFragment()); if (hasMinorFragment()) { result = result && (getMinorFragment() == other.getMinorFragment()); } result = result && (hasEndpoint() == other.hasEndpoint()); if (hasEndpoint()) { result = result && getEndpoint() .equals(other.getEndpoint()); } result = result && getUnknownFields().equals(other.getUnknownFields()); return result; }
public int getSerializedSize() { int size = memoizedSerializedSize; if (size != -1) return size; size = 0; if (((bitField0_ & 0x00000001) == 0x00000001)) { size += com.google.protobuf.CodedOutputStream .computeInt32Size(1, minorFragment_); } if (((bitField0_ & 0x00000002) == 0x00000002)) { size += com.google.protobuf.CodedOutputStream .computeMessageSize(2, endpoint_); } size += getUnknownFields().getSerializedSize(); memoizedSerializedSize = size; return size; }
public Builder toBuilder() { return newBuilder(this); }
public Builder newBuilderForType() { return newBuilder(); } public static Builder newBuilder(com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment prototype) {
public com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment getDefaultInstanceForType() { return com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.getDefaultInstance(); }
/** * <code>repeated .exec.bit.control.IncomingMinorFragment incoming_minor_fragment = 2;</code> */ public com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.Builder addIncomingMinorFragmentBuilder() { return getIncomingMinorFragmentFieldBuilder().addBuilder( com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.getDefaultInstance()); } /**
/** * <code>repeated .exec.bit.control.IncomingMinorFragment incoming_minor_fragment = 2;</code> */ public com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.Builder addIncomingMinorFragmentBuilder( int index) { return getIncomingMinorFragmentFieldBuilder().addBuilder( index, com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.getDefaultInstance()); } /**
public com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.Builder newMessage() { return com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.newBuilder(); } public java.lang.String getFieldName(int number)
public void writeTo(com.google.protobuf.CodedOutputStream output) throws java.io.IOException { getSerializedSize(); if (((bitField0_ & 0x00000001) == 0x00000001)) { output.writeInt32(1, minorFragment_); } if (((bitField0_ & 0x00000002) == 0x00000002)) { output.writeMessage(2, endpoint_); } getUnknownFields().writeTo(output); }
public void writeTo(io.protostuff.Output output, com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment message) throws java.io.IOException { if(message.hasMinorFragment()) output.writeInt32(1, message.getMinorFragment(), false); if(message.hasEndpoint()) output.writeObject(2, message.getEndpoint(), com.dremio.exec.proto.SchemaCoordinationProtos.NodeEndpoint.WRITE, false); } public boolean isInitialized(com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment message)
public static Builder newBuilder(com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment prototype) { return newBuilder().mergeFrom(prototype); } public Builder toBuilder() { return newBuilder(this); }
public Builder mergeFrom(com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment other) { if (other == com.dremio.exec.proto.CoordExecRPC.IncomingMinorFragment.getDefaultInstance()) return this; if (other.hasMinorFragment()) { setMinorFragment(other.getMinorFragment()); } if (other.hasEndpoint()) { mergeEndpoint(other.getEndpoint()); } this.mergeUnknownFields(other.getUnknownFields()); return this; }