ChorXmlWriter.java

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

package org.ow2.choreos.chors.datamodel.xml;

import java.io.StringWriter;
import java.io.Writer;

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

import org.apache.log4j.Logger;
import org.ow2.choreos.chors.datamodel.Choreography;
import org.ow2.choreos.chors.datamodel.ChoreographySpec;

public class ChorXmlWriter {

    private Logger logger = Logger.getLogger(ChorXmlWriter.class);

    public String getChorSpecXML(ChoreographySpec chorSpec) throws JAXBException {

        return getXML(chorSpec, ChoreographySpec.class);
    }

    public String getChoreographyXML(Choreography chor) throws JAXBException {

        return getXML(chor, Choreography.class);
    }

    private String getXML(Object obj, Class<?> clazz) throws JAXBException {

        Marshaller marshaller;
        try {
            marshaller = getMarshaller(clazz);
        } catch (JAXBException e) {
            logger.error("It should never happen");
            return null;
        }
        Writer writer = new StringWriter();
        marshaller.marshal(obj, writer);
        return writer.toString();
    }

    private Marshaller getMarshaller(Class<?> clazz) throws JAXBException {

        JAXBContext context = JAXBContext.newInstance(clazz);
        Marshaller marshaller = context.createMarshaller();
        marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, Boolean.TRUE);
        return marshaller;
    }

}