Contents

package org.embulk.standards;

import java.io.OutputStream;
import java.io.IOException;
import java.util.zip.GZIPOutputStream;
import org.embulk.config.Task;
import org.embulk.config.Config;
import org.embulk.config.ConfigInject;
import org.embulk.config.ConfigDefault;
import org.embulk.config.TaskSource;
import org.embulk.config.ConfigSource;
import org.embulk.spi.EncoderPlugin;
import org.embulk.spi.FileOutput;
import org.embulk.spi.BufferAllocator;
import org.embulk.spi.util.FileOutputOutputStream;
import org.embulk.spi.util.OutputStreamFileOutput;

public class GzipFileEncoderPlugin
        implements EncoderPlugin
{
    public interface PluginTask
            extends Task
    {
        // TODO java.util.zip.GZIPOutputStream doesn't support compression level
        //@Config("level")
        //@ConfigDefault("6")
        //public int getLevel();

        @ConfigInject
        public BufferAllocator getBufferAllocator();
    }

    public void transaction(ConfigSource config, EncoderPlugin.Control control)
    {
        PluginTask task = config.loadConfig(PluginTask.class);
        control.run(task.dump());
    }

    @Override
    public FileOutput open(TaskSource taskSource, final FileOutput fileOutput)
    {
        PluginTask task = taskSource.loadTask(PluginTask.class);

        final FileOutputOutputStream output = new FileOutputOutputStream(fileOutput, task.getBufferAllocator(), FileOutputOutputStream.CloseMode.CLOSE);

        return new OutputStreamFileOutput(new OutputStreamFileOutput.Provider() {
            public OutputStream openNext() throws IOException
            {
                output.nextFile();
                return new GZIPOutputStream(output);
            }

            public void finish() throws IOException
            {
                fileOutput.finish();
            }

            public void close() throws IOException
            {
                fileOutput.close();
            }
        });
    }
}

Version data entries

5 entries across 5 versions & 1 rubygems

Version	Path
embulk-0.4.8	embulk-standards/src/main/java/org/embulk/standards/GzipFileEncoderPlugin.java
embulk-0.4.7	embulk-standards/src/main/java/org/embulk/standards/GzipFileEncoderPlugin.java
embulk-0.4.6	embulk-standards/src/main/java/org/embulk/standards/GzipFileEncoderPlugin.java
embulk-0.4.5	embulk-standards/src/main/java/org/embulk/standards/GzipFileEncoderPlugin.java
embulk-0.4.4	embulk-standards/src/main/java/org/embulk/standards/GzipFileEncoderPlugin.java