Examples with PerSessionSerializationService org.apache.ignite.internal.network.serialization.PerSessionSerializationService used on opensource projects

Example 1 with PerSessionSerializationService

use of org.apache.ignite.internal.network.serialization.PerSessionSerializationService in project ignite-3 by apache.

the class InboundDecoderTest method testPartialReadWithReuseBuffer.

/**
 * Tests that an {@link InboundDecoder} can handle a {@link ByteBuf} where reader index is not {@code 0} at the start of the {@link
 * InboundDecoder#decode}.
 *
 * @throws Exception If failed.
 */
@Test
public void testPartialReadWithReuseBuffer() throws Exception {
    ChannelHandlerContext ctx = Mockito.mock(ChannelHandlerContext.class);
    var channel = new EmbeddedChannel();
    Mockito.doReturn(channel).when(ctx).channel();
    var serializationService = new SerializationService(registry, mock(UserObjectSerializationContext.class));
    var perSessionSerializationService = new PerSessionSerializationService(serializationService);
    final var decoder = new InboundDecoder(perSessionSerializationService);
    final var list = new ArrayList<>();
    var writer = new DirectMessageWriter(perSessionSerializationService, ConnectionManager.DIRECT_PROTOCOL_VERSION);
    var msg = new TestMessagesFactory().testMessage().msg("abcdefghijklmn").build();
    MessageSerializer<NetworkMessage> serializer = registry.createSerializer(msg.groupType(), msg.messageType());
    ByteBuffer nioBuffer = ByteBuffer.allocate(10_000);
    writer.setBuffer(nioBuffer);
    // Write message to the ByteBuffer.
    boolean fullyWritten = serializer.writeMessage(msg, writer);
    assertTrue(fullyWritten);
    nioBuffer.flip();
    ByteBuf buffer = allocator.buffer();
    // Write first 3 bytes of a message.
    for (int i = 0; i < 3; i++) {
        buffer.writeByte(nioBuffer.get());
    }
    decoder.decode(ctx, buffer, list);
    // At this point a header and a first byte of a message have been decoded.
    assertEquals(0, list.size());
    // Write next 3 bytes of a message.
    for (int i = 0; i < 3; i++) {
        buffer.writeByte(nioBuffer.get());
    }
    // Reader index of a buffer is not zero and it must be handled correctly by the InboundDecoder.
    decoder.decode(ctx, buffer, list);
    // Check if reader index has been tracked correctly.
    assertEquals(6, buffer.readerIndex());
    assertEquals(0, list.size());
    buffer.writeBytes(nioBuffer);
    decoder.decode(ctx, buffer, list);
    assertEquals(1, list.size());
    TestMessage actualMessage = (TestMessage) list.get(0);
    assertEquals(msg, actualMessage);
}

Example 2 with PerSessionSerializationService

use of org.apache.ignite.internal.network.serialization.PerSessionSerializationService in project ignite-3 by apache.

the class NettyServer method start.

/**
 * Starts the server.
 *
 * @return Future that resolves when the server is successfully started.
 */
public CompletableFuture<Void> start() {
    synchronized (startStopLock) {
        if (stopped) {
            throw new IgniteInternalException("Attempted to start an already stopped server");
        }
        if (serverStartFuture != null) {
            throw new IgniteInternalException("Attempted to start an already started server");
        }
        ServerBootstrap bootstrap = bootstrapFactory.createServerBootstrap();
        bootstrap.childHandler(new ChannelInitializer<SocketChannel>() {

            /**
             * {@inheritDoc}
             */
            @Override
            public void initChannel(SocketChannel ch) {
                var sessionSerializationService = new PerSessionSerializationService(serializationService);
                // Get handshake manager for the new channel.
                HandshakeManager manager = handshakeManager.get();
                ch.pipeline().addLast(/*
                                     * Decoder that uses the MessageReader
                                     * to read chunked data.
                                     */
                new InboundDecoder(sessionSerializationService), // Handshake handler.
                new HandshakeHandler(manager, (consistentId) -> new MessageHandler(messageListener, consistentId, sessionSerializationService)), /*
                                     * Encoder that uses the MessageWriter
                                     * to write chunked data.
                                     */
                new ChunkedWriteHandler(), // Converts NetworkMessage to a ChunkedNetworkMessageInput
                new OutboundEncoder(sessionSerializationService), new IoExceptionSuppressingHandler());
                manager.handshakeFuture().thenAccept(newConnectionListener);
            }
        });
        int port = configuration.port();
        int portRange = configuration.portRange();
        var bindFuture = new CompletableFuture<Channel>();
        tryBind(bootstrap, port, port + portRange, port, bindFuture);
        serverStartFuture = bindFuture.handle((channel, err) -> {
            synchronized (startStopLock) {
                if (channel != null) {
                    serverCloseFuture = NettyUtils.toCompletableFuture(channel.closeFuture());
                }
                this.channel = (ServerChannel) channel;
                if (err != null || stopped) {
                    Throwable stopErr = err != null ? err : new CancellationException("Server was stopped");
                    return CompletableFuture.<Void>failedFuture(stopErr);
                } else {
                    return CompletableFuture.<Void>completedFuture(null);
                }
            }
        }).thenCompose(Function.identity());
        return serverStartFuture;
    }
}

Example 3 with PerSessionSerializationService

use of org.apache.ignite.internal.network.serialization.PerSessionSerializationService in project ignite-3 by apache.

the class InboundDecoderTest method testPartialHeader.

/**
 * Tests that an {@link InboundDecoder} doesn't hang if it encounters a byte buffer with only partially written header.
 *
 * @throws Exception If failed.
 */
@Test
public void testPartialHeader() throws Exception {
    var serializationService = new SerializationService(registry, mock(UserObjectSerializationContext.class));
    var perSessionSerializationService = new PerSessionSerializationService(serializationService);
    var channel = new EmbeddedChannel(new InboundDecoder(perSessionSerializationService));
    ByteBuf buffer = allocator.buffer();
    buffer.writeByte(1);
    CompletableFuture.runAsync(() -> {
        channel.writeInbound(buffer);
        // In case if a header was written partially, readInbound() should not loop forever, as
        // there might not be new data anymore (example: remote host gone offline).
        channel.readInbound();
    }).get(3, TimeUnit.SECONDS);
}

Example 4 with PerSessionSerializationService

use of org.apache.ignite.internal.network.serialization.PerSessionSerializationService in project ignite-3 by apache.

the class InboundDecoderTest method sendAndReceive.

/**
 * Serializes and then deserializes the given message.
 */
private <T extends NetworkMessage> T sendAndReceive(T msg) {
    var serializationService = new SerializationService(registry, mock(UserObjectSerializationContext.class));
    var perSessionSerializationService = new PerSessionSerializationService(serializationService);
    var channel = new EmbeddedChannel(new InboundDecoder(perSessionSerializationService));
    var writer = new DirectMessageWriter(perSessionSerializationService, ConnectionManager.DIRECT_PROTOCOL_VERSION);
    MessageSerializer<NetworkMessage> serializer = registry.createSerializer(msg.groupType(), msg.messageType());
    ByteBuffer buf = ByteBuffer.allocate(10_000);
    T received;
    do {
        buf.clear();
        writer.setBuffer(buf);
        serializer.writeMessage(msg, writer);
        buf.flip();
        ByteBuf buffer = allocator.buffer(buf.limit());
        buffer.writeBytes(buf);
        channel.writeInbound(buffer);
    } while ((received = channel.readInbound()) == null);
    return received;
}

Example 5 with PerSessionSerializationService

use of org.apache.ignite.internal.network.serialization.PerSessionSerializationService in project ignite-3 by apache.

the class NettyClient method start.

/**
 * Start client.
 *
 * @param bootstrapTemplate Template client bootstrap.
 * @return Future that resolves when client channel is opened.
 */
public CompletableFuture<NettySender> start(Bootstrap bootstrapTemplate) {
    synchronized (startStopLock) {
        if (stopped) {
            throw new IgniteInternalException("Attempted to start an already stopped NettyClient");
        }
        if (clientFuture != null) {
            throw new IgniteInternalException("Attempted to start an already started NettyClient");
        }
        Bootstrap bootstrap = bootstrapTemplate.clone();
        bootstrap.handler(new ChannelInitializer<SocketChannel>() {

            /**
             * {@inheritDoc}
             */
            @Override
            public void initChannel(SocketChannel ch) {
                var sessionSerializationService = new PerSessionSerializationService(serializationService);
                ch.pipeline().addLast(new InboundDecoder(sessionSerializationService), new HandshakeHandler(handshakeManager, (consistentId) -> new MessageHandler(messageListener, consistentId, sessionSerializationService)), new ChunkedWriteHandler(), new OutboundEncoder(sessionSerializationService), new IoExceptionSuppressingHandler());
            }
        });
        clientFuture = NettyUtils.toChannelCompletableFuture(bootstrap.connect(address)).handle((channel, throwable) -> {
            synchronized (startStopLock) {
                this.channel = channel;
                if (throwable != null) {
                    channelFuture.completeExceptionally(throwable);
                } else {
                    channelFuture.complete(null);
                }
                if (stopped) {
                    return CompletableFuture.<NettySender>failedFuture(new CancellationException("Client was stopped"));
                } else if (throwable != null) {
                    return CompletableFuture.<NettySender>failedFuture(throwable);
                } else {
                    return handshakeManager.handshakeFuture();
                }
            }
        }).thenCompose(Function.identity());
        return clientFuture;
    }
}