parent
37fba96fc5
commit
f16d376674
@ -0,0 +1,59 @@
|
||||
package com.dsideal.YunXiaoTools.Utils;
|
||||
|
||||
import com.jfinal.kit.PropKit;
|
||||
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
public class CommonUtil {
|
||||
|
||||
/**
|
||||
* 获取mysqldump路径
|
||||
*/
|
||||
public static String getMySQLDumpPath() {
|
||||
try {
|
||||
String resourcePath = PropKit.get("mysqldump_path");
|
||||
// 处理路径中的空格和特殊字符
|
||||
if (resourcePath.startsWith("/")) {
|
||||
resourcePath = resourcePath.substring(1);
|
||||
}
|
||||
// URL解码
|
||||
return java.net.URLDecoder.decode(resourcePath, "UTF-8");
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("获取mysqldump路径失败", e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 从JDBC URL中提取主机地址
|
||||
*/
|
||||
public static String getHostFromJdbcUrl(String jdbcUrl) {
|
||||
String[] parts = jdbcUrl.split("://")[1].split(":");
|
||||
return parts[0];
|
||||
}
|
||||
|
||||
/**
|
||||
* 从JDBC URL中提取端口号
|
||||
*/
|
||||
public static String getPortFromJdbcUrl(String jdbcUrl) {
|
||||
String[] parts = jdbcUrl.split("://")[1].split(":");
|
||||
return parts[1].split("/")[0];
|
||||
}
|
||||
|
||||
/**
|
||||
* 从JDBC URL中提取数据库名
|
||||
*/
|
||||
public static String getDatabaseFromJdbcUrl(String jdbcUrl) {
|
||||
try {
|
||||
// 使用正则表达式匹配数据库名
|
||||
Pattern pattern = Pattern.compile("jdbc:mysql://[^/]+/(\\w+)\\??.*");
|
||||
Matcher matcher = pattern.matcher(jdbcUrl);
|
||||
if (matcher.find()) {
|
||||
return matcher.group(1);
|
||||
}
|
||||
throw new RuntimeException("无法从JDBC URL中提取数据库名: " + jdbcUrl);
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("解析数据库名失败: " + jdbcUrl, e);
|
||||
}
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,185 @@
|
||||
package com.dsideal.YunXiaoTools.Utils;
|
||||
|
||||
import com.jfinal.kit.PropKit;
|
||||
import com.obs.services.ObsClient;
|
||||
import java.io.*;
|
||||
import java.util.zip.*;
|
||||
|
||||
public class DatabaseRestore {
|
||||
private final String obsEndpoint;
|
||||
private final String obsAccessKey;
|
||||
private final String obsSecretKey;
|
||||
private final String obsBucket;
|
||||
|
||||
// 目标数据库配置
|
||||
private final String dbHost;
|
||||
private final String dbPort;
|
||||
private final String dbUser;
|
||||
private final String dbPassword;
|
||||
private final String dbName;
|
||||
|
||||
public DatabaseRestore() {
|
||||
// 从配置文件读取配置
|
||||
this.obsEndpoint = PropKit.get("obs.endPoint");
|
||||
this.obsAccessKey = PropKit.get("obs.accessKey");
|
||||
this.obsSecretKey = PropKit.get("obs.secretKey");
|
||||
this.obsBucket = PropKit.get("obs.bucketName");
|
||||
|
||||
// 目标数据库配置
|
||||
this.dbHost = CommonUtil.getHostFromJdbcUrl(PropKit.get("write.jdbcUrl"));
|
||||
this.dbPort = CommonUtil.getPortFromJdbcUrl(PropKit.get("write.jdbcUrl"));
|
||||
this.dbUser = PropKit.get("write.user");
|
||||
this.dbPassword = PropKit.get("write.password");
|
||||
this.dbName = CommonUtil.getDatabaseFromJdbcUrl(PropKit.get("write.jdbcUrl"));
|
||||
}
|
||||
|
||||
/**
|
||||
* 执行完整的还原流程
|
||||
* @param obsKey OBS上的文件路径
|
||||
*/
|
||||
public void restore(String obsKey) {
|
||||
String tempDir = System.getProperty("java.io.tmpdir");
|
||||
String zipFile = tempDir + File.separator + new File(obsKey).getName();
|
||||
String sqlFile = null;
|
||||
|
||||
try {
|
||||
// 1. 从OBS下载ZIP文件
|
||||
downloadFromObs(obsKey, zipFile);
|
||||
System.out.println("文件下载完成: " + zipFile);
|
||||
|
||||
// 2. 解压ZIP文件
|
||||
sqlFile = unzipFile(zipFile);
|
||||
System.out.println("文件解压完成: " + sqlFile);
|
||||
|
||||
// 3. 还原数据库
|
||||
restoreDatabase(sqlFile);
|
||||
System.out.println("数据库还原完成");
|
||||
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("还原过程失败: " + e.getMessage(), e);
|
||||
} finally {
|
||||
// 清理临时文件
|
||||
if (zipFile != null) {
|
||||
new File(zipFile).delete();
|
||||
}
|
||||
if (sqlFile != null) {
|
||||
new File(sqlFile).delete();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 从OBS下载文件
|
||||
*/
|
||||
private void downloadFromObs(String obsKey, String localFile) {
|
||||
try (ObsClient obsClient = new ObsClient(obsAccessKey, obsSecretKey, obsEndpoint)) {
|
||||
obsClient.getObject(obsBucket, obsKey, String.valueOf(new File(localFile)));
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("从OBS下载文件失败: " + e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 解压ZIP文件
|
||||
*/
|
||||
private String unzipFile(String zipFile) throws IOException {
|
||||
String extractedFile = null;
|
||||
|
||||
try (ZipInputStream zis = new ZipInputStream(new FileInputStream(zipFile))) {
|
||||
ZipEntry zipEntry = zis.getNextEntry();
|
||||
if (zipEntry != null) {
|
||||
extractedFile = new File(zipFile).getParent() + File.separator + zipEntry.getName();
|
||||
|
||||
try (BufferedOutputStream bos = new BufferedOutputStream(
|
||||
new FileOutputStream(extractedFile))) {
|
||||
byte[] buffer = new byte[8192];
|
||||
int len;
|
||||
while ((len = zis.read(buffer)) > 0) {
|
||||
bos.write(buffer, 0, len);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if (extractedFile == null) {
|
||||
throw new RuntimeException("ZIP文件中没有找到SQL文件");
|
||||
}
|
||||
|
||||
return extractedFile;
|
||||
}
|
||||
|
||||
/**
|
||||
* 还原数据库
|
||||
*/
|
||||
private void restoreDatabase(String sqlFile) {
|
||||
try {
|
||||
// 获取mysql命令路径
|
||||
String mysqlPath = getMySQLPath();
|
||||
|
||||
// 构建还原命令
|
||||
ProcessBuilder pb = new ProcessBuilder(
|
||||
mysqlPath,
|
||||
"-h" + dbHost,
|
||||
"-P" + dbPort,
|
||||
"-u" + dbUser,
|
||||
"--ssl-mode=DISABLED",
|
||||
dbName
|
||||
);
|
||||
|
||||
// 设置环境变量传递密码
|
||||
pb.environment().put("MYSQL_PWD", dbPassword);
|
||||
|
||||
// 执行还原命令
|
||||
Process process = pb.start();
|
||||
|
||||
// 将SQL文件内容写入mysql进程的输入流
|
||||
try (BufferedInputStream bis = new BufferedInputStream(new FileInputStream(sqlFile));
|
||||
BufferedOutputStream bos = new BufferedOutputStream(process.getOutputStream())) {
|
||||
|
||||
byte[] buffer = new byte[8192];
|
||||
int len;
|
||||
while ((len = bis.read(buffer)) > 0) {
|
||||
bos.write(buffer, 0, len);
|
||||
}
|
||||
}
|
||||
|
||||
// 读取错误输出
|
||||
StringBuilder errorOutput = new StringBuilder();
|
||||
try (BufferedReader reader = new BufferedReader(
|
||||
new InputStreamReader(process.getErrorStream()))) {
|
||||
String line;
|
||||
while ((line = reader.readLine()) != null) {
|
||||
if (!line.contains("WARNING")) {
|
||||
errorOutput.append(line).append("\n");
|
||||
System.out.println("还原进度: " + line);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// 等待命令执行完成
|
||||
int exitCode = process.waitFor();
|
||||
if (exitCode != 0) {
|
||||
throw new RuntimeException("数据库还原失败：\n" + errorOutput.toString());
|
||||
}
|
||||
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("还原数据库失败: " + e.getMessage(), e);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 获取mysql命令路径
|
||||
*/
|
||||
private String getMySQLPath() {
|
||||
try {
|
||||
String resourcePath = getClass().getClassLoader()
|
||||
.getResource("mysql.exe").getPath();
|
||||
if (resourcePath.startsWith("/")) {
|
||||
resourcePath = resourcePath.substring(1);
|
||||
}
|
||||
return java.net.URLDecoder.decode(resourcePath, "UTF-8");
|
||||
} catch (Exception e) {
|
||||
throw new RuntimeException("获取mysql路径失败", e);
|
||||
}
|
||||
}
|
||||
}
|
||||
Loading…
Reference in new issue