【自定义starter实现】@Enable开启自动装配

综合

发布日期: 2019-08-14

文章字数: 1.2k

阅读时长: 5 分

阅读次数:

自动配置非常方便，当我们要操作Mongodb的时候，只需要引入spring-boot-starter-data-mongodb的依赖，然后配置Mongodb的链接信息 spring.data.mongodb.uri=mongodb://localhost/test 就可以使用MongoTemplate来操作数据，MongoTemplate的初始化工作全部交给Starter来完成。

自动配置麻烦的是当出现错误时，排查问题的难度上升了。自动配置的逻辑都在Spring Boot Starter中，要快速的能够定位问题，那么你必须得了解Spring Boot Starter的内部原理。接下来我们自己动手来实现一个Spring Boot Starter。

开发Starter步骤：

创建Starter项目
定义Starter需要的配置（Properties）类
编写自动配置类
编写spring.factories文件加载自动配置类
编写配置提示文件spring-configuration-metadata.json（不是必须的）

创建一个项目spring-boot-starter-demo，Pom.xml配置如下：

去除xml中的build

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
    <dependency>
        <groupId>org.projectlombok</groupId>
        <artifactId>lombok</artifactId>
        <optional>true</optional>
    </dependency>
</dependencies>

创建一个配置类，用于在属性文件中配置值，相当于spring.data.mongo这种形式

import org.springframework.boot.context.properties.ConfigurationProperties;
import lombok.Data;

@Data
@ConfigurationProperties("spring.user")
public class UserPorperties {
    private String name;
}

@ConfigurationProperties指定了配置的前缀，也就是spring.user.name=XXX

再定义一个Client，相当于MongoTemplate，里面定一个方法，用于获取配置中的值

public class UserClient {
    private UserPorperties userPorperties;
    public UserClient() {
    }
    public UserClient(UserPorperties p) {
        this.userPorperties = p;
    }
    public String getName() {
        return userPorperties.getName();
    }
}

一个最基本的Starter包定义好了，但目前肯定是不能使用UserClient ，因为我们没有去自动构建UserClient 的实例，接下来开始构建UserClient

@Configuration
@EnableConfigurationProperties(UserPorperties.class)
public class UserAutoConfigure {
    @Bean
    @ConditionalOnProperty(prefix = "spring.user",value = "enabled",havingValue = "true")
    public UserClient userClient(UserPorperties userPorperties) {
        return new UserClient(userPorperties);
    }
}

Spring Boot会默认扫描跟启动类平级的包，如果我们的Starter跟启动类不在同一个主包下，如何让UserAutoConfigure 生效？

第一种方式：

在resources下创建一个META-INF文件夹，然后在META-INF文件夹中创建一个spring.factories文件，文件中指定自动配置的类

org.springframework.boot.autoconfigure.EnableAutoConfiguration=\
com.ciwei.demo.UserAutoConfigure

Spring Boot启动时会去读取spring.factories文件，然后根据配置激活对应的配置类，到底为止就简单的实现了一个Starter包。

现在可以在其他的项目中引入这个Starter包：

<dependency>
    <groupId>com.example</groupId>
    <artifactId>spring-boot-starter-demo</artifactId>
    <version>0.0.1-SNAPSHOT</version>
</dependency>

引入之后就直接可以使用UserClient，UserClient 在项目启动的时候已经自动初始化好。

@RestController
public class UserController {
    @Autowired
    private UserClient userClient;
    @GetMapping("/user/name")
    public String getUserName() {
        return userClient.getName();
    }
}

很多时候我们不想引入了Starter包就执行初始化的逻辑，想要用户来指定是否要开启Starter包的自动配置功能，比如常用的@EnableAsync这个注解就是用于开启调用方法异步执行的功能。

第二种方式：

同样的我们也可以通过注解的方式来开启是否自动配置，如果用注解的方式，那么spring.factories就不需要编写了，下面来看怎么定义启用自动配置的注解。

@Target({ElementType.TYPE})
@Retention(RetentionPolicy.RUNTIME)
@Documented
@Inherited
@Import({UserAutoConfigure.class})
public @interface EnableUserClient {

}

核心是@Import({UserAutoConfigure.class})这行代码，通过导入的方式实现把UserAutoConfigure实例加入SpringIOC容器中，这样就能开启自动配置了。

使用方式就是在启动类上加上该注解，代码入下：

@EnableUserClient
@SpringBootApplication
public class SpringBootDemoApplication {
    public static void main(String[] args) {
        SpringApplication.run(SpringBootDemoApplication.class, args);
    }
}

在某些场景下，UserAutoConfigure中会配置多个对象，对于这些对象，不想全部配置，也想让用户指定需要开启配置的时候再去构建对象，这个时候我们可以通过@ConditionalOnProperty来指定是否开启配置的功能，代码如下：

@Bean
@ConditionalOnProperty(prefix = "spring.user",value = "enabled",havingValue = "true")
public UserClient userClient(UserPorperties userPorperties) {
    return new UserClient(userPorperties);
}

通过上面的配置，只有当启动类加了@EnableUserClient并且配置文件中spring.user.enabled=true的时候才会自动配置UserClient

定义提示内容需要在META-INF中创建一个spring-configuration-metadata.json

添加依赖

<dependency>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-configuration-processor</artifactId>
    <optional>true</optional><!--true表明该依赖不会间接传递-->
</dependency>