+/*
+ * Copyright (c) 2017 Cisco Systems, Inc. and others. All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+
+package org.opendaylight.controller.clustering.it.provider.impl;
+
+import com.google.common.base.Preconditions;
+import com.google.common.util.concurrent.SettableFuture;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.ScheduledFuture;
+import java.util.concurrent.TimeUnit;
+import org.opendaylight.controller.md.sal.binding.api.NotificationPublishService;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.target.rev170215.IdSequence;
+import org.opendaylight.yang.gen.v1.tag.opendaylight.org._2017.controller.yang.lowlevel.target.rev170215.IdSequenceBuilder;
+import org.opendaylight.yangtools.yang.common.RpcError;
+import org.opendaylight.yangtools.yang.common.RpcResult;
+import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class PublishNotificationsTask implements Runnable {
+
+ private static final Logger LOG = LoggerFactory.getLogger(PublishNotificationsTask.class);
+ private static final int SECOND_AS_NANO = 1000000000;
+
+ private final NotificationPublishService notificationPublishService;
+ private final String notificationId;
+ private final long timeToTake;
+ private final long delay;
+
+ private final ScheduledExecutorService executor = Executors.newSingleThreadScheduledExecutor();
+
+ private long sequenceNumber = 1;
+ private long startTime;
+ private SettableFuture<RpcResult<Void>> completionFuture;
+ private ScheduledFuture<?> scheduledFuture;
+
+ public PublishNotificationsTask(final NotificationPublishService notificationPublishService,
+ final String notificationId, final long secondsToTake, final long maxPerSecond) {
+ Preconditions.checkNotNull(notificationPublishService);
+ Preconditions.checkNotNull(notificationId);
+ Preconditions.checkArgument(secondsToTake > 0);
+ Preconditions.checkArgument(maxPerSecond > 0);
+
+ this.notificationPublishService = notificationPublishService;
+ this.notificationId = notificationId;
+ this.timeToTake = secondsToTake * SECOND_AS_NANO;
+ this.delay = SECOND_AS_NANO / maxPerSecond;
+
+ LOG.debug("Delay : {}", delay);
+ }
+
+ @Override
+ public void run() {
+ final long current = System.nanoTime();
+
+ final IdSequence notification =
+ new IdSequenceBuilder().setId(notificationId).setSequenceNumber(sequenceNumber).build();
+ sequenceNumber++;
+
+ try {
+ LOG.debug("Publishing notification: {}", notification);
+ notificationPublishService.putNotification(notification);
+ } catch (final InterruptedException e) {
+ LOG.warn("Unexpected exception while publishing notification, : {}", notification, e);
+ completionFuture.set(RpcResultBuilder.<Void>failed()
+ .withError(RpcError.ErrorType.APPLICATION, "Unexpected-exception", e).build());
+ }
+
+ LOG.debug("current {}, starttime: {}, timetotake: {}, current-start = {}",
+ current, startTime, timeToTake, current - startTime);
+
+ if ((current - startTime) > timeToTake) {
+ completionFuture.set(RpcResultBuilder.<Void>success().build());
+ LOG.debug("Sequence number: {}", sequenceNumber);
+ scheduledFuture.cancel(false);
+ executor.shutdown();
+ }
+ }
+
+ public void start(final SettableFuture<RpcResult<Void>> settableFuture) {
+ startTime = System.nanoTime();
+ completionFuture = settableFuture;
+ scheduledFuture = executor.scheduleAtFixedRate(this, 0, delay, TimeUnit.NANOSECONDS);
+ }
+}