io.hyperfoil.core.steps.AwaitVarStep 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.steps;

import java.util.Collections;
import java.util.List;

import org.kohsuke.MetaInfServices;

import io.hyperfoil.api.config.InitFromParam;
import io.hyperfoil.api.config.Name;
import io.hyperfoil.api.config.Step;
import io.hyperfoil.api.config.StepBuilder;
import io.hyperfoil.api.session.Session;
import io.hyperfoil.api.session.Access;
import io.hyperfoil.core.session.SessionFactory;

public class AwaitVarStep extends DependencyStep {
   public AwaitVarStep(String var) {
      super(new Access[]{ access(var) });
   }

   private static Access access(String var) {
      if (var.startsWith("!")) {
         return new NegatedAccess(SessionFactory.access(var.substring(1).trim()));
      }
      return SessionFactory.access(var);
   }

   public static class NegatedAccess implements Access {
      private final Access access;

      public NegatedAccess(Access access) {
         this.access = access;
      }

      @Override
      public void declareObject(Session session) {
         access.declareObject(session);
      }

      @Override
      public void declareInt(Session session) {
         access.declareInt(session);
      }

      @Override
      public boolean isSet(Session session) {
         return !access.isSet(session);
      }

      @Override
      public Object getObject(Session session) {
         throw new UnsupportedOperationException();
      }

      @Override
      public void setObject(Session session, Object value) {
         throw new UnsupportedOperationException();
      }

      @Override
      public int getInt(Session session) {
         throw new UnsupportedOperationException();
      }

      @Override
      public void setInt(Session session, int value) {
         throw new UnsupportedOperationException();
      }

      @Override
      public Session.Var getVar(Session Session) {
         throw new UnsupportedOperationException();
      }

      @Override
      public int addToInt(Session session, int delta) {
         throw new UnsupportedOperationException();
      }

      @Override
      public Object activate(Session session) {
         throw new UnsupportedOperationException();
      }

      @Override
      public void unset(Session session) {
         throw new UnsupportedOperationException();
      }
   }

   /**
    * Block current sequence until this variable gets set/unset.
    */
   @MetaInfServices
   @Name("awaitVar")
   public static class Builder implements StepBuilder, InitFromParam {
      private String var;

      /**
       * @param param Variable name or !variable if we are waiting for it to be unset.
       * @return Self.
       */
      @Override
      public Builder init(String param) {
         return var(param);
      }

      /**
       * Variable name or !variable if we are waiting for it to be unset.
       *
       * @param var Name of the variable we're waiting for.
       * @return Self.
       */
      public Builder var(String var) {
         this.var = var;
         return this;
      }

      @Override
      public List build() {
         return Collections.singletonList(new AwaitVarStep(var));
      }
   }
}