test/Jpom
1
0
Fork 0
mirror of https://gitee.com/dromara/Jpom.git synced 2024-12-02 20:08:40 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

add H2BackupService class methods

Browse Source
This commit is contained in:
Eleven 2021-11-18 08:56:34 +08:00
parent cd0116c13a
commit 7b8be868d7
1 changed files with 21 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
modules/server/src/main/java/io/jpom/service/h2db/H2BackupService.java
View File

@ -3,12 +3,15 @@ package io.jpom.service.h2db;
import cn.hutool.core.util.StrUtil; import cn.hutool.core.util.StrUtil;
import cn.hutool.db.ds.DSFactory; import cn.hutool.db.ds.DSFactory;
import cn.jiangzeyin.common.DefaultSystemLog; import cn.jiangzeyin.common.DefaultSystemLog;
import org.h2.tools.RunScript;
import org.h2.tools.Shell; import org.h2.tools.Shell;
import org.springframework.stereotype.Service; import org.springframework.stereotype.Service;
import org.springframework.util.Assert; import org.springframework.util.Assert;
import org.springframework.util.CollectionUtils; import org.springframework.util.CollectionUtils;
import javax.sql.DataSource; import javax.sql.DataSource;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.sql.Connection; import java.sql.Connection;
import java.sql.SQLException; import java.sql.SQLException;
import java.util.List; import java.util.List;
@ -70,5 +73,22 @@ public class H2BackupService {
shell.runTool(connection, params); shell.runTool(connection, params);
} }
// 还原备份 SQL /**
* 还原备份 SQL
* @param backupSqlPath backup SQL file path, absolute path
* @throws SQLException
* @throws FileNotFoundException
*/
public void restoreBackupSql(String backupSqlPath) throws SQLException, FileNotFoundException {
// 读取数据库备份文件
FileReader fileReader = new FileReader(backupSqlPath);
// 加载数据源
DataSource dataSource = DSFactory.get();
Assert.notNull(dataSource, "Restore Backup sql error...H2 DataSource not null");
Connection connection = dataSource.getConnection();
// 执行还原
RunScript.execute(connection, fileReader);
}
} }