Examples with RecordingByteChannel org.neo4j.bolt.v1.messaging.RecordingByteChannel used on opensource projects

Example 1 with RecordingByteChannel

use of org.neo4j.bolt.v1.messaging.RecordingByteChannel in project neo4j by neo4j.

the class FragmentedMessageDeliveryTest method serialize.

private byte[] serialize(int chunkSize, RequestMessage... msgs) throws IOException {
    byte[][] serialized = new byte[msgs.length][];
    for (int i = 0; i < msgs.length; i++) {
        RecordingByteChannel channel = new RecordingByteChannel();
        BoltRequestMessageWriter writer = new BoltRequestMessageWriter(new Neo4jPack.Packer(new BufferedChannelOutput(channel)), NO_BOUNDARY_HOOK);
        writer.write(msgs[i]).flush();
        serialized[i] = channel.getBytes();
    }
    return Chunker.chunk(chunkSize, serialized);
}

Example 2 with RecordingByteChannel

use of org.neo4j.bolt.v1.messaging.RecordingByteChannel in project neo4j by neo4j.

the class TransportErrorIT method shouldHandleUnknownMarkerBytes.

@Test
public void shouldHandleUnknownMarkerBytes() throws Throwable {
    // Given I send a message with an invalid type
    final RecordingByteChannel rawData = new RecordingByteChannel();
    final BufferedChannelOutput out = new BufferedChannelOutput(rawData);
    final PackStream.Packer packer = new PackStream.Packer(out);
    packer.packStructHeader(2, RUN.signature());
    // Invalid marker byte
    out.writeByte(PackStream.RESERVED_C4);
    out.flush();
    byte[] invalidMessage = rawData.getBytes();
    // When
    client.connect(address).send(acceptedVersions(1, 0, 0, 0)).send(chunk(32, invalidMessage));
    // Then
    assertThat(client, eventuallyReceives(new byte[] { 0, 0, 0, 1 }));
    assertThat(client, eventuallyDisconnects());
}

Example 3 with RecordingByteChannel

use of org.neo4j.bolt.v1.messaging.RecordingByteChannel in project neo4j by neo4j.

the class MessageMatchers method serialize.

public static byte[] serialize(ResponseMessage... messages) throws IOException {
    final RecordingByteChannel rawData = new RecordingByteChannel();
    final BoltResponseMessageWriter packer = new BoltResponseMessageWriter(new Neo4jPack.Packer(new BufferedChannelOutput(rawData)), NO_BOUNDARY_HOOK);
    for (ResponseMessage message : messages) {
        message.dispatch(packer);
    }
    packer.flush();
    return rawData.getBytes();
}

Example 4 with RecordingByteChannel

use of org.neo4j.bolt.v1.messaging.RecordingByteChannel in project neo4j by neo4j.

the class MessageMatchers method serialize.

public static byte[] serialize(RequestMessage... messages) throws IOException {
    final RecordingByteChannel rawData = new RecordingByteChannel();
    final BoltRequestMessageWriter packer = new BoltRequestMessageWriter(new Neo4jPack.Packer(new BufferedChannelOutput(rawData)), NO_BOUNDARY_HOOK);
    for (RequestMessage message : messages) {
        packer.write(message);
    }
    packer.flush();
    return rawData.getBytes();
}

Example 5 with RecordingByteChannel

use of org.neo4j.bolt.v1.messaging.RecordingByteChannel in project neo4j by neo4j.

the class BoltResponseMessageTest method serializeAndDeserialize.

private <T extends ResponseMessage> T serializeAndDeserialize(T msg) throws IOException {
    RecordingByteChannel channel = new RecordingByteChannel();
    BoltResponseMessageReader reader = new BoltResponseMessageReader(new Neo4jPack.Unpacker(new BufferedChannelInput(16).reset(channel)));
    BoltResponseMessageWriter writer = new BoltResponseMessageWriter(new Neo4jPack.Packer(new BufferedChannelOutput(channel)), NO_BOUNDARY_HOOK);
    msg.dispatch(writer);
    writer.flush();
    channel.eof();
    return unpack(reader, channel);
}