science.aist.xes.model.impl.LogRepository Maven / Gradle / Ivy

Go to download
package science.aist.xes.model.impl;

import lombok.SneakyThrows;
import science.aist.xes.model.LogType;
import science.aist.xes.model.XMLRepository;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBElement;
import javax.xml.bind.Marshaller;
import javax.xml.bind.Unmarshaller;
import javax.xml.transform.stream.StreamSource;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * Repository for {@link LogType}
 *
 * @author Andreas Pointner
 * @since 1.0
 */
public class LogRepository implements XMLRepository {
    private final JAXBContext jaxbContext;

    @SneakyThrows
    public LogRepository() {
        jaxbContext = JAXBContext.newInstance(LogType.class);
    }

    @SneakyThrows
    @Override
    public void save(JAXBElement jaxbElement, OutputStream os) {
        Marshaller jaxbMarshaller = jaxbContext.createMarshaller();
        jaxbMarshaller.setProperty(Marshaller.JAXB_FORMATTED_OUTPUT, true);
        jaxbMarshaller.marshal(jaxbElement, os);
    }

    @SneakyThrows
    @Override
    public JAXBElement load(InputStream is) {
        final Unmarshaller unmarshaller = jaxbContext.createUnmarshaller();
        return unmarshaller.unmarshal(new StreamSource(is), LogType.class);
    }
}