fish.focus.uvms.exchange.model.mapper.JAXBMarshaller Maven / Gradle / Ivy

Go to download
/*
Developed with the contribution of the European Commission - Directorate General for Maritime Affairs and Fisheries
© European Union, 2015-2016.

This file is part of the Integrated Fisheries Data Management (IFDM) Suite. The IFDM Suite is free software: you can
redistribute it and/or modify it under the terms of the GNU General Public License as published by the
Free Software Foundation, either version 3 of the License, or any later version. The IFDM Suite is distributed in
the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more details. You should have received a
copy of the GNU General Public License along with the IFDM Suite. If not, see .
 */
package fish.focus.uvms.exchange.model.mapper;

import org.apache.commons.lang3.StringUtils;

import javax.jms.JMSException;
import javax.jms.TextMessage;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.transform.stream.StreamSource;
import java.io.StringReader;
import java.io.StringWriter;
import java.util.HashMap;
import java.util.Map;

public class JAXBMarshaller {

    private static final String COM_SUN_XML_BIND_XML_DECLARATION = "com.sun.xml.bind.xmlDeclaration";

    private static Map contexts = new HashMap<>();

    /**
     * Marshalls a JAXB Object to a XML String representation
     *
     * @param 
     * @param data
     * @return
     * @throws
     */
    public static  String marshallJaxBObjectToString(T data) {
        try {
            JAXBContext jaxbContext = contexts.get(data.getClass().getName());
            if (jaxbContext == null) {
                jaxbContext = JAXBContext.newInstance(data.getClass());
                contexts.put(data.getClass().getName(), jaxbContext);
            }
            Marshaller marshaller = jaxbContext.createMarshaller();
            marshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
            marshaller.setProperty(Marshaller.JAXB_FRAGMENT, Boolean.TRUE);
            StringWriter sw = new StringWriter();
            marshaller.marshal(data, sw);
            String marshalled = sw.toString();
            return marshalled;
        } catch (JAXBException ex) {
            throw new RuntimeException("[ Error when marshalling Object to String ]", ex);
        }
    }

    /**
     * Unmarshalls A textMessage to the desired Object. The object must be the
     * root object of the unmarchalled message!
     *
     * @param 
     * @param textMessage
     * @param clazz pperException
     * @return
     * @throws
     */
    public static  R unmarshallTextMessage(TextMessage textMessage, Class clazz) {
        try {
            JAXBContext jc = contexts.get(clazz.getName());
            if (jc == null) {
                jc = JAXBContext.newInstance(clazz);
                contexts.put(clazz.getName(), jc);
            }
            Unmarshaller unmarshaller = jc.createUnmarshaller();
            StringReader sr = new StringReader(textMessage.getText());
            StreamSource source = new StreamSource(sr);
            R object = (R) unmarshaller.unmarshal(source);
            return object;
        } catch (JMSException | JAXBException ex) {
            throw new RuntimeException("[Error when unmarshalling response in ResponseMapper ]", ex);
        }
    }


    /**
     * Unmarshalls a string to the desired class.
     *
     * @param text
     * @param clazz class to marshall to. The class must be the
     * root object of the unmarchalled message!
     * @return
     */
    public static  R unmarshallString(String text, Class clazz) {
        try {
            JAXBContext jc = contexts.get(clazz.getName());
            if (jc == null) {
                jc = JAXBContext.newInstance(clazz);
                contexts.put(clazz.getName(), jc);
            }
            Unmarshaller unmarshaller = jc.createUnmarshaller();
            StringReader sr = new StringReader(text);
            StreamSource source = new StreamSource(sr);
            R object = (R) unmarshaller.unmarshal(source);
            return object;
        } catch (JAXBException ex) {
            throw new IllegalArgumentException("Error when unmarshalling text", ex);
        }
    }

    public static  String marshallJaxBObjectToString(T data, String encoding, boolean formatted) throws JAXBException {
        JAXBContext jaxbContext = contexts.get(data.getClass().getName());
        if (jaxbContext == null) {
            jaxbContext = JAXBContext.newInstance(data.getClass());
            contexts.put(data.getClass().getName(), jaxbContext);
        }

        Marshaller marshaller = jaxbContext.createMarshaller();
        if (StringUtils.isNotEmpty(encoding)) {
            marshaller.setProperty("jaxb.encoding", encoding);
        }

        if (formatted) {
            marshaller.setProperty("jaxb.formatted.output", true);
        }

        marshaller.setProperty("jaxb.fragment", Boolean.TRUE);
        StringWriter sw = new StringWriter();
        marshaller.marshal(data, sw);
        return sw.toString();
    }
}