项目启动会执行脚本,修改mysql库表的结构、数据
1.项目每次重启都会执行脚本,修改,添加、删除表、表结构会报错。
2.数据重复新增会报错。
3.数据重复修改导致准确性问题。
解决方案:
1.通过版本来控制,检查数据库中版本和程序中版本是否一致,数据库中版本低时则依次更新每个版本的脚本,最后将数据库中版本号更新。
2.在脚本中刷入存储过程,存储过程中对每一类操作进行定义。给出操作类型和表参数, 检查参数sql是否可执行,然后拼接 sql进行执行。
说明:
方案1的业务逻辑有点复杂。 取resource,解析版本号,正确拆解sql,读取数据库版本后比对……需要一整套的逻辑处理。采用方案2.
实现:
1.借助springboot 的启动执行sql --DataSourceInitializer
@Component public class CustomizeDataSourceInitializer { @Value ("classpath:sql/schema.sql") private Resource dataScript; @Autowired DataSource dataSource; @Bean public DataSourceInitializer dataSourceInitializer( ) { final DataSourceInitializer initializer = new DataSourceInitializer(); // 设置数据源 initializer.setDataSource(dataSource); initializer.setDatabasePopulator(databasePopulator()); return initializer; } private DatabasePopulator databasePopulator() { final ResourceDatabasePopulator populator = new ResourceDatabasePopulator();
//此处修改了分隔符 populator.setSeparator( ScriptUtils.EOF_STATEMENT_SEPARATOR ); populator.addScripts(dataScript); System.out.println("==================sql脚本初始化完成=================="); return populator; } }
2.resource添加脚本sql/schema.sql
------------------------- 中文无法执行 ---------------------------------- DROP PROCEDURE IF EXISTS Pro_Temp_ColumnWork; ^^^ END OF SCRIPT ^^^ -- 1表示新增列,2表示修改列类型,3表示删除列 CREATE PROCEDURE Pro_Temp_ColumnWork(TableName VARCHAR(50),ColumnName VARCHAR(50),SqlStr VARCHAR(4000),CType INT) BEGIN DECLARE Rows1 INT ; SET Rows1=0 ; SELECT COUNT(*) INTO Rows1 FROM INFORMATION_SCHEMA.Columns WHERE table_schema= DATABASE() AND table_name=TableName AND column_name=ColumnName ; -- 新增列 IF (CType=1 AND Rows1<=0) THEN SET SqlStr := CONCAT( 'ALTER TABLE ',TableName,' ADD COLUMN ',ColumnName,' ',SqlStr) ; -- 修改列类型 ELSEIF (CType=2 AND Rows1>0) THEN SET SqlStr := CONCAT('ALTER TABLE ',TableName,' MODIFY ',ColumnName,' ',SqlStr) ; -- 删除列 ELSEIF (CType=3 AND Rows1>0) THEN SET SqlStr := CONCAT('ALTER TABLE ',TableName,' DROP COLUMN ',ColumnName) ; ELSE SET SqlStr :='' ; END IF ; -- 执行命令 IF (SqlStr<>'') THEN SET @SQL1 = SqlStr ; PREPARE stmt1 FROM @SQL1 ; EXECUTE stmt1 ; END IF ; END; ^^^ END OF SCRIPT ^^^ -- alter table gd_device_group modify column device_group_description varchar(255) COMMENT '设备分组描述'; call Pro_Temp_ColumnWork("table_name","colum_name","varchar(255) COMMENT ' testcaaa '",2) ^^^ END OF SCRIPT ^^^