com.poiji.bind.mapping.XSSFUnmarshallerStream Maven / Gradle / Ivy

Go to download
package com.poiji.bind.mapping;

import com.poiji.bind.PoijiInputStream;
import com.poiji.exception.PoijiException;
import com.poiji.option.PoijiOptions;
import java.io.IOException;
import java.util.function.Consumer;
import java.util.stream.Stream;
import javax.xml.parsers.ParserConfigurationException;
import org.apache.poi.openxml4j.exceptions.OpenXML4JException;
import org.apache.poi.openxml4j.opc.OPCPackage;
import org.apache.poi.poifs.filesystem.POIFSFileSystem;
import org.xml.sax.SAXException;

/**
 * Created by hakan on 22/10/2017
 */
final class XSSFUnmarshallerStream extends XSSFUnmarshaller {

    private final PoijiInputStream poijiInputStream;

    XSSFUnmarshallerStream(PoijiInputStream poijiInputStream, PoijiOptions options) {
        super(options);
        this.poijiInputStream = poijiInputStream;
    }

    @Override
    public  void unmarshal(Class type, Consumer consumer) {

        if (options.getPassword() != null) {
            returnFromEncryptedFile(type, consumer);
        } else {
            returnFromExcelFile(type, consumer);
        }
    }

    @Override
    public  Stream stream(final Class type) {
        try {
            if (options.getPassword() != null) {
                POIFSFileSystem fs = new POIFSFileSystem(poijiInputStream.stream());
                return streamOfEncryptedItems(type, fs);
            } else {
                OPCPackage open = OPCPackage.open(poijiInputStream.stream());
                return stream0(type, open);
            }
        } catch (ParserConfigurationException | SAXException | IOException | OpenXML4JException e) {
            throw new PoijiException("Problem occurred while reading data", e);
        }
    }

    private   void returnFromExcelFile(Class type, Consumer consumer) {
        try (OPCPackage open = OPCPackage.open(poijiInputStream.stream())) {
            unmarshal0(type, consumer, open);
        } catch (ParserConfigurationException | SAXException | IOException | OpenXML4JException e) {
            throw new PoijiException("Problem occurred while reading data", e);
        }
    }

    private   void returnFromEncryptedFile(Class type, Consumer consumer) {
        try (POIFSFileSystem fs = new POIFSFileSystem(poijiInputStream.stream())) {
            listOfEncryptedItems(type, consumer, fs);
        } catch (IOException e) {
            throw new PoijiException("Problem occurred while reading data", e);
        }
    }

}