BrBETA_dnet-hadoop/dhp-workflows/dhp-broker-events/src/main/java/eu/dnetlib/dhp/broker/model/EventFactory.java

package eu.dnetlib.dhp.broker.model;

import java.text.ParseException;
import java.util.List;
import java.util.stream.Collectors;

import org.apache.commons.codec.digest.DigestUtils;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.lang3.time.DateUtils;

import eu.dnetlib.broker.objects.OaBrokerAuthor;
import eu.dnetlib.broker.objects.OaBrokerMainEntity;
import eu.dnetlib.broker.objects.OaBrokerRelatedDatasource;
import eu.dnetlib.broker.objects.OaBrokerTypedValue;
import eu.dnetlib.dhp.broker.oa.util.BrokerConstants;
import eu.dnetlib.dhp.broker.oa.util.UpdateInfo;

public class EventFactory {

	private static final String PRODUCER_ID = "OpenAIRE";

	private static final String[] DATE_PATTERNS = {
		"yyyy-MM-dd"
	};

	private EventFactory() {
	}

	public static Event newBrokerEvent(final UpdateInfo<?> updateInfo) {

		final Event res = new Event();

		final MappedFields map = createMapFromResult(updateInfo);

		final String eventId = calculateEventId(
			updateInfo.getTopicPath(), updateInfo.getTargetDs().getOpenaireId(), updateInfo
				.getTarget()
				.getOpenaireId(),
			updateInfo.getHighlightValueAsString());

		res.setEventId(eventId);
		res.setProducerId(PRODUCER_ID);
		res.setPayload(updateInfo.asBrokerPayload().toJSON());
		res.setMap(map);
		res.setTopic(updateInfo.getTopicPath());
		res.setCreationDate(0l);
		res.setExpiryDate(Long.MAX_VALUE);
		res.setInstantMessage(false);

		return res;
	}

	private static MappedFields createMapFromResult(final UpdateInfo<?> updateInfo) {
		final MappedFields map = new MappedFields();

		final OaBrokerMainEntity source = updateInfo.getSource();
		final OaBrokerMainEntity target = updateInfo.getTarget();

		final OaBrokerRelatedDatasource targetDs = updateInfo.getTargetDs();

		map.setTargetDatasourceId(targetDs.getOpenaireId());
		map.setTargetDatasourceName(targetDs.getName());
		map.setTargetDatasourceType(targetDs.getType());

		map.setTargetResultId(target.getOpenaireId());

		final List<String> titles = target.getTitles();
		if (!titles.isEmpty()) {
			map.setTargetResultTitle(titles.get(0));
		}

		final long date = parseDateTolong(target.getPublicationdate());
		if (date > 0) {
			map.setTargetDateofacceptance(date);
		}

		map
			.setTargetSubjects(
				target.getSubjects().stream().map(OaBrokerTypedValue::getValue).collect(Collectors.toList()));
		map
			.setTargetAuthors(
				target.getCreators().stream().map(OaBrokerAuthor::getFullname).collect(Collectors.toList()));

		// PROVENANCE INFO
		map.setTrust(updateInfo.getTrust());
		map.setProvenanceResultId(source.getOpenaireId());

		source
			.getDatasources()
			.stream()
			.filter(ds -> ds.getRelType().equals(BrokerConstants.COLLECTED_FROM_REL))
			.findFirst()
			.ifPresent(ds -> {
				map.setProvenanceDatasourceId(ds.getOpenaireId());
				map.setProvenanceDatasourceName(ds.getName());
				map.setProvenanceDatasourceType(ds.getType());
			});

		return map;
	}

	private static String calculateEventId(final String topic,
		final String dsId,
		final String publicationId,
		final String value) {
		return "event-"
			+ DigestUtils.md5Hex(topic).substring(0, 4) + "-"
			+ DigestUtils.md5Hex(dsId).substring(0, 4) + "-"
			+ DigestUtils.md5Hex(publicationId).substring(0, 7) + "-"
			+ DigestUtils.md5Hex(value).substring(0, 5);
	}

	private static long parseDateTolong(final String date) {
		if (StringUtils.isBlank(date)) {
			return -1;
		}
		try {
			return DateUtils.parseDate(date, DATE_PATTERNS).getTime();
		} catch (final ParseException e) {
			return -1;
		}
	}

}