io.aeron.command.CounterMessageFlyweight Maven / Gradle / Ivy
Show all versions of aeron-client Show documentation
/*
* 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.command;
import io.aeron.ErrorCode;
import io.aeron.exceptions.ControlProtocolException;
import org.agrona.BitUtil;
import org.agrona.DirectBuffer;
import static org.agrona.BitUtil.SIZE_OF_INT;
import static org.agrona.BitUtil.SIZE_OF_LONG;
/**
* Message to denote a new counter.
*
* Note: Layout should be SBE 2.0 compliant so that the label length is aligned.
*
* @see ControlProtocolEvents
*
* 0 1 2 3
* 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1 2 3 4 5 6 7 8 9 0 1
* +-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+-+
* | Correlation ID |
* | |
* +---------------------------------------------------------------+
* | Counter Type ID |
* +---------------------------------------------------------------+
* | Key Length |
* +---------------------------------------------------------------+
* | Key Buffer ...
* ... |
* +---------------------------------------------------------------+
* | Label Length |
* +---------------------------------------------------------------+
* | Label (ASCII) ...
* ... |
* +---------------------------------------------------------------+
*
*/
public class CounterMessageFlyweight extends CorrelatedMessageFlyweight
{
private static final int COUNTER_TYPE_ID_FIELD_OFFSET = CORRELATION_ID_FIELD_OFFSET + SIZE_OF_LONG;
private static final int KEY_LENGTH_OFFSET = COUNTER_TYPE_ID_FIELD_OFFSET + SIZE_OF_INT;
private static final int MINIMUM_LENGTH = KEY_LENGTH_OFFSET + SIZE_OF_INT;
/**
* return type id field
*
* @return type id field
*/
public int typeId()
{
return buffer.getInt(offset + COUNTER_TYPE_ID_FIELD_OFFSET);
}
/**
* set counter type id field
*
* @param typeId field value
* @return flyweight
*/
public CounterMessageFlyweight typeId(final long typeId)
{
buffer.putLong(offset + COUNTER_TYPE_ID_FIELD_OFFSET, typeId);
return this;
}
/**
* Offset of the key buffer
*
* @return offset of the key buffer
*/
public int keyBufferOffset()
{
return KEY_LENGTH_OFFSET + SIZE_OF_INT;
}
/**
* Length of the key buffer in bytes
*
* @return length of key buffer in bytes
*/
public int keyBufferLength()
{
return buffer.getInt(offset + KEY_LENGTH_OFFSET);
}
/**
* Fill the key buffer.
*
* @param keyBuffer containing the optional key for the counter.
* @param keyOffset within the keyBuffer at which the key begins.
* @param keyLength of the key in the keyBuffer.
* @return flyweight
*/
public CounterMessageFlyweight keyBuffer(final DirectBuffer keyBuffer, final int keyOffset, final int keyLength)
{
buffer.putInt(KEY_LENGTH_OFFSET, keyLength);
if (null != keyBuffer && keyLength > 0)
{
buffer.putBytes(keyBufferOffset(), keyBuffer, keyOffset, keyLength);
}
return this;
}
/**
* Offset of label buffer.
*
* @return offset of label buffer
*/
public int labelBufferOffset()
{
return labelOffset() + SIZE_OF_INT;
}
/**
* Length of label buffer in bytes.
*
* @return length of label buffer in bytes
*/
public int labelBufferLength()
{
return buffer.getInt(offset + labelOffset());
}
/**
* Fill the label buffer.
*
* @param labelBuffer containing the mandatory label for the counter.
* @param labelOffset within the labelBuffer at which the label begins.
* @param labelLength of the label in the labelBuffer.
* @return flyweight
*/
public CounterMessageFlyweight labelBuffer(
final DirectBuffer labelBuffer, final int labelOffset, final int labelLength)
{
buffer.putInt(labelOffset(), labelLength);
buffer.putBytes(labelBufferOffset(), labelBuffer, labelOffset, labelLength);
return this;
}
/**
* Fill the label.
*
* @param label for the counter
* @return flyweight
*/
public CounterMessageFlyweight label(final String label)
{
buffer.putStringAscii(labelOffset(), label);
return this;
}
/**
* Get the length of the current message
*
* NB: must be called after the data is written in order to be accurate.
*
* @return the length of the current message
*/
public int length()
{
final int labelOffset = labelOffset();
return labelOffset + buffer.getInt(offset + labelOffset) + SIZE_OF_INT;
}
/**
* Validate buffer length is long enough for message.
*
* @param msgTypeId type of message.
* @param length of message in bytes to validate.
*/
public void validateLength(final int msgTypeId, final int length)
{
if (length < MINIMUM_LENGTH)
{
throw new ControlProtocolException(
ErrorCode.MALFORMED_COMMAND, "command=" + msgTypeId + " too short: length=" + length);
}
final int labelOffset = labelOffset();
if ((length - labelOffset) < SIZE_OF_INT)
{
throw new ControlProtocolException(
ErrorCode.MALFORMED_COMMAND, "command=" + msgTypeId + " too short for key: length=" + length);
}
if (length < length())
{
throw new ControlProtocolException(
ErrorCode.MALFORMED_COMMAND, "command=" + msgTypeId + " too short for label: length=" + length);
}
}
private int labelOffset()
{
return KEY_LENGTH_OFFSET + SIZE_OF_INT + BitUtil.align(buffer.getInt(offset + KEY_LENGTH_OFFSET), SIZE_OF_INT);
}
}