com.uber.cadence.internal.sync.SignalWithStartBatchRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cadence-client Show documentation
Uber Cadence Java Client
There is a newer version: 3.12.5
/*
 *  Copyright 2012-2016 Amazon.com, Inc. or its affiliates. All Rights Reserved.
 *
 *  Modifications copyright (C) 2017 Uber Technologies, Inc.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License"). You may not
 *  use this file except in compliance with the License. A copy of the License is
 *  located at
 *
 *  http://aws.amazon.com/apache2.0
 *
 *  or in the "license" file accompanying this file. This file is distributed on
 *  an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 *  express or implied. See the License for the specific language governing
 *  permissions and limitations under the License.
 */

package com.uber.cadence.internal.sync;

import com.uber.cadence.WorkflowExecution;
import com.uber.cadence.client.BatchRequest;
import com.uber.cadence.client.WorkflowStub;
import com.uber.cadence.workflow.Functions;
import java.util.ArrayList;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;

final class SignalWithStartBatchRequest implements BatchRequest {

  private final List requests = new ArrayList<>();

  private WorkflowStub stub;
  private String signalName;
  private Object[] signalArgs;
  private Object[] startArgs;
  private AtomicBoolean invoked = new AtomicBoolean();

  WorkflowExecution invoke() {
    if (!invoked.compareAndSet(false, true)) {
      throw new IllegalStateException(
          "A batch instance can be used only for a single signalWithStart call");
    }
    WorkflowInvocationHandler.initAsyncInvocation(
        WorkflowInvocationHandler.InvocationType.SIGNAL_WITH_START, this);
    try {
      for (Functions.Proc request : requests) {
        request.apply();
      }
      return signalWithStart();
    } finally {
      WorkflowInvocationHandler.closeAsyncInvocation();
    }
  }

  private WorkflowExecution signalWithStart() {
    return stub.signalWithStart(signalName, signalArgs, startArgs);
  }

  void signal(WorkflowStub stub, String signalName, Object[] args) {
    setStub(stub);
    this.signalName = signalName;
    this.signalArgs = args;
  }

  void start(WorkflowStub stub, Object[] args) {
    setStub(stub);
    this.startArgs = args;
  }

  private void setStub(WorkflowStub stub) {
    if (this.stub != null && stub != this.stub) {
      throw new IllegalArgumentException(
          "SignalWithStart Batch invoked on different workflow stubs");
    }
    this.stub = stub;
  }

  @Override
  public void add(Functions.Proc request) {
    requests.add(request);
  }

  @Override
  public  void add(Functions.Proc1 request, A1 arg1) {
    add(() -> request.apply(arg1));
  }

  @Override
  public  void add(Functions.Proc2 request, A1 arg1, A2 arg2) {
    add(() -> request.apply(arg1, arg2));
  }

  @Override
  public  void add(Functions.Proc3 request, A1 arg1, A2 arg2, A3 arg3) {
    add(() -> request.apply(arg1, arg2, arg3));
  }

  @Override
  public  void add(
      Functions.Proc4 request, A1 arg1, A2 arg2, A3 arg3, A4 arg4) {
    add(() -> request.apply(arg1, arg2, arg3, arg4));
  }

  @Override
  public  void add(
      Functions.Proc5 request, A1 arg1, A2 arg2, A3 arg3, A4 arg4, A5 arg5) {
    add(() -> request.apply(arg1, arg2, arg3, arg4, arg5));
  }

  @Override
  public  void add(
      Functions.Proc6 request,
      A1 arg1,
      A2 arg2,
      A3 arg3,
      A4 arg4,
      A5 arg5,
      A6 arg6) {
    add(() -> request.apply(arg1, arg2, arg3, arg4, arg5, arg6));
  }

  @Override
  public void add(Functions.Func request) {
    add(
        () -> {
          request.apply();
        });
  }

  @Override
  public  void add(Functions.Func1 request, A1 arg1) {
    add(() -> request.apply(arg1));
  }

  @Override
  public  void add(Functions.Func2 request, A1 arg1, A2 arg2) {
    add(() -> request.apply(arg1, arg2));
  }

  @Override
  public  void add(Functions.Func3 request, A1 arg1, A2 arg2, A3 arg3) {
    add(() -> request.apply(arg1, arg2, arg3));
  }

  @Override
  public  void add(
      Functions.Func4 request, A1 arg1, A2 arg2, A3 arg3, A4 arg4) {
    add(() -> request.apply(arg1, arg2, arg3, arg4));
  }

  @Override
  public  void add(
      Functions.Func5 request, A1 arg1, A2 arg2, A3 arg3, A4 arg4, A5 arg5) {
    add(() -> request.apply(arg1, arg2, arg3, arg4, arg5));
  }

  @Override
  public  void add(
      Functions.Func6 request,
      A1 arg1,
      A2 arg2,
      A3 arg3,
      A4 arg4,
      A5 arg5,
      A6 arg6) {
    add(() -> request.apply(arg1, arg2, arg3, arg4, arg5, arg6));
  }
}