net.bpelunit.model.bpel._2_0.Process Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of model-bpel Show documentation
The newest version!
package net.bpelunit.model.bpel._2_0;

import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Iterator;
import java.util.List;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;

import net.bpelunit.model.bpel.IActivity;
import net.bpelunit.model.bpel.IBpelObject;
import net.bpelunit.model.bpel.IImport;
import net.bpelunit.model.bpel.IPartnerLink;
import net.bpelunit.model.bpel.IProcess;
import net.bpelunit.model.bpel.IVariable;
import net.bpelunit.model.bpel.IVisitor;

import org.apache.commons.io.IOUtils;
import org.apache.commons.jxpath.JXPathContext;
import org.oasis_open.docs.wsbpel._2_0.process.executable.ObjectFactory;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TActivity;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TBoolean;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TImport;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TPartnerLink;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TPartnerLinks;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TProcess;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TVariable;
import org.oasis_open.docs.wsbpel._2_0.process.executable.TVariables;

class Process extends AbstractBpelObject implements IProcess {

	private TProcess process;
	private AbstractActivity mainActivity;
	private List partnerLinks = new ArrayList();
	private List imports = new ArrayList();
	private List variables = new ArrayList();
	private BpelFactory factory;

	Process(TProcess newProcess, BpelFactory f) {
		super(newProcess, null);
		this.factory = f;
		
		TActivity childActivity = TComplexContainerHelper
				.getChildActivity(newProcess);
		
		this.process = newProcess;
		this.setMainActivity(f.createActivity(childActivity));

		if (newProcess.getPartnerLinks() != null) {
			for (TPartnerLink p : newProcess.getPartnerLinks().getPartnerLink()) {
				partnerLinks.add(getFactory().createPartnerLink(p));
			}
		}

		for (TImport i : newProcess.getImport()) {
			imports.add(getFactory().createImport(i));
		}

		if (newProcess.getVariables() == null) {
			newProcess.setVariables(new TVariables());
		}
		for (TVariable v : newProcess.getVariables().getVariable()) {
			variables.add(getFactory().createVariable(v));
		}
	}

	@Override
	public BpelFactory getFactory() {
		return factory;
	}
	
	@Override
	public String getName() {
		return process.getName();
	}

	@Override
	public String getTargetNamespace() {
		return process.getTargetNamespace();
	}

	@Override
	public String getQueryLanguage() {
		return process.getQueryLanguage();
	}

	@Override
	public boolean getSuppressJoinFailure() {
		return process.getSuppressJoinFailure().equals(TBoolean.YES);
	}

	@Override
	public void setName(String value) {
		process.setName(value);
	}

	@Override
	public void setTargetNamespace(String value) {
		process.setTargetNamespace(value);
	}

	@Override
	public void setMainActivity(IActivity a) {
		checkForCorrectModel(a);
		AbstractActivity activity = ((AbstractActivity) a);
		TComplexContainerHelper.removeMainActivity(process);
		if (activity != null) {
			TComplexContainerHelper.setActivity(process,
					activity.getNativeActivity());
		} else {
			TComplexContainerHelper.setActivity(process, null);
		}
		this.mainActivity = activity;
	}

	@Override
	public AbstractActivity getMainActivity() {
		return this.mainActivity;
	}

	@Override
	public List getPartnerLinks() {
		return Collections.unmodifiableList(this.partnerLinks);
	}

	@Override
	public List getImports() {
		return Collections.unmodifiableList(this.imports);
	}

	@Override
	public void visit(IVisitor v) {
		v.visit(this);
		getMainActivity().visit(v);
	}

	@Override
	public List getVariables() {
		return Collections.unmodifiableList(this.variables);
	}

	@Override
	public IVariable addVariable() {
		TVariable nativeVariable = new TVariable();
		Variable variable = getFactory().createVariable(nativeVariable);

		this.process.getVariables().getVariable().add(nativeVariable);
		this.variables.add(variable);
		return variable;
	}

	@Override
	public IPartnerLink addPartnerLink() {
		TPartnerLink nativePartnerLink = new TPartnerLink();
		PartnerLink partnerLink = getFactory().createPartnerLink(nativePartnerLink);
		
		guaranteePartnerLinks(this.process);
		this.process.getPartnerLinks().getPartnerLink().add(nativePartnerLink);
		this.partnerLinks.add(partnerLink);
		
		return partnerLink;
	}
	
	private void guaranteePartnerLinks(TProcess p) {
		if(p.getPartnerLinks() == null) {
			p.setPartnerLinks(new TPartnerLinks());
		}
	}

	@Override
	public void save(OutputStream out, Class... additionalClasses) throws IOException {
		try {
			JAXBContext ctx = getFactory().getJAXBContext();
			JAXBElement rootElement = new ObjectFactory().createProcess(this.process);
			Marshaller marshaller = ctx.createMarshaller();
			marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT,
					Boolean.TRUE);
			marshaller.marshal(rootElement, out);
		} catch (JAXBException e) {
			throw new IOException("Could not marshall BPEL process: "
					+ e.getMessage(), e);
		} finally {
			IOUtils.closeQuietly(out);
		}
	}
	
	@Override
	public IImport addImport() {
		TImport nativeImport = new TImport();
		Import imp = getFactory().createImport(nativeImport);
		
		this.process.getImport().add(nativeImport);
		this.imports.add(imp);
		
		return imp;
	}

	@Override
	public List getElementsByXPath(String xpathToBpelElement) {
		JXPathContext context = JXPathContext.newContext(process);
		
		List retval = new ArrayList();
		for(Iterator iter = context.iterate(xpathToBpelElement); iter.hasNext();){
			Object o = iter.next();
			IBpelObject bo = getObjectForNativeObject(o);
			if(bo != null) {
				retval.add(bo);
			} else {
				throw new RuntimeException("No object mapping for " + o + " evaluated from " + xpathToBpelElement);
			}
		}
		
		return retval;
	}

	@Override
	IBpelObject getObjectForNativeObject(Object nativeObject) {
		if(mainActivity != null) {
			return mainActivity.getObjectForNativeObject(nativeObject);
		} else {
			return null;
		}
	}
}