org.infinispan.hotrod.event.impl.ReconnectTask Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of infinispan-hotrod Show documentation

Infinispan Hot Rod Client

The newest version!

package org.infinispan.hotrod.event.impl;

import java.util.concurrent.ScheduledFuture;
import java.util.function.Consumer;

import org.infinispan.hotrod.impl.logging.Log;
import org.infinispan.hotrod.impl.logging.LogFactory;
import org.infinispan.commons.util.Util;

public class ReconnectTask implements Runnable, Consumer {
   private static final Log log = LogFactory.getLog(ReconnectTask.class);

   private final EventDispatcher dispatcher;
   private ScheduledFuture cancellationFuture;
   private boolean completed = false;

   public ReconnectTask(EventDispatcher dispatcher) {
      this.dispatcher = dispatcher;
   }

   @Override
   public void run() {
      if (log.isTraceEnabled()) {
         log.tracef("Reconnecting client listener with id %s", Util.printArray(dispatcher.listenerId));
      }
      dispatcher.executeFailover().thenAccept(this);
   }

   public synchronized void setCancellationFuture(ScheduledFuture cancellationFuture) {
      this.cancellationFuture = cancellationFuture;
      if (completed) {
         cancellationFuture.cancel(false);
      }
   }

   @Override
   public synchronized void accept(Void address) {
      ScheduledFuture cancellationFuture = this.cancellationFuture;
      completed = true;
      if (cancellationFuture != null) {
         cancellationFuture.cancel(false);
      }
   }
}