Search in sources :

Example 1 with NewBufferSize

use of org.apache.flink.runtime.io.network.netty.NettyMessage.NewBufferSize in project flink by apache.

the class PartitionRequestServerHandler method channelRead0.

@Override
protected void channelRead0(ChannelHandlerContext ctx, NettyMessage msg) throws Exception {
    try {
        Class<?> msgClazz = msg.getClass();
        // ----------------------------------------------------------------
        if (msgClazz == PartitionRequest.class) {
            PartitionRequest request = (PartitionRequest) msg;
            LOG.debug("Read channel on {}: {}.", ctx.channel().localAddress(), request);
            try {
                NetworkSequenceViewReader reader;
                reader = new CreditBasedSequenceNumberingViewReader(request.receiverId, request.credit, outboundQueue);
                reader.requestSubpartitionView(partitionProvider, request.partitionId, request.queueIndex);
                outboundQueue.notifyReaderCreated(reader);
            } catch (PartitionNotFoundException notFound) {
                respondWithError(ctx, notFound, request.receiverId);
            }
        } else // ----------------------------------------------------------------
        if (msgClazz == TaskEventRequest.class) {
            TaskEventRequest request = (TaskEventRequest) msg;
            if (!taskEventPublisher.publish(request.partitionId, request.event)) {
                respondWithError(ctx, new IllegalArgumentException("Task event receiver not found."), request.receiverId);
            }
        } else if (msgClazz == CancelPartitionRequest.class) {
            CancelPartitionRequest request = (CancelPartitionRequest) msg;
            outboundQueue.cancel(request.receiverId);
        } else if (msgClazz == CloseRequest.class) {
            outboundQueue.close();
        } else if (msgClazz == AddCredit.class) {
            AddCredit request = (AddCredit) msg;
            outboundQueue.addCreditOrResumeConsumption(request.receiverId, reader -> reader.addCredit(request.credit));
        } else if (msgClazz == ResumeConsumption.class) {
            ResumeConsumption request = (ResumeConsumption) msg;
            outboundQueue.addCreditOrResumeConsumption(request.receiverId, NetworkSequenceViewReader::resumeConsumption);
        } else if (msgClazz == AckAllUserRecordsProcessed.class) {
            AckAllUserRecordsProcessed request = (AckAllUserRecordsProcessed) msg;
            outboundQueue.acknowledgeAllRecordsProcessed(request.receiverId);
        } else if (msgClazz == NewBufferSize.class) {
            NewBufferSize request = (NewBufferSize) msg;
            outboundQueue.notifyNewBufferSize(request.receiverId, request.bufferSize);
        } else {
            LOG.warn("Received unexpected client request: {}", msg);
        }
    } catch (Throwable t) {
        respondWithError(ctx, t);
    }
}
Also used : NewBufferSize(org.apache.flink.runtime.io.network.netty.NettyMessage.NewBufferSize) CancelPartitionRequest(org.apache.flink.runtime.io.network.netty.NettyMessage.CancelPartitionRequest) PartitionRequest(org.apache.flink.runtime.io.network.netty.NettyMessage.PartitionRequest) AckAllUserRecordsProcessed(org.apache.flink.runtime.io.network.netty.NettyMessage.AckAllUserRecordsProcessed) ResumeConsumption(org.apache.flink.runtime.io.network.netty.NettyMessage.ResumeConsumption) PartitionNotFoundException(org.apache.flink.runtime.io.network.partition.PartitionNotFoundException) AddCredit(org.apache.flink.runtime.io.network.netty.NettyMessage.AddCredit) CancelPartitionRequest(org.apache.flink.runtime.io.network.netty.NettyMessage.CancelPartitionRequest) TaskEventRequest(org.apache.flink.runtime.io.network.netty.NettyMessage.TaskEventRequest) NetworkSequenceViewReader(org.apache.flink.runtime.io.network.NetworkSequenceViewReader) CloseRequest(org.apache.flink.runtime.io.network.netty.NettyMessage.CloseRequest)

Aggregations

NetworkSequenceViewReader (org.apache.flink.runtime.io.network.NetworkSequenceViewReader)1 AckAllUserRecordsProcessed (org.apache.flink.runtime.io.network.netty.NettyMessage.AckAllUserRecordsProcessed)1 AddCredit (org.apache.flink.runtime.io.network.netty.NettyMessage.AddCredit)1 CancelPartitionRequest (org.apache.flink.runtime.io.network.netty.NettyMessage.CancelPartitionRequest)1 CloseRequest (org.apache.flink.runtime.io.network.netty.NettyMessage.CloseRequest)1 NewBufferSize (org.apache.flink.runtime.io.network.netty.NettyMessage.NewBufferSize)1 PartitionRequest (org.apache.flink.runtime.io.network.netty.NettyMessage.PartitionRequest)1 ResumeConsumption (org.apache.flink.runtime.io.network.netty.NettyMessage.ResumeConsumption)1 TaskEventRequest (org.apache.flink.runtime.io.network.netty.NettyMessage.TaskEventRequest)1 PartitionNotFoundException (org.apache.flink.runtime.io.network.partition.PartitionNotFoundException)1