com.github.dakusui.scriptiveunit.drivers.actions.Basic Maven / Gradle / Ivy

Go to download
package com.github.dakusui.scriptiveunit.drivers.actions;

import com.github.dakusui.actionunit.Action;
import com.github.dakusui.actionunit.Actions;
import com.github.dakusui.scriptiveunit.annotations.ReflectivelyReferenced;
import com.github.dakusui.scriptiveunit.annotations.Scriptable;
import com.github.dakusui.scriptiveunit.model.func.Func;
import com.github.dakusui.scriptiveunit.model.Stage;

import static java.util.Arrays.stream;
import static java.util.stream.Collectors.toList;

@ReflectivelyReferenced
public class Basic {
  @ReflectivelyReferenced
  @Scriptable
  public Func nop() {
    return input -> Actions.nop();
  }

  @SafeVarargs
  @ReflectivelyReferenced
  @Scriptable
  public final Func sequential(Func... actions) {
    return (Stage input) -> Actions.sequential(
        stream(actions)
            .map(each -> each.apply(input))
            .collect(toList())
            .toArray(new Action[actions.length]));
  }

  @SafeVarargs
  @ReflectivelyReferenced
  @Scriptable
  public final Func concurrent(Func... actions) {
    return (Stage input) -> Actions.concurrent(
        stream(actions)
            .map(each -> each.apply(input))
            .collect(toList())
            .toArray(new Action[actions.length]));
  }

  @ReflectivelyReferenced
  @Scriptable
  public Func fail(Func in) {
    return input -> Actions.simple(() -> {
      throw new RuntimeException(in.apply(input));
    });
  }

  @ReflectivelyReferenced
  @Scriptable
  public Func print(Func in) {
    return input -> Actions.simple(
        new Runnable() {
          @Override
          public void run() {
            System.out.println(in.apply(input));
          }

          @Override
          public String toString() {
            return "print";
          }
        }
    );
  }
}