dnet-applications/apps/dnet-is-application/src/main/java/eu/dnetlib/is/importer/WfHistoryImporter.java

package eu.dnetlib.is.importer;

import java.io.File;
import java.time.Instant;
import java.time.LocalDateTime;
import java.util.Iterator;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Map.Entry;
import java.util.TimeZone;

import org.apache.commons.lang3.BooleanUtils;
import org.apache.commons.lang3.math.NumberUtils;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;

import eu.dnetlib.manager.history.model.WfHistoryEntry;
import eu.dnetlib.manager.history.repository.WfHistoryEntryRepository;

@Service
public class WfHistoryImporter {

	private static final Log log = LogFactory.getLog(WfHistoryImporter.class);

	@Autowired
	private WfHistoryEntryRepository wfHistoryEntryRepository;

	public void load(final String path) throws Exception {
		final ObjectMapper mapper = new ObjectMapper();

		final JsonNode rootNode = mapper.readTree(new File(path));

		rootNode.forEach(wf -> saveWf(wf));
	}

	private void saveWf(final JsonNode node) {

		final WfHistoryEntry wf = new WfHistoryEntry();
		wf.setProcessId(node.get("system:processId").asText());
		wf.setName(node.get("system:wfName").asText());
		wf.setFamily(node.get("system:profileFamily").asText());

		if (node.has("dataprovider:id")) {
			wf.setDsId(node.get("dataprovider:id").asText());
		}
		if (node.has("dataprovider:name")) {
			wf.setDsName(node.get("dataprovider:name").asText());
		}
		if (node.has("dataprovider:interface")) {
			wf.setDsApi(node.get("dataprovider:interface").asText());
		}

		wf.setStartDate(LocalDateTime
			.ofInstant(Instant.ofEpochMilli(NumberUtils.toLong(node.get("system:startDate").asText())), TimeZone.getDefault().toZoneId()));

		wf.setEndDate(LocalDateTime.ofInstant(Instant.ofEpochMilli(NumberUtils.toLong(node.get("system:endDate").asText())), TimeZone.getDefault().toZoneId()));

		if (BooleanUtils.toBoolean(node.get("system:isCompletedSuccessfully").asText())) {
			wf.setStatus("success");
		} else {
			wf.setStatus("failure");
		}

		final Map<String, String> details = new LinkedHashMap<>();
		final Iterator<Entry<String, JsonNode>> fields = node.fields();
		while (fields.hasNext()) {
			final Entry<String, JsonNode> f = fields.next();
			if (f.getValue().isValueNode()) {
				details.put(f.getKey(), f.getValue().asText());
			}
		}

		wf.setDetails(details);

		wfHistoryEntryRepository.save(wf);

		log.info("Wf saved with id: " + wf.getProcessId());

	}
}