eu.clarussecure.proxy.protocol.plugins.pgsql.message.writer.PgsqlCommandCompleteMessageWriter 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.PgsqlCommandCompleteMessage;
import io.netty.buffer.ByteBuf;

public class PgsqlCommandCompleteMessageWriter implements PgsqlMessageWriter {

    @Override
    public int contentSize(PgsqlCommandCompleteMessage msg) {
        // Get content size
        return msg.getTag().clen();
    }

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

    @Override
    public void writeContent(PgsqlCommandCompleteMessage msg, ByteBuf buffer) throws IOException {
        // Write tag
        ByteBuf value = msg.getTag().getByteBuf();
        writeBytes(buffer, value);
    }

}