spring 源码学习(4) 之注册bean的依赖关系

代码

<bean id="A1" class="com.liubin.study.spring.beans.loader.SimpleLoaderUser"></bean>
<bean id="A2" class="com.liubin.study.spring.beans.loader.SimpleLoaderUser"></bean>
<bean id="B" depends-on="A1,A2"/>

spring在创建bean的时候，有时候会存在指定的依赖关系，既B这个bean在初始化前，A1、A2这两个bean必须初始化。

一、依赖关系的存储

/** Map between dependent bean names: bean name --> Set of dependent bean names */
private final Map<String, Set<String>> dependentBeanMap = new ConcurrentHashMap<>(64);

/** Map between depending bean names: bean name --> Set of bean names for the bean's dependencies */
private final Map<String, Set<String>> dependenciesForBeanMap = new ConcurrentHashMap<>(64);

dependentBeanMap（指定的bean与目前已经注册的依赖这个指定的bean的所有bean的依赖关系的缓存（我依赖的））
dependenciesForBeanMap(指定bean与目前已经注册的创建这个bean所需依赖的所有bean的依赖关系的缓存（依赖我的）)

二、依赖关系的使用

/**
 * 注册依赖关系
 * @param beanName bean的名称
 * @param dependentBeanName 需要依赖前面的bean的bean名称
 */
public void registerDependentBean(String beanName, String dependentBeanName) {
    // 校验依赖关系是否存在
    String canonicalName = canonicalName(beanName);
    Set<String> dependentBeans = this.dependentBeanMap.get(canonicalName);
    if (dependentBeans != null && dependentBeans.contains(dependentBeanName)) {
        return;
    }

    // 如果不存在，需要创建
    synchronized (this.dependentBeanMap) {
        dependentBeans = this.dependentBeanMap.get(canonicalName);
        if (dependentBeans == null) {
            dependentBeans = new LinkedHashSet<>(8);
            this.dependentBeanMap.put(canonicalName, dependentBeans);
        }
        dependentBeans.add(dependentBeanName);
    }
    synchronized (this.dependenciesForBeanMap) {
        Set<String> dependenciesForBean = this.dependenciesForBeanMap.get(dependentBeanName);
        if (dependenciesForBean == null) {
            dependenciesForBean = new LinkedHashSet<>(8);
            this.dependenciesForBeanMap.put(dependentBeanName, dependenciesForBean);
        }
        dependenciesForBean.add(canonicalName);
    }
}

/**
 * 校验dependentBeanName是否依赖beanName
 * 
 * @param beanName the name of the bean to check
 * @param dependentBeanName the name of the dependent bean
 * @since 4.0
 */
protected boolean isDependent(String beanName, String dependentBeanName) {
    return isDependent(beanName, dependentBeanName, null);
}

/**
 *校验dependentBeanName是否依赖beanName
 */
private boolean isDependent(String beanName, String dependentBeanName, @Nullable Set<String> alreadySeen) {
    if (alreadySeen != null && alreadySeen.contains(beanName)) {
        return false;
    }
    String canonicalName = canonicalName(beanName);
    Set<String> dependentBeans = this.dependentBeanMap.get(canonicalName);
    if (dependentBeans == null) {
        return false;
    }
    if (dependentBeans.contains(dependentBeanName)) {
        return true;
    }
    for (String transitiveDependency : dependentBeans) {
        if (alreadySeen == null) {
            alreadySeen = new HashSet<>();
        }
        alreadySeen.add(beanName);
        if (isDependent(transitiveDependency, dependentBeanName, alreadySeen)) {
            return true;
        }
    }
    return false;
}

/**
 * Determine whether a dependent bean has been registered for the given name.
 * @param beanName the name of the bean to check
 */
protected boolean hasDependentBean(String beanName) {
    return this.dependentBeanMap.containsKey(beanName);
}

/**
 * Return the names of all beans which depend on the specified bean, if any.
 * @param beanName the name of the bean
 * @return the array of dependent bean names, or an empty array if none
 */
public String[] getDependentBeans(String beanName) {
    Set<String> dependentBeans = this.dependentBeanMap.get(beanName);
    if (dependentBeans == null) {
        return new String[0];
    }
    return StringUtils.toStringArray(dependentBeans);
}

/**
 * Return the names of all beans that the specified bean depends on, if any.
 * @param beanName the name of the bean
 * @return the array of names of beans which the bean depends on,
 * or an empty array if none
 */
public String[] getDependenciesForBean(String beanName) {
    Set<String> dependenciesForBean = this.dependenciesForBeanMap.get(beanName);
    if (dependenciesForBean == null) {
        return new String[0];
    }
    return dependenciesForBean.toArray(new String[dependenciesForBean.size()]);
}