lodsve.web.webservice.WebServiceBeanFactoryPostProcessor Maven / Gradle / Ivy

Go to download
/*
 * Copyright (C) 2018  Sun.Hao
 *
 * This program 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
 * (at your option) any later version.
 *
 * This program 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 this program.  If not, see .
 */

package lodsve.web.webservice;

import lodsve.core.utils.StringUtils;
import org.apache.cxf.configuration.spring.AbstractBeanDefinitionParser;
import org.apache.cxf.configuration.spring.BusWiringType;
import org.apache.cxf.jaxws.spring.EndpointDefinitionParser;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.config.BeanFactoryPostProcessor;
import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.util.ClassUtils;

import javax.jws.WebService;

/**
 * WebService Bean FactoryPostProcessor.
 *
 * @author sunhao([email protected])
 * @date 2018-1-15-0015 09:37
 */
public class WebServiceBeanFactoryPostProcessor implements BeanFactoryPostProcessor {
    private static final Logger logger = LoggerFactory.getLogger(WebServiceBeanFactoryPostProcessor.class);
    private static final String WIRE_BUS_ATTRIBUTE = AbstractBeanDefinitionParser.class.getName() + ".wireBus";

    @Override
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {
        if (!(beanFactory instanceof DefaultListableBeanFactory)) {
            return;
        }

        String[] beanNames = beanFactory.getBeanDefinitionNames();
        for (String beanName : beanNames) {
            BeanDefinition beanDefinition = beanFactory.getBeanDefinition(beanName);
            String className = beanDefinition.getBeanClassName();
            if (StringUtils.isBlank(className)) {
                continue;
            }
            Class clazz;
            try {
                clazz = ClassUtils.forName(className, this.getClass().getClassLoader());
            } catch (ClassNotFoundException e) {
                if (logger.isErrorEnabled()) {
                    logger.error(e.getMessage(), e);
                }
                continue;
            }

            if (!clazz.isAnnotationPresent(WebService.class)) {
                continue;
            }

            String address;
            if (!clazz.isAnnotationPresent(AddressProvider.class)) {
                address = "/" + beanName;
            } else {
                AddressProvider provider = clazz.getAnnotation(AddressProvider.class);
                address = provider.value();
            }

            BeanDefinitionBuilder builder = BeanDefinitionBuilder.genericBeanDefinition(EndpointDefinitionParser.SpringEndpointImpl.class);
            builder.addPropertyValue("address", wrapAddress(address));
            builder.addPropertyValue("checkBlockConstruct", Boolean.TRUE);
            builder.getRawBeanDefinition().setAttribute(WIRE_BUS_ATTRIBUTE, BusWiringType.CONSTRUCTOR);
            builder.setInitMethodName("publish");
            builder.setDestroyMethodName("stop");
            builder.setLazyInit(false);
            builder.addConstructorArgReference(beanName);

            registerBeanDefinition((DefaultListableBeanFactory) beanFactory, beanName, builder.getBeanDefinition());
        }
    }

    private String wrapAddress(String address) {
        if (StringUtils.isBlank(address)) {
            address = "/";
        }

        if (!StringUtils.startsWith(address, "/")) {
            address = "/" + address;
        }

        return address;
    }

    private void registerBeanDefinition(DefaultListableBeanFactory beanFactory, String beanName, BeanDefinition beanDefinition) {
        beanFactory.registerBeanDefinition(generateBeanName(beanName, beanFactory), beanDefinition);
    }

    private String generateBeanName(String name, DefaultListableBeanFactory beanFactory) {
        if (beanFactory.containsBeanDefinition(name)) {
            name += "#endpoint";
        }

        return name;
    }
}