詳解SpringBoot如何自定義一個Starter

更新時間：2022年11月16日 08:22:14 作者：一只小coder

小伙伴們曾經(jīng)可能都經(jīng)歷過整天寫著CURD的業(yè)務(wù)，都沒寫過一些組件相關(guān)的東西，這篇文章記錄一下SpringBoot如何自定義一個Starter。原理和理論就不用多說了，可以在網(wǎng)上找到很多關(guān)于該方面的資料，這里主要分享如何自定義

需求

在一個項目中，用戶需要發(fā)送消息，可以通過郵件，QQ，微信，釘釘，飛書等，目前這些發(fā)送消息的方式都已經(jīng)提供了公開的API，想要實現(xiàn)在項目中通過簡單的配置各個發(fā)發(fā)送方的發(fā)送方信息，然后直接調(diào)用發(fā)送的API，發(fā)送信息即可，下面舉個例子：

配置

message:
  email:
    username: Aden
    password: 123456
    key: HJFHADJSFBDASFHUADSINF
    api-url: http://blog.qiyuan.run
  feishu:
    user-name: Aden
    pass-word: 654321
    key: HFJKADSBFJKADSJFKADSNFAD
    api-url: http://blog.qiyuan.run

調(diào)用

    @Autowired
    SendEmailMessageServiceImpl emailMessageService;
    
    @Autowired
    SendFeishuMessageServiceImpl feishuMessageService;

    public boolean sendEmail(String msg) {
        return emailMessageService.sendMessage(msg);
    }
    
    public boolean sendFeishu(String msg){
        return feishuMessageService.sendMessage(msg);
    }

效果的就是以上這樣，只要通過配置需要發(fā)送消息的配置，自動注入發(fā)送消息的API，就可以實現(xiàn)發(fā)送消息了，以下是實現(xiàn)過程。

starter創(chuàng)建

第一步，需要為你的starter取一個響亮的名字，spring的官方文檔中說明，官方的 starter 的命名格式為 spring-boot-starter-{xxxx} 比如spring-boot-starter-activemq第三方我們自己的命名格式為 {xxxx}-spring-boot-starter。比如mybatis-spring-boot-starter，此處，我命名為message-spring-boot-starter

自定義配置信息類

因為要在項目中的配置文件中寫配置信息，所以在這個starter中，我們需要通過一個配置信息類來接收配置的信息。

@ConfigurationProperties(prefix = "message")
@Data
public class MessageProperties {
    /**
     * 郵箱消息
     */
    private MessageConfigInfo email = new MessageConfigInfo();
    /**
     * 飛書消息
     */
    private MessageConfigInfo feishu = new MessageConfigInfo();

    @Data
    public static class MessageConfigInfo {
        /**
         * 用戶名
         */
        private String userName;
        /**
         * 密碼
         */
        private String passWord;
        /**
         * 秘鑰
         */
        private String key;
        /**
         * 消息發(fā)送API
         */
        private String apiUrl;
    }
}

發(fā)送消息的實現(xiàn)

由于需要通過這個starter實現(xiàn)發(fā)送消息，所以這里可能得要引入發(fā)送郵件，發(fā)送飛書的官方API，這里就不搞這么復雜了，主要還是看過程，自定義一個接口模擬一下即可。

模擬接口定義

public interface SendMessageService {
    Boolean sendMessage(String message);
}

模擬接口實現(xiàn)

public class SendEmailMessageServiceImpl implements SendMessageService {

    private MessageProperties messageProperties;

    public SendEmailMessageServiceImpl(MessageProperties messageProperties) {
        this.messageProperties = messageProperties;
    }

    @Override
    public Boolean sendMessage(String message) {
        System.out.println(messageProperties.toString() + "  開發(fā)發(fā)送郵件，發(fā)送內(nèi)容為：" + message);
        return true;
    }
}

public class SendFeishuMessageServiceImpl implements SendMessageService {

    private MessageProperties messageProperties;

    public SendFeishuMessageServiceImpl(MessageProperties messageProperties) {
        this.messageProperties = messageProperties;
    }

    @Override
    public Boolean sendMessage(String message) {
        System.out.println(messageProperties.toString() + "  開發(fā)發(fā)送郵件，發(fā)送內(nèi)容為：" + message);
        return true;
    }
}

自動配置類

@EnableConfigurationProperties(value = MessageProperties.class)
@Configuration
public class MessageAutoConfiguration {
    /**
     * 給發(fā)送郵件的實現(xiàn)類，注入配置信息
     * @param messageProperties
     * @return
     */
    @Bean
    public SendEmailMessageServiceImpl emailMessageConfig(MessageProperties messageProperties){
        return new SendEmailMessageServiceImpl(messageProperties);
    }


    /**
     * 給發(fā)送飛書的實現(xiàn)類，注入配置信息
     * @param messageProperties
     * @return
     */
    @Bean
    public SendFeishuMessageServiceImpl feishuMessageConfig(MessageProperties messageProperties){
        return new SendFeishuMessageServiceImpl(messageProperties);
    }
}

spring.factories

org.springframework.boot.autoconfigure.EnableAutoConfiguration=run.qiyuan.message.config.MessageAutoConfiguration

項目結(jié)構(gòu)

編寫完之后，mvn install即可。

如何使用該starter

在我們的項目中，引入自定義starter的坐標

        <dependency>
            <groupId>run.qiyuan</groupId>
            <artifactId>message-spring-boot-starter</artifactId>
            <version>0.0.1-SNAPSHOT</version>
        </dependency>

然后在配置文件中配置相關(guān)的信息

message:
  email:
    username: Aden
    password: 123456
    key: HJFHADJSFBDASFHUADSINF
    api-url: http://blog.qiyuan.run
  feishu:
    user-name: Aden
    pass-word: 654321
    key: HFJKADSBFJKADSJFKADSNFAD
    api-url: http://blog.qiyuan.run

測試

@SpringBootApplication
public class TeachApplication {

    public static void main(String[] args) {
        ConfigurableApplicationContext context = SpringApplication.run(TeachApplication.class, args);
        SendEmailMessageServiceImpl emailMessageService = context.getBean(SendEmailMessageServiceImpl.class);
        emailMessageService.sendMessage("你好，Starter!，這是一封郵件信息！\n\n");
        SendFeishuMessageServiceImpl feishuMessageService = context.getBean(SendFeishuMessageServiceImpl.class);
        feishuMessageService.sendMessage("你好，Starter!，這是一封飛書信息！");
    }
}