io.quarkus.jaxb.runtime.JaxbContextProducer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quarkus-jaxb Show documentation
There is a newer version: 3.17.0.CR1
package io.quarkus.jaxb.runtime;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import jakarta.enterprise.context.ApplicationScoped;
import jakarta.enterprise.context.RequestScoped;
import jakarta.enterprise.inject.Instance;
import jakarta.enterprise.inject.Produces;
import jakarta.inject.Singleton;
import jakarta.xml.bind.JAXBContext;
import jakarta.xml.bind.JAXBException;
import jakarta.xml.bind.Marshaller;
import jakarta.xml.bind.Unmarshaller;

import io.quarkus.arc.DefaultBean;

@ApplicationScoped
public class JaxbContextProducer {

    @DefaultBean
    @Singleton
    @Produces
    public JAXBContext jaxbContext(Instance customizers) {
        return createJAXBContext(customizers);
    }

    @DefaultBean
    @RequestScoped
    @Produces
    public Marshaller marshaller(JAXBContext jaxbContext, Instance customizers) {
        try {
            Marshaller marshaller = jaxbContext.createMarshaller();
            List sortedCustomizers = sortCustomizersInDescendingPriorityOrder(customizers);
            for (JaxbContextCustomizer customizer : sortedCustomizers) {
                customizer.customizeMarshaller(marshaller);
            }

            return marshaller;
        } catch (JAXBException e) {
            throw new RuntimeException(e);
        }
    }

    @DefaultBean
    @RequestScoped
    @Produces
    public Unmarshaller unmarshaller(JAXBContext jaxbContext, Instance customizers) {
        try {
            Unmarshaller unmarshaller = jaxbContext.createUnmarshaller();
            List sortedCustomizers = sortCustomizersInDescendingPriorityOrder(customizers);
            for (JaxbContextCustomizer customizer : sortedCustomizers) {
                customizer.customizeUnmarshaller(unmarshaller);
            }

            return unmarshaller;
        } catch (JAXBException e) {
            throw new RuntimeException(e);
        }
    }

    public JAXBContext createJAXBContext(Instance customizers, Class... extraClasses) {
        try {
            Map properties = new HashMap<>();
            List sortedCustomizers = sortCustomizersInDescendingPriorityOrder(customizers);
            for (JaxbContextCustomizer customizer : sortedCustomizers) {
                customizer.customizeContextProperties(properties);
            }

            Set classes = new HashSet<>();
            classes.addAll(Arrays.asList(extraClasses));
            classes.addAll(JaxbContextConfigRecorder.getClassesToBeBound());

            return JAXBContext.newInstance(classes.toArray(new Class[0]), properties);
        } catch (JAXBException e) {
            throw new RuntimeException(e);
        }
    }

    private List sortCustomizersInDescendingPriorityOrder(Instance customizers) {
        List sortedCustomizers = new ArrayList<>();
        for (JaxbContextCustomizer customizer : customizers) {
            sortedCustomizers.add(customizer);
        }
        Collections.sort(sortedCustomizers);
        return sortedCustomizers;
    }
}