VariablesCommand.java

package org.jbpm.command;

import java.util.Map;

import org.jbpm.JbpmContext;
import org.jbpm.context.exe.VariableContainer;
import org.jbpm.graph.exe.ProcessInstance;
import org.jbpm.graph.exe.Token;
import org.jbpm.taskmgmt.exe.TaskInstance;

public class VariablesCommand extends AbstractBaseCommand {

  private static final long serialVersionUID = 1L;

  long tokenId = 0;
  long taskInstanceId = 0;
  Map variables = null;

  ProcessInstance previousProcessInstance = null;
  Token previousToken = null;
  TaskInstance previousTaskInstance = null;

  public Object execute(JbpmContext jbpmContext) throws Exception {
    VariableContainer variableContainer = getVariableContainer(jbpmContext);
    if ((variableContainer != null) && (variables != null)) {
      variableContainer.addVariables(variables);
    }
    return variableContainer;
  }

  protected VariableContainer getVariableContainer(JbpmContext jbpmContext) {
    if (previousProcessInstance != null) {
      return getVariableContainer(previousProcessInstance.getRootToken());
    }
    if (previousToken != null) {
      return getVariableContainer(previousToken);
    }
    if (previousTaskInstance != null) {
      return previousTaskInstance;
    }

    if (tokenId != 0) {
      return getVariableContainer(jbpmContext.getToken(tokenId));
    }
    if (taskInstanceId != 0) {
      return jbpmContext.getTaskInstance(taskInstanceId);
    }
    return null;
  }

  protected VariableContainer getVariableContainer(Token token) {
    return token.getProcessInstance().getContextInstance().getTokenVariableMap(token);
  }

  public long getTaskInstanceId() {
    return taskInstanceId;
  }

  public void setTaskInstanceId(long taskInstanceId) {
    this.taskInstanceId = taskInstanceId;
  }

  public long getTokenId() {
    return tokenId;
  }

  public void setTokenId(long tokenId) {
    this.tokenId = tokenId;
  }

  public Map getVariables() {
    return variables;
  }

  public void setVariables(Map variables) {
    this.variables = variables;
  }

  public String getAdditionalToStringInformation() {
    return "tokenId=" + tokenId + ";taskInstanceId=" + taskInstanceId + ";variables=" + variables;
  }

  // methods for fluent programming

  public VariablesCommand taskInstanceId(long taskInstanceId) {
    setTaskInstanceId(taskInstanceId);
    return this;
  }

  public VariablesCommand tokenId(long tokenId) {
    setTokenId(tokenId);
    return this;
  }

  public VariablesCommand variables(Map variables) {
    setVariables(variables);
    return this;
  }

}