1. 程式人生 > 其它 >springboot初始化執行mysql指令碼修改庫表結構

springboot初始化執行mysql指令碼修改庫表結構

專案啟動會執行指令碼,修改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 ^^^