eu.clarussecure.proxy.protocol.plugins.pgsql.message.writer.PgsqlExecuteMessageWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of proxy-plugin-pgsql Show documentation

The newest version!

package eu.clarussecure.proxy.protocol.plugins.pgsql.message.writer;

import java.io.IOException;
import java.util.Collections;
import java.util.Map;

import eu.clarussecure.proxy.protocol.plugins.pgsql.message.PgsqlExecuteMessage;
import io.netty.buffer.ByteBuf;

public class PgsqlExecuteMessageWriter implements PgsqlMessageWriter {

    @Override
    public int contentSize(PgsqlExecuteMessage msg) {
        // Get content size
        int size = msg.getPortal().clen();
        size += Integer.BYTES;
        return size;
    }

    @Override
    public Map offsets(PgsqlExecuteMessage msg) {
        // Compute header size
        int headerSize = msg.getHeaderSize();
        // Compute buffer offsets
        int offset = headerSize;
        Map offsets = Collections.singletonMap(offset, msg.getPortal().getByteBuf());
        return offsets;
    }

    @Override
    public void writeContent(PgsqlExecuteMessage msg, ByteBuf buffer) throws IOException {
        // Write portal
        ByteBuf value = msg.getPortal().getByteBuf();
        writeBytes(buffer, value);
        // Write max rows
        buffer.writeInt(msg.getMaxRows());
    }
}