Java Code Examples for io.netty.buffer.ByteBuf#duplicate()

private void handleImage(String x, String y, ChannelHandlerContext ctx, String streamId, int latency,
        FullHttpRequest request) {
    ByteBuf image = ImageCache.INSTANCE.image(parseInt(x), parseInt(y));
    FullHttpResponse response = new DefaultFullHttpResponse(HTTP_1_1, OK, image.duplicate());
    response.headers().set(CONTENT_TYPE, "image/jpeg");
    sendResponse(ctx, streamId, latency, response, request);
}
 

@Override
public FullBinaryMemcacheRequest duplicate() {
    ByteBuf key = key();
    if (key != null) {
        key = key.duplicate();
    }
    ByteBuf extras = extras();
    if (extras != null) {
        extras = extras.duplicate();
    }
    return new DefaultFullBinaryMemcacheRequest(key, extras, content().duplicate());
}
 

@Override
public FullBinaryMemcacheResponse duplicate() {
    ByteBuf key = key();
    if (key != null) {
        key = key.duplicate();
    }
    ByteBuf extras = extras();
    if (extras != null) {
        extras = extras.duplicate();
    }
    return new DefaultFullBinaryMemcacheResponse(key, extras, content().duplicate());
}
 

private int extractResponseSequenceId(ByteBuf buffer)
        throws TTransportException
{
    TChannelBufferInputTransport inputTransport = new TChannelBufferInputTransport(buffer.duplicate());
    try {
        TMessage message = protocol.createProtocol(inputTransport).readMessageBegin();
        return message.getSequenceId();
    }
    catch (Throwable e) {
        throw new TTransportException("Could not find sequenceId in Thrift message", e);
    }
    finally {
        inputTransport.release();
    }
}
 

public static RawNetFlowV9Packet parsePacketShallow(ByteBuf bb) {
    final ByteBuf buf = bb.duplicate();

    final int dataLength = buf.readableBytes();
    final NetFlowV9Header header = parseHeader(buf);
    final Map<Integer, byte[]> allTemplates = Maps.newHashMap();
    Map.Entry<Integer, byte[]> optTemplate = null;
    final Set<Integer> usedTemplates = Sets.newHashSet();

    while (buf.isReadable()) {
        buf.markReaderIndex();
        int flowSetId = buf.readUnsignedShort();
        if (flowSetId == 0) {
            final List<Map.Entry<Integer, byte[]>> templates = parseTemplatesShallow(buf);
            for (Map.Entry<Integer, byte[]> t : templates) {
                allTemplates.put(t.getKey(), t.getValue());
            }
        } else if (flowSetId == 1) {
            optTemplate = parseOptionTemplateShallow(buf);
        } else {
            buf.resetReaderIndex();
            usedTemplates.add(parseRecordShallow(buf));
        }
    }

    return RawNetFlowV9Packet.create(header, dataLength, allTemplates, optTemplate, usedTemplates);
}
 

private RuntimeException onCheckPropertiesError(Throwable e) {
   // This is not an expected error, hence no specific logger created
   logger.warn("Could not decode properties for CoreMessage[messageID=" + messageID + ",durable=" + durable + ",userID=" + userID + ",priority=" + priority +
                  ", timestamp=" + timestamp + ",expiration=" + expiration + ",address=" + address + ", propertiesLocation=" + propertiesLocation, e);
   final ByteBuf buffer = this.buffer;
   if (buffer != null) {
      //risky: a racy modification to buffer indexes could break this duplicate operation
      final ByteBuf duplicatebuffer = buffer.duplicate();
      duplicatebuffer.readerIndex(0);
      logger.warn("Failed message has messageID=" + messageID + " and the following buffer:\n" + ByteBufUtil.prettyHexDump(duplicatebuffer));
   } else {
      logger.warn("Failed message has messageID=" + messageID + " and the buffer was null");
   }
   return new RuntimeException(e.getMessage(), e);
}
 

@Override
public FullBinaryMemcacheRequest replace(ByteBuf content) {
    ByteBuf extras = getExtras();
    if (extras != null) {
        extras = extras.duplicate();
    }
    return new DefaultFullBinaryMemcacheRequest(getKey(), extras, content);
}
 

@Override
public FullBinaryMemcacheResponse replace(ByteBuf content) {
    ByteBuf extras = getExtras();
    if (extras != null) {
        extras = extras.duplicate();
    }
    return new DefaultFullBinaryMemcacheResponse(getKey(), extras, content);
}
 

@Override
public ChannelFuture writeData(ChannelHandlerContext ctx, int streamId, ByteBuf data,
                               int padding, boolean endStream, ChannelPromise promise) {
    // duplicate 'data' to prevent readerIndex from being changed, to ease verification
    return super.writeData(ctx, streamId, data.duplicate(), padding, endStream, promise);
}
 

public MqttPublishPayload(ByteBuf buf) {
    ByteBuf b = buf.duplicate();
    this.bytes = new byte[b.readableBytes()];
    b.readBytes(this.bytes);
}
 

@Test
public void decode() throws Exception {

    ByteBuf buf = Unpooled.buffer();

    for (int i = 0; i < 9; i++) {
        buf.writeByte(i);
    }

    ByteBuf duplicate = buf.duplicate();

    EmbeddedChannel channel = new EmbeddedChannel(new FixLengthDecoder(3));
    channel.writeInbound(buf.retain());
    channel.finish();

    ByteBuf read = channel.readInbound();
    Assert.assertEquals(duplicate.readSlice(3), read);


    read = channel.readInbound();
    Assert.assertEquals(duplicate.readSlice(3), read);

    read = channel.readInbound();
    Assert.assertEquals(duplicate.readSlice(3), read);

    read.release();


}
 

@Test
public void decode2() throws Exception {

    ByteBuf buf = Unpooled.buffer();

    for (int i = 0; i < 9; i++) {
        buf.writeByte(i);
    }

    ByteBuf duplicate = buf.duplicate();

    EmbeddedChannel channel = new EmbeddedChannel(new FixLengthDecoder(3));
    Assert.assertFalse(channel.writeInbound(buf.readBytes(2)));
    Assert.assertTrue(channel.writeInbound(buf.readBytes(7)));


    channel.finish();

    ByteBuf read = channel.readInbound();
    Assert.assertEquals(duplicate.readSlice(3), read);


    read = channel.readInbound();
    Assert.assertEquals(duplicate.readSlice(3), read);

    read = channel.readInbound();
    Assert.assertEquals(duplicate.readSlice(3), read);

    read.release();


}
 

/**
 * Creates an ActiveMQBuffer wrapping an underlying ByteBuf
 *
 * The position on this buffer won't affect the position on the inner buffer
 *
 * @param underlying the underlying NIO ByteBuffer
 * @return an ActiveMQBuffer wrapping the underlying NIO ByteBuffer
 */
public static ActiveMQBuffer wrappedBuffer(final ByteBuf underlying) {
   ActiveMQBuffer buff = new ChannelBufferWrapper(underlying.duplicate());

   return buff;
}