springboot初始化執行mysql指令碼修改庫表結構
阿新 • • 發佈:2021-09-01
專案啟動會執行指令碼,修改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 ^^^