com.documents4j.job.FileSourceFromInputStreamSource Maven / Gradle / Ivy

Go to download
package com.documents4j.job;

import com.documents4j.api.IFileSource;
import com.documents4j.api.IInputStreamSource;
import com.documents4j.throwables.FileSystemInteractionException;
import com.google.common.base.MoreObjects;
import com.google.common.io.ByteStreams;
import com.google.common.io.Closeables;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;

class FileSourceFromInputStreamSource implements IFileSource {

    private static final Logger LOGGER = LoggerFactory.getLogger(FileSourceFromInputStreamSource.class);

    private final IInputStreamSource inputStreamSource;

    private final File tempStorage;

    private InputStream inputStream;

    public FileSourceFromInputStreamSource(IInputStreamSource inputStreamSource, File tempStorage) {
        this.inputStreamSource = inputStreamSource;
        this.tempStorage = tempStorage;
    }

    @Override
    public File getFile() {
        inputStream = inputStreamSource.getInputStream();
        try {
            FileOutputStream fileOutputStream = new FileOutputStream(tempStorage);
            fileOutputStream.getChannel().lock();
            try {
                ByteStreams.copy(inputStream, fileOutputStream);
                return tempStorage;
            } finally {
                // Note: This will implicitly release the file lock.
                Closeables.close(fileOutputStream, true);
            }
        } catch (IOException e) {
            throw new FileSystemInteractionException(String.format("Could not write stream to file %s", tempStorage), e);
        }
    }

    @Override
    public void onConsumed(File file) {
        try {
            if (!tempStorage.delete()) {
                LOGGER.warn("Could not delete temporary file {}", tempStorage);
            }
        } finally {
            inputStreamSource.onConsumed(inputStream);
        }
    }

    @Override
    public String toString() {
        return MoreObjects.toStringHelper(FileSourceFromInputStreamSource.class)
                .add("inputStreamSource", inputStreamSource)
                .add("temporaryFile", tempStorage)
                .toString();
    }
}