forked from D-Net/dnet-hadoop
48 lines
1.2 KiB
Java
48 lines
1.2 KiB
Java
|
|
package eu.dnetlib.dhp.aggregation.common;
|
|
|
|
import java.io.Closeable;
|
|
import java.io.IOException;
|
|
import java.util.HashMap;
|
|
import java.util.LinkedHashMap;
|
|
import java.util.Map;
|
|
import java.util.Objects;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import com.google.gson.Gson;
|
|
|
|
import eu.dnetlib.dhp.message.MessageSender;
|
|
import eu.dnetlib.dhp.utils.DHPUtils;
|
|
|
|
public class AggregatorReport extends LinkedHashMap<String, String> implements Closeable {
|
|
|
|
private static final Logger log = LoggerFactory.getLogger(AggregatorReport.class);
|
|
|
|
private MessageSender messageSender;
|
|
|
|
public AggregatorReport() {
|
|
}
|
|
|
|
public AggregatorReport(MessageSender messageSender) throws IOException {
|
|
this.messageSender = messageSender;
|
|
}
|
|
|
|
public void ongoing(Long current, Long total) {
|
|
messageSender.sendMessage(current, total);
|
|
}
|
|
|
|
@Override
|
|
public void close() throws IOException {
|
|
if (Objects.nonNull(messageSender)) {
|
|
log.info("closing report: ");
|
|
this.forEach((k, v) -> log.info("{} - {}", k, v));
|
|
|
|
Map<String, String> m = new HashMap<>();
|
|
m.put(getClass().getSimpleName().toLowerCase(), DHPUtils.MAPPER.writeValueAsString(values()));
|
|
messageSender.sendReport(m);
|
|
}
|
|
}
|
|
}
|