io.legaldocml.util.StringWriterTemplate Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of legaldocml-core Show documentation
The newest version!
package io.legaldocml.util;

import io.legaldocml.io.XmlWriter;
import io.legaldocml.io.impl.Buffers;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.MappedByteBuffer;
import java.nio.charset.StandardCharsets;
import java.util.function.Consumer;

/**
 * @author Jacques Militello
 */
public abstract class StringWriterTemplate {

    void reserveBuffer(String value) throws IOException {
        MappedByteBuffer buffer = null;
        try {
            byte[] bytes = value.getBytes(StandardCharsets.US_ASCII);
            buffer = (MappedByteBuffer) ByteBuffer.allocateDirect(bytes.length);
            buffer.put(bytes);
            buffer.flip();
            long addr = ((sun.nio.ch.DirectBuffer) buffer).address();

            onReserveBuffer(addr, bytes.length);


        } finally {
            if (buffer != null) {
                Buffers.releaseDirectByteBuffer(buffer);
            }
        }
    }

    protected abstract void onReserveBuffer(long addr, int length) throws IOException;

    public static void writeElement(XmlWriter writer, String element, Consumer consumer) throws IOException {
        new StringWriterTemplate() {
            @Override
            protected void onReserveBuffer(long addr, int length) throws IOException {
                writer.writeStart(addr, length);
                consumer.accept(writer);
                writer.writeEnd(addr, length);
            }
        }.reserveBuffer(element);
    }

    public static void writeAttribute(XmlWriter writer, String attributeName, char[] attributeValue) throws IOException {
        new StringWriterTemplate() {
            @Override
            protected void onReserveBuffer(long addr, int length) throws IOException {
                writer.writeAttribute(addr, length, attributeValue);
            }
        }.reserveBuffer(attributeName);
    }


}