/* Copyright (c) 2011 Danish Maritime Authority
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 3 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this library. If not, see <http://www.gnu.org/licenses/>.
*/
package dk.dma.ais.abnormal.event.db.csv;
import com.google.inject.Inject;
import dk.dma.ais.abnormal.event.db.EventRepository;
import dk.dma.ais.abnormal.event.db.domain.Behaviour;
import dk.dma.ais.abnormal.event.db.domain.Event;
import dk.dma.ais.abnormal.event.db.domain.TrackingPoint;
import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVPrinter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.annotation.concurrent.GuardedBy;
import javax.annotation.concurrent.ThreadSafe;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;
import java.time.format.DateTimeFormatter;
import java.util.Date;
import java.util.HashMap;
import java.util.List;
import java.util.Locale;
import java.util.Map;
import java.util.concurrent.locks.ReentrantLock;
import static java.time.temporal.ChronoUnit.MINUTES;
/**
* CsvEventRepository is an implementation of the EventRepository interface which
* manages persistent Event objects in a flat CSV file.
*/
@ThreadSafe
public class CsvEventRepository implements EventRepository {
private static final Logger LOG = LoggerFactory.getLogger(CsvEventRepository.class);
{
LOG.info(this.getClass().getSimpleName() + " created (" + this + ").");
}
private final ReentrantLock lock = new ReentrantLock();
private final CSVPrinter printer;
private final boolean readonly;
@GuardedBy("lock")
private final Map<Integer, Map<Class<? extends Event>, Event>> ongoingEvents = new HashMap<>();
@Inject
public CsvEventRepository(OutputStream out, boolean readonly) throws IOException {
Writer writer = new OutputStreamWriter(out);
printer = CSVFormat.RFC4180
.withHeader(
"eventId", "eventType", "startTime", "endTime", "title",
"description", "mmsis",
"pMmsi", "pName", "pCallsign", "pType", "pLength", "pLat", "pLon", "pSog", "pCog", "pHdg",
"sMmsi", "sName", "sCallsign", "sType", "sLength", "sLat", "sLon", "sSog", "sCog", "sHdg",
"filterSuggest"
).print(writer);
this.readonly = readonly;
}
@Override
public List<String> getEventTypes() {
throw new UnsupportedOperationException();
}
@Override
public void save(Event event) {
if (readonly == false) {
if (event.getState() == Event.State.ONGOING)
insertOngoingEvent(event);
else
removeOngoingEvent(event);
if (event.getState() == Event.State.PAST) {
try {
Behaviour primaryBehaviour = event.primaryBehaviour();
Behaviour secondaryBehaviour = event.arbitraryNonPrimaryBehaviour();
TrackingPoint primaryLastTrackingPoint = primaryBehaviour.getTrackingPoints().last();
TrackingPoint secondaryLastTrackingPoint = secondaryBehaviour == null ? null : secondaryBehaviour.getTrackingPoints().last();
printer.printRecord(
event.getId(),
event.getEventType(),
event.getStartTime(),
event.getEndTime(),
event.getTitle(),
event.getDescription(),
event.involvedMmsis(),
primaryBehaviour.getVessel().getMmsi(),
primaryBehaviour.getVessel().getName(),
primaryBehaviour.getVessel().getCallsign(),
primaryBehaviour.getVessel().getType(),
primaryBehaviour.getVessel().getLength(),
String.format(Locale.ENGLISH, "%.4f", primaryLastTrackingPoint.getLatitude()),
String.format(Locale.ENGLISH, "%.4f", primaryLastTrackingPoint.getLongitude()),
String.format(Locale.ENGLISH, "%.1f", primaryBehaviour.mostRecentTrackingPoint().getSpeedOverGround()),
String.format(Locale.ENGLISH, "%.0f", primaryBehaviour.mostRecentTrackingPoint().getCourseOverGround()),
String.format(Locale.ENGLISH, "%.0f", primaryBehaviour.mostRecentTrackingPoint().getTrueHeading()),
secondaryBehaviour == null ? null : secondaryBehaviour.getVessel().getMmsi(),
secondaryBehaviour == null ? null : secondaryBehaviour.getVessel().getName(),
secondaryBehaviour == null ? null : secondaryBehaviour.getVessel().getCallsign(),
secondaryBehaviour == null ? null : secondaryBehaviour.getVessel().getType(),
secondaryBehaviour == null ? null : secondaryBehaviour.getVessel().getLength(),
secondaryLastTrackingPoint == null ? null : String.format(Locale.ENGLISH, "%.4f", secondaryLastTrackingPoint.getLatitude()),
secondaryLastTrackingPoint == null ? null : String.format(Locale.ENGLISH, "%.4f", secondaryLastTrackingPoint.getLongitude()),
secondaryBehaviour == null ? null : String.format(Locale.ENGLISH, "%.1f", secondaryBehaviour.mostRecentTrackingPoint().getSpeedOverGround()),
secondaryBehaviour == null ? null : String.format(Locale.ENGLISH, "%.0f", secondaryBehaviour.mostRecentTrackingPoint().getCourseOverGround()),
secondaryBehaviour == null ? null : String.format(Locale.ENGLISH, "%.0f", secondaryBehaviour.mostRecentTrackingPoint().getTrueHeading()),
filterSuggestion(event)
);
printer.flush();
} catch (IOException e) {
LOG.error(e.getMessage(), e);
}
}
}
}
private String filterSuggestion(Event event) {
DateTimeFormatter fmt = DateTimeFormatter.ofPattern("yyyy-MM-dd-HH:mm");
TrackingPoint tp = event.primaryBehaviour().mostRecentTrackingPoint();
StringBuilder sb = new StringBuilder();
sb.append("-start \"")
.append(fmt.format(event.getStartTime().minus(10, MINUTES)))
.append("\" ")
.append("-end \"")
.append(fmt.format(event.getEndTime() == null ? event.getStartTime().plus(20, MINUTES) : event.getEndTime().plus(10, MINUTES)))
.append("\" ")
.append("-exp \"")
.append("m.pos within circle(")
.append(String.format(Locale.ENGLISH, "%.4f", tp.getLatitude()))
.append(",")
.append(String.format(Locale.ENGLISH, "%.4f", tp.getLongitude()))
.append(",")
.append(1000)
.append(")")
.append("\"");
return sb.toString();
}
@Override
public Event getEvent(long eventId) {
throw new UnsupportedOperationException();
}
@Override
public List<Event> findEventsByFromAndToAndTypeAndVesselAndArea(Date from, Date to, String type, String vessel, Double north, Double east, Double south, Double west) {
throw new UnsupportedOperationException();
}
@Override
public List<Event> findEventsByFromAndTo(Date from, Date to) {
throw new UnsupportedOperationException();
}
@Override
public List<Event> findRecentEvents(int numberOfEvents) {
throw new UnsupportedOperationException();
}
@Override
public <T extends Event> T findOngoingEventByVessel(int mmsi, Class<T> eventClass) {
Event event = null;
lock.lock();
try {
Map<Class<? extends Event>, Event> eventMap = ongoingEvents.get(mmsi);
if (eventMap != null)
event = eventMap.get(eventClass);
} finally {
lock.unlock();
}
return event == null ? null : (T) event;
}
private void insertOngoingEvent(Event event) {
lock.lock();
try {
if (event.getState() == Event.State.ONGOING) {
event.involvedMmsis().forEach(mmsi -> {
Map<Class<? extends Event>, Event> eventMap = ongoingEvents.get(mmsi);
if (eventMap == null) {
eventMap = new HashMap<>();
ongoingEvents.put(mmsi, eventMap);
}
eventMap.put(event.getClass(), event);
});
}
} finally {
lock.unlock();
}
}
private void removeOngoingEvent(Event event) {
lock.lock();
try {
event.involvedMmsis().forEach(mmsi -> {
Map<Class<? extends Event>, Event> eventMap = ongoingEvents.get(mmsi);
if (eventMap != null) {
eventMap.remove(event.getClass());
}
});
} finally {
lock.unlock();
}
}
}