com.github.youyinnn.youdbutils.ioc.YouServiceIocContainer Maven / Gradle / Ivy

Go to download
package com.github.youyinnn.youdbutils.ioc;

import com.github.youyinnn.youdbutils.dao.YouDao;
import com.github.youyinnn.youdbutils.dao.YouDaoContainer;
import com.github.youyinnn.youdbutils.exceptions.AutowiredException;
import com.github.youyinnn.youdbutils.ioc.annotations.Autowired;
import com.github.youyinnn.youdbutils.ioc.annotations.YouService;
import com.github.youyinnn.youdbutils.ioc.proxy.TransactionProxyGenerator;
import com.github.youyinnn.youdbutils.utils.ReflectionUtils;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Map;

/**
 * YouService的Ioc容器管理.
 * 分为单例容器和多例容器.
 * 提供:
 *  1.注册单例/多例YouService的方法;
 *  2.获取YouService对象的方法;
 *  3.查看当前Ioc内注册过的YouService的方法;
 *
 * @author youyinnn
 */
public class YouServiceIocContainer {

    private static HashMap              prototypeServiceMap             = new HashMap<>();
    private static HashMap              singletonServiceMap             = new HashMap<>();

    private YouServiceIocContainer() {}

    private static void addSingletonYouService(ServiceIocBean serviceBean) {
        String className = serviceBean.getClassName();
        serviceBean.setSingleton(TransactionProxyGenerator.getProxyObject(serviceBean.getServiceClass()));
        singletonServiceMap.put(className,serviceBean);
    }

    private static void addPrototypeYouService(ServiceIocBean serviceBean) {
        String className = serviceBean.getClassName();
        prototypeServiceMap.put(className,serviceBean);
    }

    /**
     * Gets you service.
     *
     * @param serviceClass the service class
     * @return the you service
     * @throws AutowiredException the autowired limited exception
     */
    public static Object getYouService(Class serviceClass) throws AutowiredException {
        ServiceIocBean serviceIocBean = singletonServiceMap.get(serviceClass.getName());
        if (serviceIocBean == null) {
            serviceIocBean = prototypeServiceMap.get(serviceClass.getName());
            if (serviceIocBean == null) {
                return null;
            } else {
                return autowired(TransactionProxyGenerator.getProxyObject(serviceIocBean.getServiceClass()));
            }
        } else {
            return autowired(serviceIocBean.getSingleton());
        }
    }

    /**
     * Show service map.
     */
    public static void showServiceMap(){
        System.out.println("[ prototype service ]:");
        for (Map.Entry stringIocBeanEntry : prototypeServiceMap.entrySet()) {
            System.out.println(stringIocBeanEntry.getKey()+" : "+stringIocBeanEntry.getValue());
        }
        System.out.println("[ singleton service ]:");
        for (Map.Entry stringIocBeanEntry : singletonServiceMap.entrySet()) {
            System.out.println(stringIocBeanEntry.getKey()+" : "+stringIocBeanEntry.getValue());
        }
    }

    /**
     * 这个方法检查当前ioc容器中是否已经存在该YouService
     * @param youService
     * @return
     */
    private static boolean hasYouService(Class youService) {
        return singletonServiceMap.containsKey(youService.getName()) | prototypeServiceMap.containsKey(youService.getName());
    }

    /**
     * 对于youService对象的自动装配方法 只装配Dao和YouService
     * @param youService
     * @return
     * @throws AutowiredException
     */
    private static Object autowired(Object youService) throws AutowiredException {
        // Autowired实现 只接受YouDao和YouService的自动装配
        ArrayList declaredFields = ReflectionUtils.getDeclaredFields(youService, Autowired.class);
        for (Field declaredField : declaredFields) {
            Class type = declaredField.getType();
            YouService youServiceAnnotation = type.getAnnotation(YouService.class);
            // 装载YouDao和YouService
            if (youServiceAnnotation == null) {
                if ("YouDao".equals(type.getSuperclass().getSimpleName())) {
                    YouDao autowiredDao = YouDaoContainer.getDao(type);
                    ReflectionUtils.setFieldValue(youService,declaredField.getName(),autowiredDao);
                } else {
                    throw new AutowiredException("不支持的自动装配类型：["+type.getSimpleName()+" "+declaredField.getName()+"].");
                }
            } else {
                Object autowiredDaoYouService = getYouService(type);
                ReflectionUtils.setFieldValue(youService,declaredField.getName(),autowiredDaoYouService);
            }
        }
        return youService;
    }

    /**
     * 注册YouService到Ioc容器中.
     *
     * @param aClass the a class
     */
    static void registerYouService(Class aClass) {
        YouService youService = aClass.getAnnotation(YouService.class);
        if (!hasYouService(aClass)) {
            // 单例service
            if (youService.scope().equals(ServiceIocBean.SINGLETON)){
                addSingletonYouService(new ServiceIocBean(aClass, ServiceIocBean.SINGLETON));
            } else {
                addPrototypeYouService(new ServiceIocBean(aClass, ServiceIocBean.PROTOTYPE));
            }
        }
    }

}