转载

Spring-Boot项目集成Flyway和MybatisPlus执行先后问题解决办法

使用mybatis-plus很爽，但没法自动创建数据库和表结构或者一些初始化数据，所以我在自己项目里加上了flyway(数据库迁移工具)，还有一个类似的工具，liquibase,只不过后者用的是xml配置的，flyway直接执行的是sql，相比之下我更喜欢flyway

但集成flyway跟mybatis-plus却出现了问题，执行先后的问题，下面具体说明

引入依赖

不用指定版本号，springboot已经内置了

<dependency>
  <groupId>org.flywaydb</groupId>
  <artifactId>flyway-core</artifactId>
</dependency>

flyway的配置

前言：如果程序里在启动的过程中不使用mybatis-plus去操作数据库，那就直接配置即可，默认flyway就是提前执行的

spring:
  flyway:
    locations: ["classpath:db/migration"]
    baseline-on-migrate: true

flyway的sql文件格式网上很多介绍，这里不多说，在 src/main/resources/ 下创建文件夹 db 然后在 db 下再创建文件夹 migration ，最后将sql文件放在 migration 里就可以了

启动项目，可以看到数据库里会多一个表 flyway_schema_history 这就算成功了

但是！！！我在程序里把大部分配置都放到数据库里了，也就是说程序启动有些配置要先去查数据库，拿到数据后，再继续配置相关的服务，比如 shiro redis 等初始化工作，这时候就出问题了，flyway还没有执行，数据库里还没有数据，就开始配置其它的一些服务了，到数据库里查不到数据，配置自然就报错了

创建Flyway配置类

一样的，不用springboot内置的flyway配置类了，自己创建一个，就可以指定它在哪个阶段执行了

flyway的配置类非常简单，代码如下

@Configuration
public class FlywayConfig {

  @Autowired
  private DataSource dataSource;

  @PostConstruct
  public void migrate() {
    Flyway flyway = Flyway.configure()
        .dataSource(dataSource)
        .locations("db/migration")
        .baselineOnMigrate(true)
        .load();
    flyway.migrate();
  }

}

有了这个，还要把springboot默认的flyway初始化类给去掉，配置如下

// 不用默认配置的Flyway，自己配置
@SpringBootApplication(exclude = {
    FlywayAutoConfiguration.class
})
public class PybbsApplication {

  public static void main(String[] args) {
    SpringApplication.run(PybbsApplication.class, args);
  }
}

另外其它一些初始化服务的类，比如 shiroConfig 的配置类，就可以放在 mybatisPlusConfig 后再执行，而 mybatisPlusConfig 可以指定在 flywayConfig 初始化后再初始化，具体配置如下

MyBatisPlusConfig

@Configuration
@MapperScan("co.yiiu.pybbs.mapper")
public class MybatisPlusConfig {

  @Bean("mybatisSqlSession")
  @DependsOn("flywayConfig") // 这里指定依赖于 flywayConfig 类
  public SqlSessionFactory sqlSessionFactory(@Qualifier("dataSource") DataSource dataSource) throws Exception {
    MybatisSqlSessionFactoryBean sqlSessionFactory = new MybatisSqlSessionFactoryBean();
    MybatisConfiguration configuration = new MybatisConfiguration();
    configuration.setDefaultScriptingLanguage(MybatisXMLLanguageDriver.class);
    configuration.setJdbcTypeForNull(JdbcType.NULL);
    //*注册Map 下划线转驼峰*
    configuration.setObjectWrapperFactory(new MybatisMapWrapperFactory());

    // 添加数据源，官方文档里没有下面这一行，启动一直报错，脑瓜疼。。。
    sqlSessionFactory.setDataSource(dataSource);

    sqlSessionFactory.setConfiguration(configuration);

    // 添加分页插件，这里有个坑
    // 在没有配置Map 下划线转驼峰这个配置之前，配置了上面注释掉的分页插件配置，结果配置了这个之后，分页就失效了
    // 找了一圈，无解，只好自己尝试解决，这个配置里就用了两个类，MybatisSqlSessionFactoryBean，MybatisConfiguration
    // 先看一下分页插件是实现的哪个接口，没错就是这货 Interceptor
    // 逐个去上面两个配置类里找，还真在MybatisSqlSessionFactoryBean类里找到了，有个plugins的字段，类型正是 Interceptor
    // 然后尝试把分页插件设置进去，启动项目，问题解决
    // 官方文档太简陋，坑新手呀！！！！！！
    PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
    sqlSessionFactory.setPlugins(new Interceptor[]{paginationInterceptor});

    return sqlSessionFactory.getObject();
  }
}

ShiroConfig

@Configuration
public class ShiroConfig {
  // ... 其它一些配置 ...

  // 配置记住我功能
  @Bean
  @DependsOn("mybatisPlusConfig") // 这个方法要用mybatisPlus查数据库拿数据，就在这个方法上配置上依赖于 mybatisPlusConfig 类
  public SimpleCookie rememberMeCookie() {
    //这个参数是cookie的名称，对应前端的checkbox的name = rememberMe
    SimpleCookie simpleCookie = new SimpleCookie("rememberMe");
    // 记住我cookie生效时间 单位秒
    int adminRememberMeMaxAge = Integer.parseInt(systemConfigService.selectAllConfig().get("admin_remember_me_max_age").toString());
    simpleCookie.setMaxAge(adminRememberMeMaxAge * 24 * 60 * 60);
    return simpleCookie;
  }

  @Bean
  @DependsOn("mybatisPlusConfig") // 这个方法要用mybatisPlus查数据库拿数据，就在这个方法上配置上依赖于 mybatisPlusConfig 类
  public CookieRememberMeManager rememberMeManager() {
    //System.out.println("ShiroConfiguration.rememberMeManager()");
    CookieRememberMeManager cookieRememberMeManager = new CookieRememberMeManager();
    cookieRememberMeManager.setCookie(rememberMeCookie());
    //rememberMe cookie加密的密钥 建议每个项目都不一样 默认AES算法 密钥长度(128 256 512 位)
    cookieRememberMeManager.setCipherKey(Base64.encode("pybbs is the best!".getBytes()));
    return cookieRememberMeManager;
  }
}

从上面代码可以看出，哪个 bean 有依赖于其它配置类，就在哪个 bean 上配置上 @DependsOn() 即可，完全自己掌控，爽歪歪