From f4e4b47d0d5e850c616e83afab0fe8f008b4c307 Mon Sep 17 00:00:00 2001
From: Mingliang Tao <mintao@microsoft.com>
Date: Fri, 13 Nov 2020 09:49:20 +0800
Subject: [PATCH] Disable stop button immediately after click (#5079)

---
 .../app/job/job-view/fabric/JobList/index.jsx | 72 +++++++++----------
 .../app/job/job-view/fabric/job-detail.jsx    |  7 +-
 2 files changed, 42 insertions(+), 37 deletions(-)

diff --git a/src/webportal/src/app/job/job-view/fabric/JobList/index.jsx b/src/webportal/src/app/job/job-view/fabric/JobList/index.jsx
index 8a1bc7ee9d..931129a8f4 100644
--- a/src/webportal/src/app/job/job-view/fabric/JobList/index.jsx
+++ b/src/webportal/src/app/job/job-view/fabric/JobList/index.jsx
@@ -12,7 +12,7 @@ import React, {
   useEffect,
   useRef,
 } from 'react';
-import { debounce, isEmpty } from 'lodash';
+import { debounce, isEmpty, cloneDeep } from 'lodash';
 
 import { ColorClassNames, getTheme } from '@uifabric/styling';
 import { Fabric } from 'office-ui-fabric-react/lib/Fabric';
@@ -39,7 +39,6 @@ export default function JobList() {
   const admin = userAuth.checkAdmin();
   const username = cookies.get('user');
 
-  const [allJobs, setAllJobs] = useState(null);
   const [selectedJobs, setSelectedJobs] = useState([]);
   const [error, setError] = useState(null);
 
@@ -164,42 +163,43 @@ export default function JobList() {
     applyOrdering(ordering);
   }, [applyOrdering, ordering]);
 
-  const stopJob = useCallback(
-    (...jobs) => {
-      userAuth.checkToken(token => {
-        jobs.forEach(job => {
-          const { name, username } = job;
-          const client = new PAIV2.OpenPAIClient({
-            rest_server_uri: new URL(
-              webportalConfig.restServerUri,
-              window.location.href,
-            ),
-            username: username,
-            token: token,
-            https: window.location.protocol === 'https:',
-          });
-          client.job
-            .updateJobExecutionType(username, name, 'STOP')
-            .then(() => {
-              job.executionType = 'STOP';
-              delete job._statusText;
-              delete job._statusIndex;
-              setAllJobs(allJobs.slice());
-            })
-            .catch(err => {
-              if (err.data.code === 'UnauthorizedUserError') {
-                alert(err.data.message);
-                clearToken();
-              } else {
-                alert(err.data.message || err.message);
-                throw new Error(err.data.message || err.message);
-              }
-            });
+  const stopJob = (...jobs) => {
+    userAuth.checkToken(token => {
+      jobs.forEach(job => {
+        const { name, username } = job;
+        const client = new PAIV2.OpenPAIClient({
+          rest_server_uri: new URL(
+            webportalConfig.restServerUri,
+            window.location.href,
+          ),
+          username: username,
+          token: token,
+          https: window.location.protocol === 'https:',
         });
+        client.job
+          .updateJobExecutionType(username, name, 'STOP')
+          .then(() => {
+            job.executionType = 'STOP';
+            delete job._statusText;
+            delete job._statusIndex;
+            const newFilteredJobsInfo = cloneDeep(filteredJobsInfo);
+            setFilteredJobsInfo(newFilteredJobsInfo);
+          })
+          .catch(err => {
+            if (err.data && err.data.code === 'UnauthorizedUserError') {
+              alert(err.data.message);
+              clearToken();
+            } else if (err.data) {
+              alert(err.data.message);
+              throw new Error(err.data.message);
+            } else {
+              alert(err.message);
+              throw new Error(err.message);
+            }
+          });
       });
-    },
-    [allJobs],
-  );
+    });
+  };
 
   const getJobs = async query => {
     const token = userAuth.checkToken();
diff --git a/src/webportal/src/app/job/job-view/fabric/job-detail.jsx b/src/webportal/src/app/job/job-view/fabric/job-detail.jsx
index 12befdb18f..9ed39fc3c6 100644
--- a/src/webportal/src/app/job/job-view/fabric/job-detail.jsx
+++ b/src/webportal/src/app/job/job-view/fabric/job-detail.jsx
@@ -15,7 +15,7 @@
 // DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 
-import { capitalize, isEmpty, isNil, get } from 'lodash';
+import { capitalize, isEmpty, isNil, get, cloneDeep } from 'lodash';
 import { DateTime, Interval } from 'luxon';
 import {
   MessageBar,
@@ -162,6 +162,11 @@ class JobDetail extends React.Component {
 
   async stop() {
     await stopJob();
+    const newJobInfo = cloneDeep(this.state.jobInfo);
+    newJobInfo.jobStatus.executionType = 'STOP';
+    this.setState({
+      jobInfo: newJobInfo,
+    });
     await this.reload();
   }