From 6b4a67d5b9d2cbc43b087c090bd3f5371f320ac8 Mon Sep 17 00:00:00 2001
From: Saturneric <eric@bktus.com>
Date: Sun, 22 May 2022 19:10:22 +0800
Subject: perf: improve ListedKeyServerTestThread

1. fixes #9
---
 src/ui/thread/ListedKeyServerTestTask.cpp | 88 +++++++++++++++++++++++++++++++
 1 file changed, 88 insertions(+)
 create mode 100644 src/ui/thread/ListedKeyServerTestTask.cpp

(limited to 'src/ui/thread/ListedKeyServerTestTask.cpp')

diff --git a/src/ui/thread/ListedKeyServerTestTask.cpp b/src/ui/thread/ListedKeyServerTestTask.cpp
new file mode 100644
index 00000000..9d4ca74d
--- /dev/null
+++ b/src/ui/thread/ListedKeyServerTestTask.cpp
@@ -0,0 +1,88 @@
+/**
+ * Copyright (C) 2021 Saturneric
+ *
+ * This file is part of GpgFrontend.
+ *
+ * GpgFrontend is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * GpgFrontend is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with GpgFrontend. If not, see <https://www.gnu.org/licenses/>.
+ *
+ * The initial version of the source code is inherited from
+ * the gpg4usb project, which is under GPL-3.0-or-later.
+ *
+ * The source code version of this software was modified and released
+ * by Saturneric<eric@bktus.com><eric@bktus.com> starting on May 12, 2021.
+ *
+ */
+
+#include "ListedKeyServerTestTask.h"
+
+#include <vector>
+
+GpgFrontend::UI::ListedKeyServerTestTask::ListedKeyServerTestTask(
+    const QStringList& urls, int timeout, QWidget* parent)
+    : urls_(urls),
+      timeout_(timeout),
+      network_manager_(new QNetworkAccessManager(this)),
+      result_(urls_.size(), kTestResultType_Error) {
+  qRegisterMetaType<std::vector<KeyServerTestResultType>>(
+      "std::vector<KeyServerTestResultType>");
+}
+
+void GpgFrontend::UI::ListedKeyServerTestTask::run() {
+  SetFinishAfterRun(false);
+
+  size_t index = 0;
+  for (const auto& url : urls_) {
+    auto key_url = QUrl{url};
+    LOG(INFO) << "key server request: " << key_url.host().toStdString();
+
+    auto* network_reply = network_manager_->get(QNetworkRequest{key_url});
+    auto* timer = new QTimer(this);
+
+    connect(network_reply, &QNetworkReply::finished, this,
+            [this, index, network_reply]() {
+              LOG(INFO) << "key server domain reply"
+                        << urls_[index].toStdString();
+              this->slot_process_network_reply(index, network_reply);
+            });
+
+    connect(timer, &QTimer::timeout, this, [this, index, network_reply]() {
+      LOG(INFO) << "timeout for key server" << urls_[index].toStdString();
+      if (network_reply->isRunning()) {
+        network_reply->abort();
+        this->slot_process_network_reply(index, network_reply);
+      }
+    });
+
+    timer->start(timeout_);
+
+    index++;
+  }
+}
+
+void GpgFrontend::UI::ListedKeyServerTestTask::slot_process_network_reply(
+    int index, QNetworkReply* reply) {
+  if (!reply->isRunning() && reply->error() == QNetworkReply::NoError) {
+    result_[index] = kTestResultType_Success;
+  } else {
+    if (!reply->isFinished())
+      result_[index] = kTestResultType_Timeout;
+    else
+      result_[index] = kTestResultType_Error;
+  }
+
+  if (++result_count_ == urls_.size()) {
+    emit SignalKeyServerListTestResult(result_);
+    emit SignalTaskFinished();
+  }
+}
-- 
cgit v1.2.3