org.graylog2.plugin.journal.RawMessage Java Examples

@Test
public void decodeMessagesReturnsNullIfNetFlowParserThrowsFlowException() throws Exception {
    final byte[] b = "Foobar".getBytes(StandardCharsets.UTF_8);
    final InetSocketAddress source = new InetSocketAddress(InetAddress.getLocalHost(), 12345);
    final RawMessage rawMessage = new RawMessage(b, source) {
        private boolean triggered = false;
        @Override
        public byte[] getPayload() {
            if (triggered) {
                return new byte[]{};
            }
            triggered = true;
            throw new FlowException("Boom!");
        }
    };

    final Collection<Message> messages = codec.decodeMessages(rawMessage);
    assertThat(messages).isNull();
}
 

@Nullable
@Override
public Message decode(@Nonnull RawMessage rawMessage) {
    try {
        final CloudTrailRecord record = objectMapper.readValue(rawMessage.getPayload(), CloudTrailRecord.class);
        final String source = configuration.getString(Config.CK_OVERRIDE_SOURCE, "aws-cloudtrail");
        final Message message = new Message(record.getConstructedMessage(), source, DateTime.parse(record.eventTime));

        message.addFields(record.additionalFieldsAsMap());
        message.addField("full_message", record.getFullMessage());
        message.addField(AWS.SOURCE_GROUP_IDENTIFIER, true);

        return message;
    } catch (Exception e) {
        throw new RuntimeException("Could not deserialize CloudTrail record.", e);
    }
}
 

@Nullable
@Override
public Message decode(@Nonnull RawMessage rawMessage) {
    final byte[] payload = rawMessage.getPayload();
    final JsonNode event;
    try {
        event = objectMapper.readTree(payload);
    } catch (IOException e) {
        LOG.error("Couldn't decode raw message {}", rawMessage);
        return null;
    }

    return parseEvent(event);
}
 

@Override
protected void channelRead0(ChannelHandlerContext ctx, ByteBuf message) throws Exception {
    final int readableBytes = message.readableBytes();
    final byte[] messageBytes = new byte[readableBytes];
    message.readBytes(messageBytes);
    final RawMessage rawMessage = new RawMessage(messageBytes);

    final Message decodedMessage = beatsCodec.decode(rawMessage);
    System.out.println(decodedMessage);

    ctx.fireChannelRead(decodedMessage);
}
 

@Test
public void decodeMessagesReturnsNullIfMessageWasInvalid() throws Exception {
    final byte[] b = "Foobar".getBytes(StandardCharsets.UTF_8);
    final InetSocketAddress source = new InetSocketAddress(InetAddress.getLocalHost(), 12345);
    final RawMessage rawMessage = new RawMessage(b, source);

    final Collection<Message> messages = codec.decodeMessages(rawMessage);
    assertThat(messages).isNull();
}
 

@Test
public void decodeMessagesThrowsEmptyTemplateExceptionWithIncompleteNetFlowV9() throws Exception {
    final byte[] b = Resources.toByteArray(Resources.getResource("netflow-data/netflow-v9-3_incomplete.dat"));
    final InetSocketAddress source = new InetSocketAddress(InetAddress.getLocalHost(), 12345);

    assertThat(codec.decodeMessages(new RawMessage(b, source))).isNull();
}
 

private RawMessage convertToRawMessage(CodecAggregator.Result result, SocketAddress remoteAddress) {
    final ByteBuf buffer = result.getMessage();
    assertThat(buffer).isNotNull();

    final byte[] payload = ByteBufUtil.getBytes(buffer);

    return new RawMessage(payload, (InetSocketAddress) remoteAddress);
}
 

@Test
public void decodeReturnsNullIfPayloadCouldNotBeDecoded() throws Exception {
    assertThat(codec.decode(new RawMessage(new byte[0]))).isNull();
}
 

private RawMessage messageFromJson(String resourceName) throws IOException {
    final URL resource = Resources.getResource(resourceName);
    final byte[] json = Resources.toByteArray(resource);
    return new RawMessage(json);
}
 

@Nullable
@Override
public Message decode(@Nonnull RawMessage rawMessage) {
    throw new UnsupportedOperationException("MultiMessageCodec " + getClass() + " does not support decode()");
}
 

@Test
public void decodeThrowsUnsupportedOperationException() throws Exception {
    assertThatExceptionOfType(UnsupportedOperationException.class)
            .isThrownBy(() -> codec.decode(new RawMessage(new byte[0])))
            .withMessage("MultiMessageCodec " + NetFlowCodec.class + " does not support decode()");
}
 

private Consumer<byte[]> kinesisCallback(final MessageInput input) {
    return (data) -> input.processRawMessage(new RawMessage(data));
}