D-Net

package eu.dnetlib.data.mapreduce.hbase.statsExport.utils;

import java.io.BufferedWriter;

import java.io.FileOutputStream;

import java.io.OutputStreamWriter;

import java.io.Writer;

import org.apache.log4j.Logger;

public class ContextExporter {

	private String outputPath;

	private String contextMap;

	private boolean readFromURL = false;

	private ContextTransformer contextTransformer;

	private Logger log = Logger.getLogger(this.getClass());

	public ContextExporter(String outputPath, String contextMap, boolean readFromURL) {

		this.outputPath = outputPath;

		this.contextMap = contextMap;

		this.readFromURL = readFromURL;

		contextTransformer = new ContextTransformer();

		if (!outputPath.endsWith("/")) {

			outputPath += "/";

		}

	}

	public void exportToFile() throws Exception {

		try {

			String data = contextTransformer.transformXSL(contextMap);

			String[] split = data.split("COPY");

			log.info("Importing context, concept and category...");

			split[0]=split[0].replaceFirst("","");

			writeData(split[0], "context");

			writeData(split[1], "category");

			writeData(split[2], "concept");

		} catch (Exception e) {

			String msg = " Unable to create file with context, " + "concept and category values in output path " + outputPath + ". Reason: ";

			log.error(msg + e);

			throw new Exception(msg, e);

		}

	}

	private void writeData(String data, String name) throws Exception {

		Writer fileWriter = null;

		try {

			data=data.replaceFirst("\n","");

			fileWriter = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(outputPath + name + ".txt"), "UTF8"));

			fileWriter.write(data);

		} catch (Exception e) {

			log.error(e);

			throw new Exception("Failed  to write exported data to a file", e);

		} finally {

			fileWriter.close();

		}

	}

	public String getOutputPath() {

		return outputPath;

	}

	public void setOutputPath(String outputPath) {

		this.outputPath = outputPath;

	}

	public String getContextMap() {

		return contextMap;

	}

	public void setContextMap(String contextMap) {

		this.contextMap = contextMap;

	}

	public boolean isReadFromURL() {

		return readFromURL;

	}

	public void setReadFromURL(boolean readFromURL) {

		this.readFromURL = readFromURL;

	}

}