com.iqiny.silly.spring.SpringSillyContext Maven / Gradle / Ivy

Go to download
/*
 *  Copyright  iqiny.com
 *
 *  https://gitee.com/iqiny/silly
 *
 *  project name：silly-spring
 *  project description：top silly project pom.xml file
 */
package com.iqiny.silly.spring;

import com.iqiny.silly.core.base.SillyContext;
import com.iqiny.silly.core.base.SillyOrdered;
import com.iqiny.silly.core.common.SillyCoreUtil;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.config.BeanDefinition;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.BeanDefinitionRegistry;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.lang.NonNull;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.function.Consumer;
import java.util.function.Supplier;

/**
 * 傻瓜 Spring 相关操作
 */
public class SpringSillyContext implements SillyContext, ApplicationContextAware {

    private final static Log log = LogFactory.getLog(SpringSillyContext.class);

    private static ApplicationContext applicationContext;

    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    /**
     * 获取Bean 同时会在容器中寻找相应Bean 注册到Spring环境中
     *
     * @param clazz
     * @param 
     * @return
     */
    @Override
    public  List getBeanList(String category, Class clazz) {
        List beanList = new ArrayList<>();
        try {
            final Set> entries = applicationContext.getBeansOfType(clazz).entrySet();
            for (Map.Entry entry : entries) {
                T value = entry.getValue();
                boolean available = SillyCoreUtil.available(category, value);
                if (available) {
                    beanList.add(value);
                }
            }
        } catch (BeansException ignore) {
        }

        return SillyCoreUtil.orderCollection(beanList);
    }


    @Override
    public  T getBean(String category, String beanName) {
        try {
            Object bean = applicationContext.getBean(beanName);
            return (T) SillyCoreUtil.availableOrNull(category, bean);
        } catch (BeansException ignore) {
        }
        return null;
    }

    @Override
    public  T getBean(String category, Class clazz) {
        List list = getBeanList(category, clazz);
        return SillyCoreUtil.availableOne(category, list);
    }


    @Override
    public  T getNextBean(SillyOrdered order, String category, Class clazz) {
        List list = getBeanList(category, clazz);
        if (!list.isEmpty() && order == null) {
            return list.get(0);
        }

        for (T t : list) {
            if (t.order() == order.order()) {
                if (t.hashCode() > order.hashCode()) {
                    return t;
                }
            } else if (t.order() > order.order()) {
                return t;
            }
        }
        return null;
    }

    @Override
    public  T registerBean(String category, String beanName, Class clazz, Consumer