Move streams support classes
[netconf.git] / restconf / restconf-nb-rfc8040 / src / main / java / org / opendaylight / restconf / nb / rfc8040 / streams / listeners / AbstractCommonSubscriber.java
index 751ee83394e495879101c45743a90e17613a9787..d83e15e63bb67206f020db6ba81ad5a634da7b79 100644 (file)
  */
 package org.opendaylight.restconf.nb.rfc8040.streams.listeners;
 
-import com.google.common.base.Preconditions;
+import static com.google.common.base.Preconditions.checkArgument;
+import static com.google.common.base.Preconditions.checkState;
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.base.MoreObjects;
+import java.time.Instant;
+import java.time.format.DateTimeFormatter;
+import java.time.format.DateTimeFormatterBuilder;
+import java.time.format.DateTimeParseException;
+import java.time.temporal.ChronoField;
+import java.time.temporal.TemporalAccessor;
 import java.util.HashSet;
 import java.util.Iterator;
 import java.util.Set;
+import java.util.concurrent.ExecutionException;
+import javax.xml.xpath.XPathExpressionException;
+import org.checkerframework.checker.lock.qual.GuardedBy;
+import org.checkerframework.checker.lock.qual.Holding;
+import org.eclipse.jdt.annotation.NonNull;
+import org.opendaylight.restconf.common.errors.RestconfDocumentedException;
+import org.opendaylight.restconf.nb.rfc8040.NotificationQueryParams;
 import org.opendaylight.restconf.nb.rfc8040.streams.StreamSessionHandler;
-import org.opendaylight.yangtools.concepts.ListenerRegistration;
+import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.DateAndTime;
+import org.opendaylight.yang.gen.v1.urn.sal.restconf.event.subscription.rev140708.NotificationOutputTypeGrouping.NotificationOutputType;
+import org.opendaylight.yangtools.concepts.Registration;
+import org.opendaylight.yangtools.yang.common.QName;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
 /**
  * Features of subscribing part of both notifications.
  */
-abstract class AbstractCommonSubscriber extends AbstractQueryParams implements BaseListenerInterface {
-
+abstract class AbstractCommonSubscriber<P, T> extends AbstractNotificationsData implements BaseListenerInterface {
     private static final Logger LOG = LoggerFactory.getLogger(AbstractCommonSubscriber.class);
+    private static final DateTimeFormatter FORMATTER = new DateTimeFormatterBuilder()
+        .appendValue(ChronoField.YEAR, 4).appendLiteral('-')
+        .appendValue(ChronoField.MONTH_OF_YEAR, 2).appendLiteral('-')
+        .appendValue(ChronoField.DAY_OF_MONTH, 2).appendLiteral('T')
+        .appendValue(ChronoField.HOUR_OF_DAY, 2).appendLiteral(':')
+        .appendValue(ChronoField.MINUTE_OF_HOUR, 2).appendLiteral(':')
+        .appendValue(ChronoField.SECOND_OF_MINUTE, 2)
+        .appendFraction(ChronoField.NANO_OF_SECOND, 0, 9, true)
+        .appendOffset("+HH:MM", "Z").toFormatter();
 
+    private final EventFormatterFactory<T> formatterFactory;
+    private final NotificationOutputType outputType;
+    private final String streamName;
+    private final P path;
+
+    @GuardedBy("this")
     private final Set<StreamSessionHandler> subscribers = new HashSet<>();
-    private volatile ListenerRegistration<?> registration;
+    @GuardedBy("this")
+    private Registration registration;
+
+    // FIXME: these should be final
+    private Instant start = null;
+    private Instant stop = null;
+    private boolean leafNodesOnly = false;
+    private boolean skipNotificationData = false;
+    private EventFormatter<T> formatter;
+
+    AbstractCommonSubscriber(final QName lastQName, final String streamName, final P path,
+            final NotificationOutputType outputType, final EventFormatterFactory<T> formatterFactory) {
+        super(lastQName);
+        this.streamName = requireNonNull(streamName);
+        checkArgument(!streamName.isEmpty());
+        this.path = requireNonNull(path);
+
+        this.outputType = requireNonNull(outputType);
+        this.formatterFactory = requireNonNull(formatterFactory);
+        formatter = formatterFactory.getFormatter();
+    }
+
+    @Override
+    public final String getStreamName() {
+        return streamName;
+    }
+
+    @Override
+    public final String getOutputType() {
+        return outputType.getName();
+    }
 
     @Override
     public final synchronized boolean hasSubscribers() {
-        return !this.subscribers.isEmpty();
+        return !subscribers.isEmpty();
     }
 
     @Override
     public final synchronized Set<StreamSessionHandler> getSubscribers() {
-        return new HashSet<>(this.subscribers);
+        return new HashSet<>(subscribers);
     }
 
     @Override
-    public final synchronized void close() throws Exception {
-        if (this.registration != null) {
-            this.registration.close();
-            this.registration = null;
+    public final synchronized void close() throws InterruptedException, ExecutionException {
+        if (registration != null) {
+            registration.close();
+            registration = null;
         }
-        deleteDataInDS();
-        this.subscribers.clear();
+        deleteDataInDS().get();
+        subscribers.clear();
     }
 
     @Override
     public synchronized void addSubscriber(final StreamSessionHandler subscriber) {
         final boolean isConnected = subscriber.isConnected();
-        Preconditions.checkState(isConnected);
+        checkState(isConnected);
         LOG.debug("Subscriber {} is added.", subscriber);
         subscribers.add(subscriber);
     }
@@ -57,7 +121,7 @@ abstract class AbstractCommonSubscriber extends AbstractQueryParams implements B
     @Override
     public synchronized void removeSubscriber(final StreamSessionHandler subscriber) {
         final boolean isConnected = subscriber.isConnected();
-        Preconditions.checkState(isConnected);
+        checkState(isConnected);
         LOG.debug("Subscriber {} is removed", subscriber);
         subscribers.remove(subscriber);
         if (!hasSubscribers()) {
@@ -65,14 +129,85 @@ abstract class AbstractCommonSubscriber extends AbstractQueryParams implements B
         }
     }
 
-    @Override
-    public void setRegistration(final ListenerRegistration<?> registration) {
-        this.registration = registration;
+    public final Instant getStart() {
+        return start;
     }
 
-    @Override
-    public boolean isListening() {
-        return this.registration != null;
+    /**
+     * Set query parameters for listener.
+     *
+     * @param params NotificationQueryParams to use.
+     */
+    public final void setQueryParams(final NotificationQueryParams params) {
+        final var startTime = params.startTime();
+        start = startTime == null ? Instant.now() : parseDateAndTime(startTime.value());
+
+        final var stopTime = params.stopTime();
+        stop = stopTime == null ? null : parseDateAndTime(stopTime.value());
+
+        final var leafNodes = params.leafNodesOnly();
+        leafNodesOnly = leafNodes == null ? false : leafNodes.value();
+
+        final var skipData = params.skipNotificationData();
+        skipNotificationData = skipData == null ? false : skipData.value();
+
+        final var filter = params.filter();
+        final String filterValue = filter == null ? null : filter.paramValue();
+        if (filterValue != null && !filterValue.isEmpty()) {
+            try {
+                formatter = formatterFactory.getFormatter(filterValue);
+            } catch (XPathExpressionException e) {
+                throw new IllegalArgumentException("Failed to get filter", e);
+            }
+        } else {
+            formatter = formatterFactory.getFormatter();
+        }
+    }
+
+    final P path() {
+        return path;
+    }
+
+    /**
+     * Check whether this query should only notify about leaf node changes.
+     *
+     * @return true if this query should only notify about leaf node changes
+     */
+    final boolean getLeafNodesOnly() {
+        return leafNodesOnly;
+    }
+
+    /**
+     * Check whether this query should notify changes without data.
+     *
+     * @return true if this query should notify about changes with  data
+     */
+    final boolean isSkipNotificationData() {
+        return skipNotificationData;
+    }
+
+    final EventFormatter<T> formatter() {
+        return formatter;
+    }
+
+    /**
+     * Sets {@link Registration} registration.
+     *
+     * @param registration a listener registration registration.
+     */
+    @Holding("this")
+    final void setRegistration(final Registration registration) {
+        this.registration = requireNonNull(registration);
+    }
+
+    /**
+     * Checks if {@link Registration} registration exists.
+     *
+     * @return {@code true} if exists, {@code false} otherwise.
+     */
+    @Holding("this")
+    final boolean isListening() {
+        return registration != null;
     }
 
     /**
@@ -96,4 +231,54 @@ abstract class AbstractCommonSubscriber extends AbstractQueryParams implements B
             }
         }
     }
+
+    @SuppressWarnings("checkstyle:IllegalCatch")
+    final boolean checkStartStop(final Instant now) {
+        if (stop != null) {
+            if (start.compareTo(now) < 0 && stop.compareTo(now) > 0) {
+                return true;
+            }
+            if (stop.compareTo(now) < 0) {
+                try {
+                    close();
+                } catch (final Exception e) {
+                    throw new RestconfDocumentedException("Problem with unregister listener." + e);
+                }
+            }
+        } else if (start != null) {
+            if (start.compareTo(now) < 0) {
+                start = null;
+                return true;
+            }
+        } else {
+            return true;
+        }
+        return false;
+    }
+
+    @Override
+    public final String toString() {
+        return MoreObjects.toStringHelper(this)
+            .add("path", path)
+            .add("stream-name", streamName)
+            .add("output-type", getOutputType())
+            .toString();
+    }
+
+    /**
+     * Parse input of query parameters - start-time or stop-time - from {@link DateAndTime} format
+     * to {@link Instant} format.
+     *
+     * @param uriValue Start-time or stop-time as string in {@link DateAndTime} format.
+     * @return Parsed {@link Instant} by entry.
+     */
+    private static @NonNull Instant parseDateAndTime(final DateAndTime dateAndTime) {
+        final TemporalAccessor accessor;
+        try {
+            accessor = FORMATTER.parse(dateAndTime.getValue());
+        } catch (final DateTimeParseException e) {
+            throw new RestconfDocumentedException("Cannot parse of value in date: " + dateAndTime, e);
+        }
+        return Instant.from(accessor);
+    }
 }