From bcc21ccded5f62a4075181418cfca03b3da4bc8e Mon Sep 17 00:00:00 2001
From: Srivats P <pstavirs@users.noreply.github.com>
Date: Sun, 27 Dec 2020 17:00:28 +0530
Subject: [PATCH] Gracefully handle portgroup disconnect during Apply

Fixes #326
---
 client/portgroup.cpp | 11 +++++++++++
 1 file changed, 11 insertions(+)

diff --git a/client/portgroup.cpp b/client/portgroup.cpp
index e2b48e2..7b259ac 100644
--- a/client/portgroup.cpp
+++ b/client/portgroup.cpp
@@ -253,6 +253,16 @@ void PortGroup::on_rpcChannel_disconnected()
     emit portListChanged(mPortGroupId);
     emit portGroupDataChanged(mPortGroupId);
 
+    // Disconnected during apply? Restore UI.
+    if (applyTimer_.isValid()) {
+        applyTimer_.invalidate();
+
+        emit applyFinished();
+
+        mainWindow->setEnabled(true);
+        QApplication::restoreOverrideCursor();
+    }
+
     isGetStatsPending_ = false;
 
     if (reconnect)
@@ -817,6 +827,7 @@ void PortGroup::processApplyBuildAck(int portIndex, PbRpcController *controller)
     logInfo(id(), mPorts[portIndex]->id(),
             QString("All port changes applied - in %1s")
                 .arg(applyTimer_.elapsed()/1e3));
+    applyTimer_.invalidate();
 
     if (controller->Failed())
     {