通过flyway实现项目启动自动建库建表

前言

我之前研究过一个开源论坛项目，pybbs，他代码下载下来直接启动会自动建库建表，所以我拿过来研究一下（不难），这样以后方便开发萌新部署项目。

代码实现

添加依赖

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

将数据库导出成sql

然后在springBoot静态资源路径下建立db/migration（一定要看图）

sql文件命名方式也有讲究

sql 文件的命名具有一定规则，以V开头，接着两个下划线 __，接着可以写注释，然后以 .sql 结尾，如V3__alter_table.sql 版本号支持小版本x.y.z格式，但是为了简单起见，直接用一个数字递增更方便。

代码配置

import org.flywaydb.core.Flyway;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.DependsOn;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import javax.sql.DataSource;

/**
 * Created by tomoya.
 * Copyright (c) 2018, All Rights Reserved.
 * https://yiiu.co
 */
@Configuration
public class FlywayConfig {

    @Resource
    private DataSource dataSource;

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

}

启动建库

使用flyway，在启动的时候如果没有表会自动建表

package co.yiiu.pybbs.config;

import co.yiiu.pybbs.util.SpringContextUtil;
import org.springframework.context.annotation.Configuration;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;
import java.net.URI;
import java.net.URISyntaxException;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.sql.Statement;

/**
 * Created by tomoya.
 * Copyright (c) 2018, All Rights Reserved.
 * https://yiiu.co
 */
@Configuration
public class DataSourceHelper {

    @Resource
    private DataSourceProperties dataSourceProperties;

    @PostConstruct
    public void init() {
        if (siteConfig == null) siteConfig = SpringContextUtil.getBean(SiteConfig.class);
        try {
            Class.forName(dataSourceProperties.getDriverClassName());
            URI uri = new URI(dataSourceProperties.getUrl().replace("jdbc:", ""));
            String host = uri.getHost();
            int port = uri.getPort();
            String path = uri.getPath();
            String query = uri.getQuery();
            Connection connection = DriverManager.getConnection("jdbc:mysql://" + host + ":" + port + "?" + query,
                    dataSourceProperties.getUsername(), ataSourceProperties.getPassword());
            Statement statement = connection.createStatement();
            statement.executeUpdate("CREATE DATABASE IF NOT EXISTS `" + path.replace("/", "") + "` DEFAULT CHARACTER SET = "
                    + "" + "`utf8` COLLATE `utf8_general_ci`;");
            statement.close();
            connection.close();
        } catch (URISyntaxException | ClassNotFoundException | SQLException e) {
            e.printStackTrace();
            System.exit(0);
        }
    }
}

后面正常启动即可，但是基本代码，只做记录，不做叙述
如有问题可加论坛群：1048094312