Java XML JAXB Context getJAXBContextForWebService(Class webService)

Description

get JAXB Context For Web Service

License

Apache License

Declaration

public static JAXBContext getJAXBContextForWebService(Class<?> webService) 

Method Source Code

//package com.java2s;
//License from project: Apache License 

import javax.jws.WebService;
import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.annotation.XmlRootElement;
import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.Set;

public class Main {
    public static JAXBContext getJAXBContextForWebService(Class<?> webService) {
        if (!hasWebServiceAnnotation(webService)) {
            throw new IllegalArgumentException("no webservice class");
        }/*from w  ww. j a v a 2s  . co  m*/
        Set<Class<?>> classes = new HashSet<>();
        collect(webService, classes);
        try {
            Class[] classesToBeBound = classes.stream()
                    .filter(clazz -> clazz.getDeclaredAnnotation(XmlRootElement.class) != null)
                    .toArray(Class[]::new);
            return JAXBContext.newInstance(classesToBeBound);
        } catch (JAXBException e) {
            throw new RuntimeException(e);
        }
    }

    private static boolean hasWebServiceAnnotation(Class<?> clazz) {
        while (clazz != null) {
            if (clazz.getDeclaredAnnotation(WebService.class) != null) {
                return true;
            }
            for (Class<?> anInterface : clazz.getInterfaces()) {
                if (anInterface.getDeclaredAnnotation(WebService.class) != null) {
                    return true;
                }
            }
            clazz = clazz.getSuperclass();
        }
        return false;
    }

    private static void collect(Class<?> clazz, Set<Class<?>> classes) {
        for (Method method : clazz.getDeclaredMethods()) {
            if (method.getParameterCount() <= 1 && method.getReturnType() != null) {
                if (classes.add(method.getReturnType())) {
                    collect(method.getReturnType(), classes);
                }
                if (method.getParameterCount() > 0) {
                    Class<?> parameterType = method.getParameterTypes()[0];
                    if (classes.add(parameterType)) {
                        collect(parameterType, classes);
                    }
                }

                collectExceptions(classes, method);

            }
        }
    }

    private static void collectExceptions(Set<Class<?>> classes, Method method) {
        for (Class<?> exceptionClass : method.getExceptionTypes()) {
            if (classes.add(exceptionClass)) {
                collect(exceptionClass, classes);
            }
        }
        Class<?> thisClass = method.getDeclaringClass();
        for (Class<?> i : thisClass.getInterfaces()) {
            try {
                Method parentMethod = i.getDeclaredMethod(method.getName(), method.getParameterTypes());
                collectExceptions(classes, parentMethod);
            } catch (NoSuchMethodException ignored) {
            }
        }
        Class<?> parent = thisClass.getSuperclass();
        while (parent != null) {
            try {
                Method parentMethod = parent.getDeclaredMethod(method.getName(), method.getParameterTypes());
                collectExceptions(classes, parentMethod);
                return;
            } catch (NoSuchMethodException ignored) {
            }
            parent = parent.getSuperclass();
        }
    }
}

Related

1. getJaxbContext(Class valueType)
2. getJaxbContext(String contextPath)
3. getJAXBContext(String contextPath)
4. getJAXBContext(String namespaces)
5. getJAXBContext(String packagePrefix)
6. getRequestContext()
7. getRightsContext()
8. isElement(String contextPath, Object object)
9. parseXMLFile(String fileName, String contextPath)