variablescommand.java

workflow first jbpm

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 implements Command {
  
  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;
  }
}