Spring BeanPostProcessor : DataSourceInitializerPostProcessor

最新推荐文章于 2025-06-29 22:30:35 发布

原创最新推荐文章于 2025-06-29 22:30:35 发布 · 1.9k 阅读

0 ·

CC 4.0 BY-SA版权

Spring Boot 专栏收录该内容

103 篇文章

订阅专栏

本文深入探讨了SpringBoot中DataSourceInitializerPostProcessor的工作原理，详细解释了如何通过DataSourceInitializerInvoker执行数据库表创建(schema-*.sql)及数据填充(data-*.sql)的流程。

概述

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

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;
	}

}