D-Net

package eu.dnetlib.msro.workflows.nodes.info;

import java.io.StringReader;

import java.util.Map;

import javax.annotation.Resource;

import org.dom4j.Document;

import org.dom4j.Node;

import org.dom4j.io.SAXReader;

import org.springframework.beans.factory.annotation.Required;

import com.google.common.collect.Maps;

import com.googlecode.sarasvati.Arc;

import com.googlecode.sarasvati.NodeToken;

import eu.dnetlib.datasource.common.utils.DatasourceUpdater;

import eu.dnetlib.enabling.is.lookup.rmi.ISLookUpService;

public class ObjectStoreToApiExtraFieldJobNode extends SimpleJobNode {

	private String objId;

	private String datasourceId;

	private String datasourceInterface;

	private String extraFieldForTotal;

	private String extraFieldForDate;

	private String extraFieldForObjId;

	private UniqueServiceLocator serviceLocator;

	private DatasourceUpdater datasourceUpdater;

	@Override

	protected String execute(final NodeToken token) throws Exception {

		final String xq = "for $x in collection('/db/DRIVER/ObjectStoreDSResources/ObjectStoreDSResourceType') " + "where $x//RESOURCE_IDENTIFIER/@value='"

				+ objId + "' " + "return concat($x//COUNT_STORE, ' @=@ ', $x//LAST_STORAGE_DATE)";

		final String[] arr = s.split(" @=@ ");

		final Map<String, String> map = getCurrentExtraFields(datasourceId, datasourceInterface);

		map.put(extraFieldForTotal, arr[0].trim());

		map.put(extraFieldForDate, arr[1].trim());

		map.put(extraFieldForObjId, objId);

		datasourceUpdater.updateApiExtraFields(datasourceId, datasourceInterface, map);

		return Arc.DEFAULT_ARC;

	}

	private Map<String, String> getCurrentExtraFields(final String repoId, final String ifaceId) throws Exception {

		final Map<String, String> res = Maps.newHashMap();

		final SAXReader reader = new SAXReader();

		final Document doc = reader.read(new StringReader(profile));

		final Node ifcNode = doc.selectSingleNode("//INTERFACE[@id='" + ifaceId + "']");

		if (ifcNode != null) {

			for (Object o : ifcNode.selectNodes("./INTERFACE_EXTRA_FIELD")) {

				res.put(((Node) o).valueOf("@name"), ((Node) o).getText());

			}

		}

		return res;

	}

	public String getDatasourceId() {

		return datasourceId;

	}

	public void setDatasourceId(final String datasourceId) {

		this.datasourceId = datasourceId;

	}

	public String getDatasourceInterface() {

		return datasourceInterface;

	}

	public void setDatasourceInterface(final String datasourceInterface) {

		this.datasourceInterface = datasourceInterface;

	}

	public String getExtraFieldForTotal() {

		return extraFieldForTotal;

	}

	public void setExtraFieldForTotal(final String extraFieldForTotal) {

		this.extraFieldForTotal = extraFieldForTotal;

	}

	public String getExtraFieldForDate() {

		return extraFieldForDate;

	}

	public void setExtraFieldForDate(final String extraFieldForDate) {

		this.extraFieldForDate = extraFieldForDate;

	}

	public DatasourceUpdater getDatasourceUpdater() {

		return datasourceUpdater;

	}

	@Required

	public void setDatasourceUpdater(final DatasourceUpdater datasourceUpdater) {

		this.datasourceUpdater = datasourceUpdater;

	}

	/**

	 * @return the objId

	 */

	public String getObjId() {

		return objId;

	}

	/**

	 * @param objId

	 *            the objId to set

	 */

	public void setObjId(final String objId) {

		this.objId = objId;

	}

	/**

	 * @return the extraFieldForObjId

	 */

	public String getExtraFieldForObjId() {

		return extraFieldForObjId;

	}

	/**

	 * @param extraFieldForObjId

	 *            the extraFieldForObjId to set

	 */

	public void setExtraFieldForObjId(final String extraFieldForObjId) {

		this.extraFieldForObjId = extraFieldForObjId;

	}

}