前言
我之前研究过一个开源论坛项目,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