io.higgs.http.client.readers.FileReader Maven / Gradle / Ivy

Go to download
package io.higgs.http.client.readers;

import io.higgs.core.func.Function2;
import io.higgs.http.client.Response;
import io.netty.buffer.ByteBuf;

import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.nio.file.Files;

/**
 * @author Courtney Robinson 
 */
public class FileReader extends Reader {
    protected File file;
    protected FileOutputStream out;

    public FileReader() throws IOException {
        this(null);
    }

    public FileReader(Function2 function) throws IOException {
        if (function != null) {
            listen(function);
        }
        file = Files.createTempFile("higgs-http-client-temp-" + new Double(Math.random()).longValue(), ".tmp").toFile();
        out = new FileOutputStream(file);
    }

    @Override
    public void data(ByteBuf data) {
        byte[] tmp = new byte[data.readableBytes()];
        data.readBytes(tmp);
        try {
            out.write(tmp);
        } catch (IOException e) {
            log.warn("Error writing data to file", e);
        }
    }

    @Override
    public void done() {
        try {
            out.close();
        } catch (IOException e) {
            log.warn("Unable to close output stream of the downloaded file", e);
        }
        for (Function2 function : functions) {
            function.apply(file, response);
        }
    }
}