org.mariadb.r2dbc.client.MariadbPacketEncoder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of r2dbc-mariadb-0.9.1-spec Show documentation

R2DBC driver for MariaDB

The newest version!

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2020-2024 MariaDB Corporation Ab

package org.mariadb.r2dbc.client;

import io.netty.buffer.ByteBuf;
import io.netty.buffer.ByteBufAllocator;
import io.netty.buffer.CompositeByteBuf;
import io.netty.buffer.Unpooled;
import org.mariadb.r2dbc.message.ClientMessage;
import org.mariadb.r2dbc.message.Context;
import reactor.core.publisher.Mono;

public class MariadbPacketEncoder {
  private Context context = null;

  public Mono encodeFlux(ClientMessage msg) {
    ByteBufAllocator allocator = context.getByteBufAllocator();

    return msg.encode(context, allocator)
        .map(
            buf -> {
              CompositeByteBuf out = allocator.compositeBuffer();

              int initialReaderIndex = buf.readerIndex();
              int packetLength;
              do {
                packetLength = Math.min(0xffffff, buf.readableBytes());

                ByteBuf header = Unpooled.buffer(4, 4);
                header.writeMediumLE(packetLength);
                header.writeByte(msg.getSequencer().next());

                out.addComponent(true, header);
                out.addComponent(true, buf.readRetainedSlice(packetLength));

              } while (buf.readableBytes() > 0);

              if (packetLength == 0xffffff) {
                // in case last packet is full, sending an empty packet to indicate that command is
                // complete
                ByteBuf header = Unpooled.buffer(4, 4);
                header.writeMediumLE(0);
                header.writeByte(msg.getSequencer().next());
                out.addComponent(true, header);
              }

              context.saveRedo(msg, buf, initialReaderIndex);
              buf.release();
              return out;
            });
  }

  public void setContext(Context context) {
    this.context = context;
  }
}