io.github.factoryfx.soap.JAXBSoapUtil Maven / Gradle / Ivy

Go to download
package io.github.factoryfx.soap;

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 JAXBSoapUtil {

    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;
    }

    public static JAXBContext getJAXBContextForWebService(Class webService) {
        if (!hasWebServiceAnnotation(webService)){
            throw new IllegalArgumentException("no webservice class");
        }
        Set> 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 void collect(Class clazz, Set> classes) {
        for (Method method : clazz.getDeclaredMethods()) {
            if (method.getReturnType() != null) {
                if (classes.add(method.getReturnType())) {
                    collect(method.getReturnType(),classes);
                }
                for (int i = 0; i < method.getParameterCount(); ++i) {
                    Class parameterType = method.getParameterTypes()[0];
                    if (classes.add(parameterType)) {
                        collect(parameterType, classes);
                    }
                }


                collectExceptions(classes, method);

            }
        }
    }

    private static void collectExceptions(Set> 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();
        }
    }
}