Reorganize codes

Reorganize all parts of codes; Fix a bug that index page color is not correct when initialized.
2025-12-14 21:02:04 +08:00 · 2021-04-17 15:43:11 +08:00
parent 6fd3c40e97
commit e64e7fcae2
19 changed files with 525 additions and 410 deletions
--- a/src/downloadworker.cpp
+++ b/src/downloadworker.cpp
@@ -1,10 +1,9 @@
 #include "downloadworker.h"
-#include <QIODevice>
+
 #include <QEventLoop>
 #include <QNetworkAccessManager>
 #include <QNetworkRequest>
 #include <QNetworkReply>
-#include <QDebug>
 #include <QThread>
 #include <QRegularExpression>
 #include <QFileInfo>
@@ -12,7 +11,7 @@

 DownloadWorker::DownloadWorker(QObject *parent)
 {
-
+    Q_UNUSED(parent)
 }

 void DownloadWorker::setIdentifier(int identifier)
@@ -35,28 +34,28 @@ qint64 DownloadWorker::getReceivedPos()

 void DownloadWorker::doWork()
 {
-        mgr = new QNetworkAccessManager(this);
-        QNetworkRequest request;
-        request.setUrl(url);
-        request.setAttribute(QNetworkRequest::FollowRedirectsAttribute, true);
-        request.setRawHeader("Range", QString("bytes=%1-%2").arg(startPos)
-                    .arg(endPos).toLocal8Bit());
-        reply = mgr->get(request);
-        qDebug() << "开始下载数据：" << QString(" %1~%2 -> writePos Start %3")
-            .arg(startPos).arg(endPos).arg(receivedPos);
-        connect(reply, static_cast<void(QNetworkReply::*)(QNetworkReply::NetworkError)>(&QNetworkReply::error),
-        [this](QNetworkReply::NetworkError error){
-            if (error != QNetworkReply::NoError) {
-                qDebug() << "出错了：" << reply->errorString();
-            }
-        });
-        connect(reply, &QNetworkReply::finished, mgr, &QNetworkAccessManager::deleteLater);
-        connect(reply, &QNetworkReply::readyRead, this, &DownloadWorker::dataReady);
-        connect(reply, &QNetworkReply::finished, this, &DownloadWorker::slotFinish);
-        connect(reply, &QNetworkReply::downloadProgress, this, &DownloadWorker::handleProcess);
-//        connect(reply, &QNetworkReply::finished, reply, &QNetworkReply::deleteLater);
-        connect(reply, &QNetworkReply::finished, this, &DownloadWorker::doStop);
+    mgr = new QNetworkAccessManager(this);
+    QNetworkRequest request;
+    request.setUrl(url);
+    request.setAttribute(QNetworkRequest::FollowRedirectsAttribute, true);
+    request.setRawHeader("Range", QString("bytes=%1-%2").arg(startPos).arg(endPos).toLocal8Bit());
+    reply = mgr->get(request);
+    qDebug() << "开始下载数据：" << QString(" %1~%2 -> writePos Start %3").arg(startPos).arg(endPos).arg(receivedPos);

+    connect(reply, static_cast<void(QNetworkReply::*)(QNetworkReply::NetworkError) > (&QNetworkReply::error),
+            [this](QNetworkReply::NetworkError error)
+    {
+        if(error != QNetworkReply::NoError)
+        {
+            qDebug() << "出错了：" << reply->errorString();
+        }
+    });
+    connect(reply, &QNetworkReply::finished, mgr, &QNetworkAccessManager::deleteLater);
+    connect(reply, &QNetworkReply::readyRead, this, &DownloadWorker::dataReady);
+    connect(reply, &QNetworkReply::finished, this, &DownloadWorker::slotFinish);
+    connect(reply, &QNetworkReply::downloadProgress, this, &DownloadWorker::handleProcess);
+    // connect(reply, &QNetworkReply::finished, reply, &QNetworkReply::deleteLater);
+    connect(reply, &QNetworkReply::finished, this, &DownloadWorker::doStop);
 }

 void DownloadWorker::doStop()
@@ -80,8 +79,7 @@ void DownloadWorker::dataReady()
 void DownloadWorker::slotFinish()
 {
    file->flush();
-    qDebug() << "数据块下载完毕：" << QString(" %1~%2 -> writePos Start %3")
-        .arg(startPos).arg(endPos).arg(receivedPos);
+    qDebug() << "数据块下载完毕：" << QString(" %1~%2 -> writePos Start %3").arg(startPos).arg(endPos).arg(receivedPos);
    emit workFinished();
 }

@@ -93,6 +91,8 @@ void DownloadWorker::handleProcess(qint64, qint64)

 DownloadController::DownloadController(QObject *parent)
 {
+    Q_UNUSED(parent)
+
    domains = {
        "d1.store.deepinos.org.cn",
        "d2.store.deepinos.org.cn",
@@ -100,13 +100,15 @@ DownloadController::DownloadController(QObject *parent)
        "d4.store.deepinos.org.cn",
        "d5.store.deepinos.org.cn"
    };
-    this->threadNum =  domains.size() > 5 ? 5 : domains.size();
+    this->threadNum = domains.size() > 5 ? 5 : domains.size();
 }

 DownloadController::~DownloadController()
 {
-    if (workers.size() > 0) {
-        for(int i = 0; i < workers.size(); i++) {
+    if(workers.size() > 0)
+    {
+        for(int i = 0; i < workers.size(); i++)
+        {
            workers.at(i)->doStop();
            workers.at(i)->disconnect();
            workers.at(i)->deleteLater();
@@ -130,45 +132,52 @@ void DownloadController::setThreadNum(int threadNum)
 */
 void DownloadController::startDownload(const QString &url)
 {
-
-
    finish = 0;

    // 下载任务等分，计算每个线程的下载数据
    fileSize = getFileSize(url);
-    if (fileSize == 0) {
+    if(fileSize == 0)
+    {
        emit errorOccur("文件大小获取失败");
        return; 
    }
+
    qint64 segmentSize = fileSize / threadNum;
    ranges.resize(threadNum);
    QVector<qint64> receivedBytes;
    receivedBytes.resize(threadNum);
-    for (int i = 0; i < threadNum; i++) {
+    for(int i = 0; i < threadNum; i++)
+    {
        ranges[i].first = i * segmentSize;
        ranges[i].second = i * segmentSize + segmentSize - 1;
        receivedBytes[i] = 0;
    }
-    ranges[threadNum-1].second = fileSize; // 余数部分加入最后一个
+    ranges[threadNum - 1].second = fileSize; // 余数部分加入最后一个

    // 打开文件
    QDir tmpdir("/tmp/spark-store");
    file = new QFile;
    file->setFileName(tmpdir.absoluteFilePath(filename));

-    if (file->exists())
+    if(file->exists())
+    {
        file->remove();
-    if (!file->open(QIODevice::WriteOnly)) {
+    }
+
+    if(!file->open(QIODevice::WriteOnly))
+    {
        delete file;
        file = nullptr;
        emit errorOccur(file->errorString());
        return;
    }
+
    file->resize(fileSize);

    // 创建下载线程
    workers.clear();
-    for(int i = 0; i < ranges.size(); i++) {
+    for(int i = 0; i < ranges.size(); i++)
+    {
        qDebug() << QString("第%1个下载请求：%2-%3").arg(i).arg(ranges.at(i).first).arg(ranges.at(i).second);
        auto worker = new DownloadWorker(this);
        auto range = ranges.at(i);
@@ -176,8 +185,10 @@ void DownloadController::startDownload(const QString &url)
        worker->setIdentifier(i);
        worker->setParamter(chunkUrl, range, file);
        workers.append(worker);
+
        connect(worker, &DownloadWorker::downloadProcess, this, &DownloadController::handleProcess);
        connect(worker, &DownloadWorker::workFinished, this, &DownloadController::chunkDownloadFinish);
+
        worker->doWork();
    }
 }
@@ -187,14 +198,15 @@ void DownloadController::startDownload(const QString &url)
 */
 void DownloadController::stopDownload()
 {
-    for(int i = 0; i < workers.size(); i++) {
+    for(int i = 0; i < workers.size(); i++)
+    {
        workers.at(i)->doStop();
        workers.at(i)->disconnect();
        workers.at(i)->deleteLater();
    }
    workers.clear();
-
    qDebug() << "文件下载路径：" << QFileInfo(file->fileName()).absoluteFilePath();
+
    file->flush();
    file->close();
    delete file;
@@ -205,7 +217,8 @@ void DownloadController::stopDownload()
 void DownloadController::handleProcess()
 {
    qint64 bytesReceived = 0;
-    for(int i = 0; i < workers.size(); i++) {
+    for(int i = 0; i < workers.size(); i++)
+    {
        bytesReceived += workers.at(i)->getReceivedPos();
    }
    qDebug() << QString("下载进度 %1-%2").arg(bytesReceived).arg(fileSize);
@@ -216,7 +229,8 @@ void DownloadController::chunkDownloadFinish()
 {
    finish++;
    qDebug() << QString("已下载了%1块，共%2块！！！").arg(finish).arg(threadNum);
-    if (finish == threadNum) {
+    if(finish == threadNum)
+    {
        stopDownload();
        emit downloadFinished();
    }
@@ -230,17 +244,21 @@ qint64 DownloadController::getFileSize(const QString& url)
    request.setUrl(QUrl(url));
    request.setAttribute(QNetworkRequest::FollowRedirectsAttribute, true);
    QNetworkReply *reply = requestManager.head(request);
-    connect(reply, static_cast<void(QNetworkReply::*)(QNetworkReply::NetworkError)>(&QNetworkReply::error),
-    [this, reply](QNetworkReply::NetworkError error){
-        if (error != QNetworkReply::NoError) {
+    connect(reply, static_cast<void(QNetworkReply::*)(QNetworkReply::NetworkError) > (&QNetworkReply::error),
+            [this, reply](QNetworkReply::NetworkError error)
+    {
+        if(error != QNetworkReply::NoError)
+        {
            emit errorOccur(reply->errorString());
        }
    });
    connect(reply, &QNetworkReply::finished, &event, &QEventLoop::quit);
    event.exec();
+
    qint64 fileSize = 0;
-    if (reply->rawHeader("Accept-Ranges") == QByteArrayLiteral("bytes")
-            && reply->hasRawHeader(QString("Content-Length").toLocal8Bit())) {
+    if(reply->rawHeader("Accept-Ranges") == QByteArrayLiteral("bytes")
+            && reply->hasRawHeader(QString("Content-Length").toLocal8Bit()))
+    {
        fileSize = reply->header(QNetworkRequest::ContentLengthHeader).toUInt();
    }
    qDebug() << "文件大小为：" << fileSize;
@@ -251,9 +269,9 @@ qint64 DownloadController::getFileSize(const QString& url)
 QString DownloadController::replaceDomain(const QString& url, const QString domain)
 {
    QRegularExpression regex(R"((?:[a-z0-9](?:[a-z0-9-]{0,61}[a-z0-9])?\.)+[a-z0-9][a-z0-9-]{0,61}[a-z0-9])");
-    if (regex.match(url).hasMatch()) {
+    if(regex.match(url).hasMatch())
+    {
        return QString(url).replace(regex.match(url).captured(), domain);
    }
    return url;
 }
-