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

Go to download
package com.documents4j.job;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.concurrent.atomic.AtomicBoolean;

class DeleteFileOnCloseInputStream extends InputStream {

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

    private final File file;
    private final InputStream underlyingInputStream;

    private final AtomicBoolean deletedMark;

    public DeleteFileOnCloseInputStream(File file) throws IOException {
        this.file = file;
        FileInputStream fileInputStream = new FileInputStream(file);
        fileInputStream.getChannel().lock(0L, Long.MAX_VALUE, true);
        this.underlyingInputStream = fileInputStream;
        this.deletedMark = new AtomicBoolean(false);
    }

    @Override
    public int read() throws IOException {
        return underlyingInputStream.read();
    }

    @Override
    public int read(byte[] b) throws IOException {
        return underlyingInputStream.read(b);
    }

    @Override
    public int read(byte[] b, int off, int len) throws IOException {
        return underlyingInputStream.read(b, off, len);
    }

    @Override
    public long skip(long n) throws IOException {
        return underlyingInputStream.skip(n);
    }

    @Override
    public int available() throws IOException {
        return underlyingInputStream.available();
    }

    @Override
    public void close() throws IOException {
        try {
            // Note: This will implicitly release the file lock.
            underlyingInputStream.close();
        } finally {
            if (deletedMark.compareAndSet(false, true)) {
                if (!file.delete()) {
                    LOGGER.warn("Could not delete file {} after input stream was closed", file);
                }
            }
        }
    }

    @Override
    public synchronized void mark(int readlimit) {
        underlyingInputStream.mark(readlimit);
    }

    @Override
    public synchronized void reset() throws IOException {
        underlyingInputStream.reset();
    }

    @Override
    public boolean markSupported() {
        return underlyingInputStream.markSupported();
    }
}