Spring BeanPostProcessor : DataSourceInitializerPostProcessor

概述

DataSourceInitializerPostProcessor是Springboot提供的一个用于初始化数据源的BeanPostProcessor。它存在的目的是检测数据源DataSource bean实例的初始化，一旦检测到，它会尝试获取一个实现类为DataSourceInitializerInvoker的bean。而bean DataSourceInitializerInvoker实现了InitializingBean接口，所以此bean随后会被实例化，而在此bean的初始化过程中，它对数据源DataSource bean执行了初始化操作，具体来讲，是执行数据库表创建文件schema-*.sql创建数据库表(schema)，然后在表结构创建完成时执行表数据初始化文件data-*.sql填充有关表数据(data)。

DataSourceInitializerInvoker初始化数据源的过程如下图所示:

DataSourceInitializerInvoker初始化方法开始

调用DataSourceInitializer执行schema-*.sql

调用DataSourceInitializer执行 data-*.sql

DataSourceInitializerInvoker初始化方法结束

源代码分析

package org.springframework.boot.autoconfigure.jdbc;

import javax.sql.DataSource;

import org.springframework.beans.BeansException;
import org.springframework.beans.factory.BeanFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.core.Ordered;

class DataSourceInitializerPostProcessor implements BeanPostProcessor, Ordered {

	@Override
	public int getOrder() {
		return Ordered.HIGHEST_PRECEDENCE + 1;
	}

	@Autowired
	private BeanFactory beanFactory;

	// BeanPostProcessor 定义的方法，用于在每个bean创建时初始化前执行，
	// 这里什么都不做，直接返回目标bean
	@Override
	public Object postProcessBeforeInitialization(Object bean, String beanName)
			throws BeansException {
		return bean;
	}

	// BeanPostProcessor 定义的方法，用于在每个bean创建时初始化后执行，
	// 这里检测目标bean是否是一个DataSource，如果是的话，尝试从容器获取一个类型
	// 为 DataSourceInitializerInvoker 的 bean。
	@Override
	public Object postProcessAfterInitialization(Object bean, String beanName)
			throws BeansException {
		if (bean instanceof DataSource) {
			// force initialization of this bean as soon as we see a DataSource
			// 一旦检测到bean DataSource初始化完成，就开始获取bean DataSourceInitializerInvoker。	
			this.beanFactory.getBean(DataSourceInitializerInvoker.class);
		}
		return bean;
	}

}