DatasetDBProducer.java

package eu.dnetlib.iis.wf.referenceextraction.dataset;

import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.HashMap;
import java.util.Map;

import org.apache.commons.io.IOUtils;
import org.apache.hadoop.conf.Configuration;
import org.apache.hadoop.fs.FileSystem;
import org.apache.hadoop.fs.Path;

import eu.dnetlib.iis.common.java.PortBindings;
import eu.dnetlib.iis.common.java.Process;
import eu.dnetlib.iis.common.java.io.FileSystemPath;
import eu.dnetlib.iis.common.java.porttype.AnyPortType;
import eu.dnetlib.iis.common.java.porttype.PortType;


public class DatasetDBProducer implements Process {
	
	private final static String datasetDBPort = "dataset_db";

    @Override
	public Map<String, PortType> getInputPorts() {
		return new HashMap<String, PortType>();
	}
	
	@Override
	public Map<String, PortType> getOutputPorts() {
		return createOutputPorts();
	}

	private static Map<String, PortType> createOutputPorts(){
		Map<String, PortType> outputPorts = new HashMap<String, PortType>();
		outputPorts.put(datasetDBPort, new AnyPortType());
		return outputPorts;
	}
	
    @Override
    public void run(PortBindings portBindings, Configuration conf,
            Map<String, String> parameters)	throws IOException{
        Map<String, Path> output = portBindings.getOutput();
        
        FileSystem fs = FileSystem.get(conf);
        
        try (
                InputStream inStream = this.getClass().getResourceAsStream("/eu/dnetlib/iis/wf/referenceextraction/dataset/datacite/data/datasets.db");
                OutputStream outStream = fs.create(new FileSystemPath(fs, output.get(datasetDBPort)).getPath())) {
            
            IOUtils.copy(inStream, outStream);
        }
        
	}

}