159 lines
4.2 KiB
C++
Executable File
159 lines
4.2 KiB
C++
Executable File
#include "udiskdatarestoreproxy.h"
|
||
#include <QDateTime>
|
||
#include <QDir>
|
||
#include <QTimer>
|
||
#include <QDebug>
|
||
#include <unistd.h>
|
||
#include <sys/reboot.h>
|
||
#include "../common/utils.h"
|
||
#include "mymountproxy.h"
|
||
|
||
IMPLEMENT_DYNCREATE(UDiskDataRestoreProxy)
|
||
|
||
/**
|
||
* @brief 构造函数
|
||
*/
|
||
UDiskDataRestoreProxy::UDiskDataRestoreProxy()
|
||
{
|
||
m_bSuccess = false;
|
||
m_isFinished = false;
|
||
m_p = nullptr;
|
||
}
|
||
|
||
/**
|
||
* @brief 析构函数
|
||
*/
|
||
UDiskDataRestoreProxy::~UDiskDataRestoreProxy()
|
||
{
|
||
delete m_p;
|
||
}
|
||
|
||
/**
|
||
* @brief 环境检测
|
||
* @return false,检测失败;true,检测成功
|
||
*/
|
||
bool UDiskDataRestoreProxy::checkEnvEx()
|
||
{
|
||
qDebug() << "UDiskDataRestoreProxy::checkEnvEx invoke begin";
|
||
|
||
// 1、检测.user.txt是否存在
|
||
m_userFile = m_backupWrapper.m_prefixDestPath + BACKUP_SNAPSHOTS_PATH + "/" + m_backupWrapper.m_uuid + "/" + PATHS_USER_FILE;
|
||
if (!Utils::filsExists(m_userFile)) {
|
||
qCritical(".user.txt文件不存在");
|
||
emit checkResult(int(BackupResult::WRITE_BACKUP_PATHS_TO_USER_FAILED));
|
||
return false;
|
||
}
|
||
|
||
// 2、检测.exclude.user.txt是否存在
|
||
m_excludeUserFile = m_backupWrapper.m_prefixDestPath + BACKUP_SNAPSHOTS_PATH + "/" + m_backupWrapper.m_uuid + "/" + EXCLUDE_PATHS_USER_FILE;
|
||
if (!Utils::filsExists(m_excludeUserFile)) {
|
||
qCritical(".exclude.user.txt文件不存在");
|
||
emit checkResult(int(BackupResult::WRITE_EXCLUDE_BACKUP_PATHS_TO_USER_FAILED));
|
||
return false;
|
||
}
|
||
|
||
// 3、检测还原点是否存在
|
||
m_backupPath = m_backupWrapper.m_prefixDestPath + BACKUP_SNAPSHOTS_PATH + "/" + m_backupWrapper.m_uuid + "/data";
|
||
if (Utils::isDirEmpty(m_backupPath)) {
|
||
qCritical("还原点{uuid}/data目录不存在");
|
||
emit checkResult(int(BackupResult::INC_NOT_FOUND_DIR));
|
||
return false;
|
||
}
|
||
|
||
emit checkResult(int(BackupResult::CHECK_ENV_SUCCESS));
|
||
|
||
qDebug() << "UDiskDataRestoreProxy::checkEnvEx invoke end";
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* @brief 执行还原逻辑
|
||
*/
|
||
void UDiskDataRestoreProxy::doWorkEx()
|
||
{
|
||
qDebug() << "UDiskDataRestoreProxy::doWorkEx invoke begin";
|
||
|
||
// 1、校验
|
||
if (!checkEnvEx())
|
||
return ;
|
||
|
||
// 2、还原数据
|
||
restoreData();
|
||
|
||
qDebug() << "UDiskDataRestoreProxy::doWorkEx invoke end";
|
||
}
|
||
|
||
/**
|
||
* @brief 根据场景获取rsync命令参数
|
||
* @param scene，场景
|
||
* @return 组装好的rsync的参数信息
|
||
*/
|
||
QStringList UDiskDataRestoreProxy::getRsyncArgs(DataRestoreScene scene)
|
||
{
|
||
QStringList args;
|
||
args << "-avAHXr";
|
||
args << "--info=progress2";
|
||
args << "--no-inc-recursive";
|
||
args << "--ignore-missing-args";
|
||
args << "--delete";
|
||
|
||
switch (scene) {
|
||
case DataRestoreScene::DATA_RESTORE :
|
||
args << "--files-from" << m_userFile;
|
||
|
||
break ;
|
||
default:
|
||
return args;
|
||
}
|
||
|
||
return args;
|
||
}
|
||
|
||
/**
|
||
* @brief 数据还原
|
||
*/
|
||
void UDiskDataRestoreProxy::restoreData()
|
||
{
|
||
QStringList args = getRsyncArgs(DataRestoreScene::DATA_RESTORE);
|
||
|
||
m_srcPath = m_backupPath;
|
||
QString destPath = Utils::getSysRootPath();
|
||
destPath.replace("//", "/");
|
||
args << m_srcPath + "/";
|
||
args << destPath + "/";
|
||
|
||
m_p = new RsyncPathToDirProcess(this);
|
||
connect(m_p, &RsyncPathToDirProcess::progress, this, &UDiskDataRestoreProxy::progress);
|
||
connect(m_p, &RsyncPathToDirProcess::finished, this, [&](bool result) {
|
||
m_bSuccess = result;
|
||
m_isFinished = true;
|
||
if (result) {
|
||
QString time = QDateTime::currentDateTime().toString("yy-MM-dd hh:mm:ss");
|
||
Utils::writeBackupLog(time + "," + m_backupWrapper.m_uuid + "," + QString::number(m_backupWrapper.m_type) + ",,," + QString::number(m_backupWrapper.m_frontUid)+ "," + m_backupWrapper.m_backupName);
|
||
}
|
||
emit this->workResult(result);
|
||
});
|
||
|
||
QTimer::singleShot(1*1000, this, &UDiskDataRestoreProxy::checkUdiskExists);
|
||
m_isFinished = false;
|
||
m_bSuccess = false;
|
||
m_p->start(args, false);
|
||
}
|
||
|
||
/**
|
||
* @brief 监控移动设备是否还在
|
||
* @return true-在；false-不在
|
||
*/
|
||
bool UDiskDataRestoreProxy::checkUdiskExists()
|
||
{
|
||
if (!m_isFinished) {
|
||
if (Utils::isDirEmpty(m_backupPath) && m_p != nullptr)
|
||
m_p->stop();
|
||
else
|
||
QTimer::singleShot(1*1000, this, &UDiskDataRestoreProxy::checkUdiskExists);
|
||
}
|
||
|
||
return true;
|
||
}
|
||
|