io.hyperfoil.core.util.CountDown Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of hyperfoil-core Show documentation

The core of Hyperfoil

There is a newer version: 0.27.1

package io.hyperfoil.core.util;

import io.vertx.core.AsyncResult;
import io.vertx.core.Future;
import io.vertx.core.Handler;

public class CountDown implements Handler> {
   private Handler> handler;
   private int value;

   public CountDown(Handler> handler, int initialValue) {
      if (initialValue <= 0) {
         throw new IllegalArgumentException();
      }
      this.handler = handler;
      this.value = initialValue;
   }

   public CountDown(int initialValue) {
      this(null, initialValue);
   }

   public CountDown setHandler(Handler> handler) {
      if (this.handler != null) {
         throw new IllegalStateException();
      } else if (handler == null) {
         throw new IllegalArgumentException();
      }
      this.handler = handler;
      return this;
   }

   public void increment() {
      if (value < 0) {
         throw new IllegalStateException();
      }
      ++value;
   }

   public void countDown() {
      if (value <= 0) {
         throw new IllegalStateException();
      }
      if (--value == 0) {
         value = -1;
         handler.handle(Future.succeededFuture());
      }
   }

   @Override
   public void handle(AsyncResult event) {
      if (event.succeeded()) {
         countDown();
      } else {
         value = -1;
         handler.handle(event);
      }
   }
}