top.wboost.common.base.service.impl.ConfigForAutowiredService Maven / Gradle / Ivy

Go to download
package top.wboost.common.base.service.impl;

import org.springframework.context.ApplicationContext;
import org.springframework.context.event.ContextRefreshedEvent;
import org.springframework.stereotype.Service;
import org.springframework.util.ClassUtils;
import top.wboost.common.base.annotation.AutoRootApplicationConfig;
import top.wboost.common.base.repository.BaseRepository;
import top.wboost.common.log.entity.Logger;
import top.wboost.common.log.util.LoggerUtil;
import top.wboost.common.system.exception.BeanNotFindException;
import top.wboost.common.system.exception.SystemException;
import top.wboost.common.util.ReflectUtil;
import top.wboost.common.utils.web.interfaces.context.EzApplicationListener;
import top.wboost.common.utils.web.utils.AopUtils;
import top.wboost.common.utils.web.utils.SpringBeanUtil;

import java.util.Map;

@SuppressWarnings({ "rawtypes", "unchecked" })
@AutoRootApplicationConfig
public class ConfigForAutowiredService implements EzApplicationListener {

    private Logger log = LoggerUtil.getLogger(getClass());

    @Override
    public void onWebApplicationEvent(ContextRefreshedEvent event) {
        onRootApplicationEvent(event);
    }

    @Override
    public void onRootApplicationEvent(ContextRefreshedEvent event) {
        ApplicationContext context = event.getApplicationContext();
        Map servicess = context.getBeansWithAnnotation(Service.class);
        servicess.forEach((name, bean) -> {
            Object target = AopUtils.getTarget(bean);
            if (ClassUtils.isAssignable(BaseJpaServiceImpl.class, target.getClass())) {
                autowired((BaseJpaServiceImpl) target);
            }
        });
    }

    protected void autowired(BaseJpaServiceImpl service) {
        Class clazz = ClassUtils.getUserClass(service);
        Object repository = null;
        Class genericInterfaces = ReflectUtil.getGenericInterfaces(clazz, 1);
        repository = SpringBeanUtil.getBean(genericInterfaces);
        String defaultBeanName = null;
        if (repository == null) {
            if (clazz.getSimpleName().indexOf("ServiceImpl") == -1) {
                throw new SystemException("服务层类名必须以ServiceImpl结尾");
            }
            String className = clazz.getSimpleName().substring(0, clazz.getSimpleName().indexOf("ServiceImpl"));
            className = Character.toLowerCase(className.charAt(0)) + className.substring(1, className.length());
            defaultBeanName = className + "Repository";
            repository = SpringBeanUtil.getBean(defaultBeanName);
            if (repository == null) {
                log.warn("cant find bean:" + defaultBeanName + ",please check");
                return;
            }
        } else {
            defaultBeanName = genericInterfaces.getName();
        }
        if (repository != null && repository instanceof BaseRepository) {
            BaseRepository findRepository = (BaseRepository) repository;
            log.info("auto config bean: " + defaultBeanName + " to service:" + clazz.getName());
            service.setRepository(findRepository);
        } else {
            throw new BeanNotFindException(defaultBeanName, BaseRepository.class);
        }
    }

    @Override
    public void onBootApplicationEvent(ContextRefreshedEvent event) {
        onWebApplicationEvent(event);
    }

    public boolean doWebAndRootApplicationListener(ContextRefreshedEvent event) {
        return false;
    }

}