SpringBoot AutoConfiguration解密[译]

本文是http://sivalabs.in/2016/03/how-springboot-autoconfiguration-magic/的翻译。

我们知道SpringBoot的应用可以以非常简洁的代码去做很多的事情, 可以自动帮你注入数据库的Bean，消息队列的Bean等等等等，那么SpringBoot是怎么做到的呢？

但是在探索SpringBoot的神秘之前，我们先了解一下Spring的@Conditional注解，这是SpringBoot的AutoConfiguration的神奇之处所依赖的底层机制。

@Conditional 探秘

我们开发Spring应用的时候有时候会碰到要根据外界条件注册不同Bean的情况。

比如如果你的应用是跑在本地机器的话，你会想要让你的DataSource bean指向一个本地的数据库，而如果是跑在线上机器的话，你会想让DataSource bean指向一个生产的数据库。

你可以把这些数据库连接信息抽取到几个不同的配置文件里面去，然后在不同的环境下使用不同的配置文件。但是当你的应用被部署到新的环境下的时候，你还是要添加新的配置文件，并且重新打包。(译者注:这里其实只要把配置文件抽取到代码之外不就好了么?不知道原作者在想什么)

为了解决这个问题，Spring 3.1引入了Profiles的概念。你可以注册同一个类型bean的不同实例，然后把这些不同的实例绑定到不同的Profile, 当你运行这个Spring应用的时候，你可以指定你要激活的Profile, 这样只有跟这些被激活的Profile相关联的bean才会被注册:

@Configuration
public class AppConfig
{
    @Bean
    @Profile("DEV")
    public DataSource devDataSource() {
        ...
    }

    @Bean
    @Profile("PROD")
    public DataSource prodDataSource() {
        ...
    }
}

然后你可以通过系统属性指定激活的profile:

-Dspring.profiles.active=DEV

这种方式对于你要基于profile来决定是否注册一个bean的情况工作得很好。但是如果你要基于一些条件性的判断逻辑来决定是否注册一个bean的话，那么光靠profile是不行的。

为了给条件性地注册bean提供更高的灵活性，Spring 4提供了@Conditional的概念。通过使用@Conditional你可以基于任何条件来决定是否注册一个bean。

你的条件可能是这样的:

CLASSPATH里面有一个特定的Class
ApplicationContext里面有一个特定类型的bean
在指定的位置有指定的文件
配置文件里面有指定的配置项
系统属性里面配置了指定的属性

这些只是我能想到的一些，实际上你可以基于任何条件。下面让我们来看看Spring的@Conditional到底是如何工作的。我们先设定一个场景:

我们有一个UserDAO接口用来从数据库里面获取数据。这个接口我们有两个实现: JdbcUserDAO从MySQL数据库里面获取数据；MongoUserDAO从MongoDB里面获取数据。

通过系统属性来决定

我们想通过一个名为dbType的系统属性来决定到底使用JdbcUserDAO还是MongoUserDAO。期望的效果是，如果通过java -jar myapp.jar -DdbType=MySQL那么使用的是JdbcUserDAO, 如果通过java -jar myapp.jar -DdbType=MONGO启动，这使用MongoUserDAO。几个类的实现是这样的:

public interface UserDAO
{
    List<String> getAllUserNames();
}

public class JdbcUserDAO implements UserDAO
{
    @Override
    public List<String> getAllUserNames()
    {
        System.out.println("**** Getting usernames from RDBMS *****");
        return Arrays.asList("Siva","Prasad","Reddy");
    }
}

public class MongoUserDAO implements UserDAO
{
    @Override
    public List<String> getAllUserNames()
    {
        System.out.println("**** Getting usernames from MongoDB *****");
        return Arrays.asList("Bond","James","Bond");
    }
}

我们可以实现这样的一个MySQLDatabaseTypeCondition来检测系统属性dbType是否是MYSQL:

public class MySQLDatabaseTypeCondition implements Condition
{
    @Override
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata metadata)
    {
        String enabledDBType = System.getProperty("dbType");
        return (enabledDBType != null && enabledDBType.equalsIgnoreCase("MYSQL"));
    }
}

类似的MongoDBDatabaseTypeCondition:

public class MongoDBDatabaseTypeCondition implements Condition
{
    @Override
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata metadata)
    {
        String enabledDBType = System.getProperty("dbType");
        return (enabledDBType != null && enabledDBType.equalsIgnoreCase("MONGODB"));
    }
}

现在我们就可以通过@Conditional来判断使用JdbcUserDAO还是MongoUserDAO:

@Configuration
public class AppConfig
{
    @Bean
    @Conditional(MySQLDatabaseTypeCondition.class)
    public UserDAO jdbcUserDAO(){
        return new JdbcUserDAO();
    }

    @Bean
    @Conditional(MongoDBDatabaseTypeCondition.class)
    public UserDAO mongoUserDAO(){
        return new MongoUserDAO();
    }
}

通过CLASSPATH上是否有指定的类来判断

类似地我们可以通过判断CLASSPATH里面是否有com.mongodb.Server这个Driver类来决定是使用MongoUserDAO还是JdbcUserDAO:

public class MongoDriverPresentsCondition implements Condition
{
    @Override
    public boolean matches(ConditionContext conditionContext,AnnotatedTypeMetadata metadata)
    {
        try {
            Class.forName("com.mongodb.Server");
            return true;
        } catch (ClassNotFoundException e) {
            return false;
        }
    }
}

public class MongoDriverNotPresentsCondition implements Condition
{
    @Override
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata metadata)
    {
        try {
            Class.forName("com.mongodb.Server");
            return false;
        } catch (ClassNotFoundException e) {
            return true;
        }
    }
}

通过容器里面是否存在指定类型bean来判断

如果我们只在容器里面没有任何类型的UserDAO的bean的时候才注册MongoUserDAO。我们通过创建一个Condition来检测是否存在一个指定类型的bean:

public class UserDAOBeanNotPresentsCondition implements Condition
{
    @Override
    public boolean matches(ConditionContext conditionContext, AnnotatedTypeMetadata metadata)
    {
        UserDAO userDAO = conditionContext.getBeanFactory().getBean(UserDAO.class);
        return (userDAO == null);
    }
}

如果想通过配置文件里面的配置来决定DAO类型呢?

public class MongoDbTypePropertyCondition implements Condition
{
    @Override
    public boolean matches(ConditionContext conditionContext,
    AnnotatedTypeMetadata metadata)
    {
        String dbType = conditionContext.getEnvironment()
                            .getProperty("app.dbType");
        return "MONGO".equalsIgnoreCase(dbType);
    }
}

更优雅的实现方式: 注解

我们已经试了通过各种不同条件来实现Condition。但是其实有更优雅的、通过注解来实现Condition的方式。我们不再为MYSQL和MongoDB实现各自Condition, 我们可以实现下面这样一个DatabaseType注解:

@Target({ ElementType.TYPE, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
@Conditional(DatabaseTypeCondition.class)
public @interface DatabaseType
{
    String value();
}

然后我们实现DatabaseTypeCondition来使用这个注解来判断DAO的类型了:

public class DatabaseTypeCondition implements Condition
{
    @Override
    public boolean matches(ConditionContext conditionContext,
    AnnotatedTypeMetadata metadata)
    {
        Map<String, Object> attributes = metadata.getAnnotationAttributes(DatabaseType.class.getName());
        String type = (String) attributes.get("value");
        String enabledDBType = System.getProperty("dbType","MYSQL");
        return (enabledDBType != null && type != null && enabledDBType.equalsIgnoreCase(type));
    }
}

现在我们就可以在我们的bean上使用DatabaseType注解了:

@Configuration
@ComponentScan
public class AppConfig
{
    @Bean
    @DatabaseType("MYSQL")
    public UserDAO jdbcUserDAO(){
        return new JdbcUserDAO();
    }

    @Bean
    @DatabaseType("MONGO")
    public UserDAO mongoUserDAO(){
        return new MongoUserDAO();
    }
}

这里我们从DatabaseType注解里面获取元数据，并且把获取的值跟系统属性里面的dbType进行对比来决定是否激活bean。

我们已经看了很多例子来看@Conditional注解的作用。SpringBoot大量的使用@Conditional来实现基于条件的注册bean。你可以在spring-boot-autoconfigure-{version}.jar 的 org.springframework.boot.autoconfigure 包里面看到SpringBoot使用的大量的Condition的实现。

那么我们已经知道了SpringBoot使用@Conditional来决定是否初始化一个bean, 但是是什么机制触发了auto-configuration机制呢？我们下一节来聊聊这个事情:

SprintBoot的自动配置

SpringBoot的自动配置的关键在于@EnableAutoConfiguration这个注解。一般来说我们把我们程序的入口类加上@SpringBootApplication注解，或者如果我们想要更加细致的定制这些默认值的话:

@Configuration
@EnableAutoConfiguration
@ComponentScan
public class Application
{

}

@EnableAutoConfiguration注解通过扫描CLASSPATH里面所有的组件，然后基于条件来决定是否注册bean来使得Spring的ApplicationContext自动配置。

SpringBoot在spring-boot-autoconfigure-{version}.jar里面提供了很多AutoConfiguration的类来负责注册各种不同的组件。

一般来说AutoConfiguration类上面会标上@Configuration注解来标明它是一个Spring的配置类，标上@EnableConfigurationProperties来绑定自定义的配置值，并且会在一到多个方法上标上@Conditional注解来标记注册方法。

比如我们来看看org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration:

@Configuration
@ConditionalOnClass({ DataSource.class, EmbeddedDatabaseType.class })
@EnableConfigurationProperties(DataSourceProperties.class)
@Import({ Registrar.class, DataSourcePoolMetadataProvidersConfiguration.class })
public class DataSourceAutoConfiguration
{
    ...
    ...
    @Conditional(DataSourceAutoConfiguration.EmbeddedDataSourceCondition.class)
    @ConditionalOnMissingBean({ DataSource.class, XADataSource.class })
    @Import(EmbeddedDataSourceConfiguration.class)
    protected static class EmbeddedConfiguration {

    }

    @Configuration
    @ConditionalOnMissingBean(DataSourceInitializer.class)
    protected static class DataSourceInitializerConfiguration {
        @Bean
        public DataSourceInitializer dataSourceInitializer() {
        return new DataSourceInitializer();
        }
    }

    @Conditional(DataSourceAutoConfiguration.NonEmbeddedDataSourceCondition.class)
    @ConditionalOnMissingBean({ DataSource.class, XADataSource.class })
    protected static class NonEmbeddedConfiguration {
        @Autowired
        private DataSourceProperties properties;

        @Bean
        @ConfigurationProperties(prefix = DataSourceProperties.PREFIX)
        public DataSource dataSource() {
            DataSourceBuilder factory = DataSourceBuilder
                    .create(this.properties.getClassLoader())
                    .driverClassName(this.properties.getDriverClassName())
                    .url(this.properties.getUrl()).username(this.properties.getUsername())
                    .password(this.properties.getPassword());
            if (this.properties.getType() != null) {
                factory.type(this.properties.getType());
            }
            return factory.build();
        }
    }
    ...
    ...
    @Configuration
    @ConditionalOnProperty(prefix = "spring.datasource", name = "jmx-enabled")
    @ConditionalOnClass(name = "org.apache.tomcat.jdbc.pool.DataSourceProxy")
    @Conditional(DataSourceAutoConfiguration.DataSourceAvailableCondition.class)
    @ConditionalOnMissingBean(name = "dataSourceMBean")
    protected static class TomcatDataSourceJmxConfiguration {
        @Bean
        public Object dataSourceMBean(DataSource dataSource) {
        ....
        ....
        }
    }
    ...
    ...
}

这里DataSourceAutoConfiguration上标记了一个@ConditionalOnClass({ DataSource.class,EmbeddedDatabaseType.class }), 这样只有当CLASSPATH上有DataSource.class和EmbeddedDatabaseType.class的时候，DataSourceAutoConfiguration 才会生效。

这个类还被@EnableConfigurationProperties(DataSourceProperties.class)标记了，这样application.properties里面相关的配置值会自动绑定到DataSourceProperties上面。

@ConfigurationProperties(prefix = DataSourceProperties.PREFIX)
public class DataSourceProperties implements BeanClassLoaderAware, EnvironmentAware, InitializingBean {

    public static final String PREFIX = "spring.datasource";
    ...
    ...
    private String driverClassName;
    private String url;
    private String username;
    private String password;
    ...
    //setters and getters
}

有了这个配置属性类，所有spring.datasource.*的配置都会自动绑定到DataSourceProperties:

spring.datasource.url=jdbc:mysql://localhost:3306/test
spring.datasource.username=root
spring.datasource.password=secret
spring.datasource.driver-class-name=com.mysql.jdbc.Driver

你还能看到一些内部类以及bean的定义方法被SpringBoot的@ConditionalOnMissingBean, @ConditionOnClass, @ConditionalOnProperty等等标记。

这些bean的定义只有在那些条件满足的时候才会注册。

在spring-boot-autoconfigure-{version}.jar里面你还能看到其它的AutoConfiguration类:

org.springframework.boot.autoconfigure.web.DispatcherServletAutoConfiguration
org.springframework.boot.autoconfigure.orm.jpa.HibernateJpaAutoConfiguration
org.springframework.boot.autoconfigure.data.jpa.JpaRepositoriesAutoConfiguration
org.springframework.boot.autoconfigure.jackson.JacksonAutoConfiguration

等等。我希望通过今天文章的介绍大家已经理解SpringBoot是怎么做到自动配置bean的了。

最后编辑于：2017.12.06 07:02:17

人面猴
序言：七十年代末，一起剥皮案震惊了整个滨河市，随后出现的几起案子，更是在滨河造成了极大的恐慌，老刑警刘岩，带你破解...
沈念sama阅读 160,165评论 4赞 364
死咒
序言：滨河连续发生了三起死亡事件，死亡现场离奇诡异，居然都是意外死亡，警方通过查阅死者的电脑和手机，发现死者居然都...
沈念sama阅读 67,720评论 1赞 298
救了他两次的神仙让他今天三更去死
文/潘晓璐我一进店门，熙熙楼的掌柜王于贵愁眉苦脸地迎上来，“玉大人，你说我怎么就摊上这事。” “怎么了？”我有些...
开封第一讲书人阅读 109,849评论 0赞 244
道士缉凶录：失踪的卖姜人
文/不坏的土叔我叫张陵，是天一观的道长。经常有香客问我，道长，这世上最难降的妖魔是什么？我笑而不...
开封第一讲书人阅读 44,245评论 0赞 213
港岛之恋（遗憾婚礼）
正文为了忘掉前任，我火速办了婚礼，结果婚礼上，老公的妹妹穿的比我还像新娘。我一直安慰自己，他们只是感情好，可当我...
茶点故事阅读 52,596评论 3赞 288
恶毒庶女顶嫁案：这布局不是一般人想出来的
文/花漫我一把揭开白布。她就那样静静地躺着，像睡着了一般。火红的嫁衣衬着肌肤如雪。梳的纹丝不乱的头发上，一...
开封第一讲书人阅读 40,747评论 1赞 222
城市分裂传说
那天，我揣着相机与录音，去河边找鬼。笑死，一个胖子当着我的面吹牛，可吹牛的内容都是我干的。我是一名探鬼主播，决...
沈念sama阅读 31,977评论 2赞 315
双鸳鸯连环套：你想象不到人心有多黑
文/苍兰香墨我猛地睁开眼，长吁一口气：“原来是场噩梦啊……” “哼！你这毒妇竟也来了？” 一声冷哼从身侧响起，我...
开封第一讲书人阅读 30,708评论 0赞 204
万荣杀人案实录
序言：老挝万荣一对情侣失踪，失踪者是张志新（化名）和其女友刘颖，没想到半个月后，有当地人在树林里发现了一具尸体，经...
沈念sama阅读 34,448评论 1赞 246
护林员之死
正文独居荒郊野岭守林人离奇死亡，尸身上长有42处带血的脓包…… 初始之章·张勋以下内容为张勋视角年9月15日...
茶点故事阅读 30,657评论 2赞 249
白月光启示录
正文我和宋清朗相恋三年，在试婚纱的时候发现自己被绿了。大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
茶点故事阅读 32,141评论 1赞 261
活死人
序言：一个原本活蹦乱跳的男人离奇死亡，死状恐怖，灵堂内的尸体忽然破棺而出，到底是诈尸还是另有隐情，我是刑警宁泽，带...
沈念sama阅读 28,493评论 3赞 258
日本核电站爆炸内幕
正文年R本政府宣布，位于F岛的核电站，受9级特大地震影响，放射性物质发生泄漏。R本人自食恶果不足惜，却给世界环境...
茶点故事阅读 33,153评论 3赞 238
男人毒药：我在死后第九天来索命
文/蒙蒙一、第九天我趴在偏房一处隐蔽的房顶上张望。院中可真热闹，春花似锦、人声如沸。这庄子的主人今日做“春日...
开封第一讲书人阅读 26,108评论 0赞 8
一桩弑父案，背后竟有这般阴谋
文/苍兰香墨我抬头看了看天上的太阳。三九已至，却和暖如春，着一层夹袄步出监牢的瞬间，已是汗流浃背。一阵脚步声响...
开封第一讲书人阅读 26,890评论 0赞 198
情欲美人皮
我被黑心中介骗来泰国打工，没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留，地道东北人。一个月前我还...
沈念sama阅读 35,799评论 2赞 277
代替公主和亲
正文我出身青楼，却偏偏与公主长得像，于是被迫代替她去往敌国和亲。传闻我的和亲对象是个残疾皇子，可洞房花烛夜当晚...
茶点故事阅读 35,685评论 2赞 272