io.aeron.protocol.StatusMessageFlyweight Maven / Gradle / Ivy
/*
* Copyright 2014-2019 Real Logic Ltd.
*
* Licensed 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
*
* https://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 io.aeron.protocol;
import io.aeron.exceptions.AeronException;
import org.agrona.concurrent.UnsafeBuffer;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import static java.nio.ByteOrder.LITTLE_ENDIAN;
/**
* Flyweight for a Status Message Frame.
*
*
* Status Message wiki page.
*/
public class StatusMessageFlyweight extends HeaderFlyweight
{
/**
* Length of the Status Message Frame
*/
public static final int HEADER_LENGTH = 36;
/**
* Publisher should send SETUP frame
*/
public static final short SEND_SETUP_FLAG = 0x80;
private static final int SESSION_ID_FIELD_OFFSET = 8;
private static final int STREAM_ID_FIELD_OFFSET = 12;
private static final int CONSUMPTION_TERM_ID_FIELD_OFFSET = 16;
private static final int CONSUMPTION_TERM_OFFSET_FIELD_OFFSET = 20;
private static final int RECEIVER_WINDOW_FIELD_OFFSET = 24;
private static final int RECEIVER_ID_FIELD_OFFSET = 28;
private static final int APP_SPECIFIC_FEEDBACK_FIELD_OFFSET = 36;
public StatusMessageFlyweight()
{
}
public StatusMessageFlyweight(final ByteBuffer buffer)
{
super(buffer);
}
public StatusMessageFlyweight(final UnsafeBuffer buffer)
{
super(buffer);
}
/**
* return session id field
*
* @return session id field
*/
public int sessionId()
{
return getInt(SESSION_ID_FIELD_OFFSET, LITTLE_ENDIAN);
}
/**
* set session id field
*
* @param sessionId field value
* @return flyweight
*/
public StatusMessageFlyweight sessionId(final int sessionId)
{
putInt(SESSION_ID_FIELD_OFFSET, sessionId, LITTLE_ENDIAN);
return this;
}
/**
* return stream id field
*
* @return stream id field
*/
public int streamId()
{
return getInt(STREAM_ID_FIELD_OFFSET, LITTLE_ENDIAN);
}
/**
* set stream id field
*
* @param streamId field value
* @return flyweight
*/
public StatusMessageFlyweight streamId(final int streamId)
{
putInt(STREAM_ID_FIELD_OFFSET, streamId, LITTLE_ENDIAN);
return this;
}
/**
* return highest consumption term offset field
*
* @return highest consumption term offset field
*/
public int consumptionTermOffset()
{
return getInt(CONSUMPTION_TERM_OFFSET_FIELD_OFFSET, LITTLE_ENDIAN);
}
/**
* set highest consumption term offset field
*
* @param termOffset field value
* @return flyweight
*/
public StatusMessageFlyweight consumptionTermOffset(final int termOffset)
{
putInt(CONSUMPTION_TERM_OFFSET_FIELD_OFFSET, termOffset, LITTLE_ENDIAN);
return this;
}
/**
* return highest consumption term id field
*
* @return highest consumption term id field
*/
public int consumptionTermId()
{
return getInt(CONSUMPTION_TERM_ID_FIELD_OFFSET, LITTLE_ENDIAN);
}
/**
* set highest consumption term id field
*
* @param termId field value
* @return flyweight
*/
public StatusMessageFlyweight consumptionTermId(final int termId)
{
putInt(CONSUMPTION_TERM_ID_FIELD_OFFSET, termId, LITTLE_ENDIAN);
return this;
}
/**
* return receiver window field
*
* @return receiver window field
*/
public int receiverWindowLength()
{
return getInt(RECEIVER_WINDOW_FIELD_OFFSET, LITTLE_ENDIAN);
}
/**
* set receiver window field
*
* @param receiverWindowLength field value
* @return flyweight
*/
public StatusMessageFlyweight receiverWindowLength(final int receiverWindowLength)
{
putInt(RECEIVER_WINDOW_FIELD_OFFSET, receiverWindowLength, LITTLE_ENDIAN);
return this;
}
/**
* Identifier for the receiver to distinguish them for FlowControl strategies.
*
* @return identifier for the receiver to distinguish them for FlowControl strategies.
*/
public long receiverId()
{
final long value;
if (ByteOrder.nativeOrder() == LITTLE_ENDIAN)
{
value =
(
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 7)) << 56) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 6) & 0xFF) << 48) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 5) & 0xFF) << 40) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 4) & 0xFF) << 32) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 3) & 0xFF) << 24) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 2) & 0xFF) << 16) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 1) & 0xFF) << 8) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 0) & 0xFF))
);
}
else
{
value =
(
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 0)) << 56) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 1) & 0xFF) << 48) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 2) & 0xFF) << 40) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 3) & 0xFF) << 32) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 4) & 0xFF) << 24) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 5) & 0xFF) << 16) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 6) & 0xFF) << 8) |
(((long)getByte(RECEIVER_ID_FIELD_OFFSET + 7) & 0xFF))
);
}
return value;
}
/**
* Identifier for the receiver to distinguish them for FlowControl strategies.
*
* @param id for the receiver to distinguish them for FlowControl strategies.
* @return flyweight
*/
public StatusMessageFlyweight receiverId(final long id)
{
if (ByteOrder.nativeOrder() == LITTLE_ENDIAN)
{
putByte(RECEIVER_ID_FIELD_OFFSET + 7, (byte)(id >> 56));
putByte(RECEIVER_ID_FIELD_OFFSET + 6, (byte)(id >> 48));
putByte(RECEIVER_ID_FIELD_OFFSET + 5, (byte)(id >> 40));
putByte(RECEIVER_ID_FIELD_OFFSET + 4, (byte)(id >> 32));
putByte(RECEIVER_ID_FIELD_OFFSET + 3, (byte)(id >> 24));
putByte(RECEIVER_ID_FIELD_OFFSET + 2, (byte)(id >> 16));
putByte(RECEIVER_ID_FIELD_OFFSET + 1, (byte)(id >> 8));
putByte(RECEIVER_ID_FIELD_OFFSET + 0, (byte)(id));
}
else
{
putByte(RECEIVER_ID_FIELD_OFFSET + 0, (byte)(id >> 56));
putByte(RECEIVER_ID_FIELD_OFFSET + 1, (byte)(id >> 48));
putByte(RECEIVER_ID_FIELD_OFFSET + 2, (byte)(id >> 40));
putByte(RECEIVER_ID_FIELD_OFFSET + 3, (byte)(id >> 32));
putByte(RECEIVER_ID_FIELD_OFFSET + 4, (byte)(id >> 24));
putByte(RECEIVER_ID_FIELD_OFFSET + 5, (byte)(id >> 16));
putByte(RECEIVER_ID_FIELD_OFFSET + 6, (byte)(id >> 8));
putByte(RECEIVER_ID_FIELD_OFFSET + 7, (byte)(id));
}
return this;
}
/**
* Retrieve the Application Specific Feedback (if present) from the Status Message.
*
* @param destination to store the feedback
* @return the number of bytes in the feedback copied into the destination
*/
public int applicationSpecificFeedback(final byte[] destination)
{
final int frameLength = frameLength();
int result = 0;
if (frameLength > HEADER_LENGTH)
{
if (frameLength > capacity())
{
throw new AeronException(
"SM application specific feedback (" + (frameLength - HEADER_LENGTH) + ") is truncated (" +
(capacity() - HEADER_LENGTH) + ")");
}
final int copyLength = Math.min(destination.length, frameLength - HEADER_LENGTH);
getBytes(APP_SPECIFIC_FEEDBACK_FIELD_OFFSET, destination, 0, copyLength);
result = copyLength;
}
return result;
}
/**
* Set the Application Specific Feedback for the Status Message.
*
* @param source of the feedback to set
* @param offset of the feedback in the source
* @param length of the feedback in bytes
* @return flyweight
*/
public StatusMessageFlyweight applicationSpecificFeedback(final byte[] source, final int offset, final int length)
{
frameLength(HEADER_LENGTH + length);
putBytes(APP_SPECIFIC_FEEDBACK_FIELD_OFFSET, source, offset, length);
return this;
}
public String toString()
{
return "STATUS{" +
"frame-length=" + frameLength() +
" version=" + version() +
" flags=" + String.valueOf(flagsToChars(flags())) +
" type=" + headerType() +
" session-id=" + sessionId() +
" stream-id=" + streamId() +
" consumption-term-id=" + consumptionTermId() +
" consumption-term-offset=" + consumptionTermOffset() +
" receiver-window-length=" + receiverWindowLength() +
"}";
}
}