From 08e7df4ad39b42cccadcbe50113743cbece8c62a Mon Sep 17 00:00:00 2001
From: Stephan Porada <sporada@uni-bielefeld.de>
Date: Thu, 26 Mar 2020 15:28:11 +0100
Subject: [PATCH] Add function to resend queries during running query

---
 app/corpora/events.py                        | 81 ++++++++++++++++----
 app/static/js/nopaque.analyse_corpus.js      | 67 ++++++++++++++--
 app/templates/corpora/analyse_corpus.html.j2 | 18 ++++-
 3 files changed, 141 insertions(+), 25 deletions(-)

diff --git a/app/corpora/events.py b/app/corpora/events.py
index 19d0fd0d..0f3828ec 100644
--- a/app/corpora/events.py
+++ b/app/corpora/events.py
@@ -15,22 +15,36 @@ corpus_analysis_sessions = {}
 '''
 ' A dictionary containing Socket.IO session id - CQi client pairs.
 ' {<sid>: CQiClient, ...}
+' A dictionary containing Socket.IO session id - and its status.
+' {<sid>: 'running'|'idle'|'abort'}
 '''
 corpus_analysis_clients = {}
+corpus_analysis_clients_status = {}
+
+
+@socketio.on('send_analysis_status_cli')
+@login_required
+def update_status(response):
+    update_analysis_status(response['status'])
 
 
 @socketio.on('corpus_analysis_init')
 @login_required
 def init_corpus_analysis(corpus_id):
-    logger.warning("Initiating corpus analysis.")
+    logger.warning('Initiating corpus analysis.')
     socketio.start_background_task(corpus_analysis_session_handler,
                                    current_app._get_current_object(),
                                    corpus_id, current_user.id, request.sid)
+    update_analysis_status('idle')
 
 
 @socketio.on('corpus_analysis_query')
 @login_required
 def corpus_analysis_query(query):
+    update_analysis_status('running')
+    logger.warning('Recieved a query.')
+    logger.warning('Clients current session ID is: {}'.format(request.sid))
+    current_client_status = get_analysis_status()
     client = corpus_analysis_clients.get(request.sid)
     if client is None:
         socketio.emit('query', '[424]: Failed Dependency', room=request.sid)
@@ -41,7 +55,7 @@ def corpus_analysis_query(query):
     except Exception as e:
         logger.warning(e)
         response = str(e)
-        if response == "CQI_CQP_ERROR_GENERAL":
+        if response == 'CQI_CQP_ERROR_GENERAL':
             response = {'code': 1}
             socketio.emit('corpus_analysis_query', response, room=request.sid)
     else:
@@ -50,21 +64,52 @@ def corpus_analysis_query(query):
         context = 100
         progress = 0
         while chunk_start <= results.size:
-            chunk = results.dump_values(context=context,
-                                        first_result=chunk_start,
-                                        num_results=chunk_size)
-            if (results.size == 0):
-                progress = 100
+            current_client_status = get_analysis_status()
+            logger.warning('Status at while start: {}'.format(current_client_status))
+            if current_client_status == 'abort':
+                update_analysis_status('idle')
+                logger.warning('Aborting due to status being "abort".')
+                break
             else:
-                progress = ((chunk_start + chunk_size) / results.size) * 100
-                progress = min(100, int(math.ceil(progress)))
-            socketio.emit('corpus_analysis_query',
-                          {'chunk': chunk,
-                           'progress': progress,
-                           'num_matches_total': results.size,
-                           'code': 0},
-                          room=request.sid)
-            chunk_start += chunk_size
+                chunk = results.dump_values(context=context,
+                                            first_result=chunk_start,
+                                            num_results=chunk_size)
+                if (results.size == 0):
+                    progress = 100
+                else:
+                    progress = ((chunk_start + chunk_size) / results.size) * 100
+                    progress = min(100, int(math.ceil(progress)))
+                socketio.emit('corpus_analysis_query',
+                              {'chunk': chunk,
+                               'progress': progress,
+                               'num_matches_total': results.size,
+                               'code': 0},
+                              room=request.sid)
+                chunk_start += chunk_size
+                current_client_status = get_analysis_status()
+                if (progress == 100):
+                    update_analysis_status('idle')
+                elif (current_client_status != 'abort'):
+                    update_analysis_status('running')
+                logger.warning('Status at while end: {}'.format(get_analysis_status()))
+
+
+def update_analysis_status(status_str):
+    # first: sets new status on server side
+    session_id = request.sid
+    corpus_analysis_clients_status[session_id] = status_str
+    logger.warning('Set new status to "{}" on server side for {}.'.format(status_str, session_id))
+    # second: send new status to client side
+    if status_str != 'abort':
+        message = {'status': status_str}
+        socketio.emit('send_analysis_status_server', message, room=session_id)
+        logger.warning('Sending status "{}" to client.'.format(status_str))
+
+
+def get_analysis_status():
+    status = corpus_analysis_clients_status[request.sid]
+    logger.warning('Current status on server side is: {}'.format(status))
+    return status
 
 
 @socketio.on('inspect_match')
@@ -88,10 +133,12 @@ def corpus_analysis_session_handler(app, corpus_id, user_id, session_id):
         corpus = Corpus.query.get(corpus_id)
         user = User.query.get(user_id)
         if corpus is None:
+            logger.warning("ERROR 1")
             response = {'code': 404, 'msg': 'Not Found'}
             socketio.emit('corpus_analysis_init', response, room=session_id)
             return
         elif not (corpus.creator == user or user.is_administrator()):
+            logger.warning("ERROR 2")
             response = {'code': 403, 'msg': 'Forbidden'}
             socketio.emit('corpus_analysis_init', response, room=session_id)
             return
@@ -102,6 +149,7 @@ def corpus_analysis_session_handler(app, corpus_id, user_id, session_id):
         try:
             client.connect()
         except Exception:
+            logger.warning("ERROR 3")
             response = {'code': 500, 'msg': 'Internal Server Error'}
             socketio.emit('corpus_analysis_init', response, room=session_id)
             return
@@ -111,6 +159,7 @@ def corpus_analysis_session_handler(app, corpus_id, user_id, session_id):
         else:
             corpus_analysis_sessions[corpus_id].append(session_id)
         response = {'code': 201, 'msg': 'Created'}
+        logger.warning("ERROR 4")
         socketio.emit('corpus_analysis_init', response, room=session_id)
         ''' Observe analysis session '''
         while session_id in connected_sessions:
diff --git a/app/static/js/nopaque.analyse_corpus.js b/app/static/js/nopaque.analyse_corpus.js
index 8ec17397..da75ec20 100644
--- a/app/static/js/nopaque.analyse_corpus.js
+++ b/app/static/js/nopaque.analyse_corpus.js
@@ -1,3 +1,24 @@
+// ### corpus analysis client status functions
+
+function sendAbortSignal() {
+  console.log(`Sending status "abort signal" from client side.`);
+  message = {'status': "abort"};
+  nopaque.socket.emit("send_analysis_status_cli", message);
+  setAnalysisStatus("abort");
+}
+
+function setAnalysisStatus(statusStr) {
+  analysisClientStatus[sessionId] = statusStr;
+  console.log(`Set new status "${statusStr}" for "${sessionId}" on client side.`);
+  console.log("Object with status is:", analysisClientStatus);
+}
+
+function checkAnalysisStatus(sessionId) {
+  let status = analysisClientStatus[sessionId];
+  console.log(`Check status on client side for ${sessionId}.`, analysisClientStatus);
+  return status;
+}
+
 // ### Send query functions ###
 
 //function to get current queryData from a given queryFormElement
@@ -12,11 +33,13 @@ function getQueryData(queryFormElement) {
 }
 
 // Function to send a query to the CQP server using queryData gatherd with getQueryData
-function sendQuery(event) {
+async function sendQuery(event) {
+  const delay = ms => new Promise(res => setTimeout(res, ms));
   result;  // global declaration, holds query results as JSON
   resultList;  // global declaration, list.js object displaying the results
   let queryData;  // holds data of query form
   let resultListOptions;  // holding the options for the ResultList object
+  let analysisStatus;
 
   // checks if one query has been finished before
   // if true, result download will be disabled again until query is finished
@@ -26,9 +49,32 @@ function sendQuery(event) {
     queryResultsDeterminateElement.parentNode.parentNode.classList.remove("hide");
   }
 
+  console.log("Sending query.");
+  console.log("Session status are: ", analysisClientStatus);
   event.preventDefault();
+  analysisStatus = checkAnalysisStatus(sessionId);
   queryData = getQueryData(queryFormElement);
-  nopaque.socket.emit("corpus_analysis_query", queryData.query);
+
+  if (analysisStatus === "idle") {
+    nopaque.toast("Query has been sent!");
+    nopaque.socket.emit("corpus_analysis_query", queryData.query);
+    helperSendQuery(queryData);
+    analysisStatus = checkAnalysisStatus(sessionId);
+  } else if (analysisStatus === "running" ) {
+    sendAbortSignal();
+    analysisStatus = checkAnalysisStatus(sessionId);
+    while (analysisStatus === "abort") {
+      queryResultsTableElement.classList.add("hide");
+      queryLoadingElement.classList.remove("hide");
+      analysisStatus = checkAnalysisStatus(sessionId);
+      await delay(100);
+    }
+    sendQuery(event)
+  }
+}
+
+
+function helperSendQuery(queryData) {
   // full results object declaration, global declaration!
   // will always be reset if a query is sent, so that only the chunks of the
   // current query will be saved in it
@@ -42,7 +88,6 @@ function sendQuery(event) {
   // some hiding/showing for loading animation
   queryLoadingElement.classList.remove("hide");
   queryResultsTableElement.classList.add("hide");
-  nopaque.toast("Query has been sent!");
 
   resultListOptions = {page: queryData["hits_per_page"],
     pagination: [{
@@ -114,10 +159,16 @@ function recieveResults(response) {
   result["query"] = queryData["query"];
   console.log(result);
   // Some hiding and showing of loading animations
-  queryLoadingElement.classList.add("hide");
-  queryResultsTableElement.classList.remove("hide");
-  queryResultsElement = document.getElementById("query-results");
-  queryResultsElement.innerHTML = "";
+  analysisStatus = checkAnalysisStatus();
+  if (analysisStatus === "abort") {
+    queryResultsTableElement.classList.add("hide");
+    queryLoadingElement.classList.remove("hide");
+  } else {
+    queryResultsTableElement.classList.remove("hide");
+    queryLoadingElement.classList.add("hide");
+    queryResultsElement = document.getElementById("query-results");
+    queryResultsElement.innerHTML = "";
+  }
 
   // check if query has any results
   if (chunk["matches"].length === 0) {
@@ -252,7 +303,7 @@ function changeContext(event) {
       console.log("This error is expected.");
   } finally {
       newContextValue = document.getElementById("context").value;
-      console.log(newContextValue);
+      console.log("Context value is:", newContextValue);
       lc = document.getElementsByClassName("left-context");
       rc = document.getElementsByClassName("right-context");
       for (let element of lc) {
diff --git a/app/templates/corpora/analyse_corpus.html.j2 b/app/templates/corpora/analyse_corpus.html.j2
index 1473ce0e..af6efe66 100644
--- a/app/templates/corpora/analyse_corpus.html.j2
+++ b/app/templates/corpora/analyse_corpus.html.j2
@@ -239,7 +239,9 @@ server side -->
   // create some global variables
   var result;  // will gradually be filled with JSON result chunks of a query
   var resultList;  // the list.js new ResultList object displaying the results
-  var queryFinished;  // bool flag that shows if query is finished or nor
+  var queryFinished;  // bool flag that shows if query is finished or not
+  var analysisClientStatus;  // Object holding: {<sid>: 'running'|'idle'}
+  var sessionId;  // the clients current session id
   // create some global elements
   var exportQueryResultsElement;  // export button opens onclick download modal
   var queryResultsDeterminateElement; // progress bar for recieved query status
@@ -254,6 +256,13 @@ server side -->
 
   // set queryFinished to false initially
   queryFinished = false;
+  analysisClientStatus = {};
+
+  nopaque.socket.on("connect", function() {
+    sessionId = nopaque.socket.id;
+    console.log("Clients current session id:", sessionId);
+    setAnalysisStatus("idle", sessionId);
+  });
 
   // creates some models on DOMContentLoaded
   document.addEventListener("DOMContentLoaded", function() {
@@ -295,6 +304,13 @@ server side -->
   queryFormElement = document.getElementById("query-form");
   queryFormElement.addEventListener("submit", sendQuery);
 
+  // recieves server side analysis status updates and sets those accordingly in
+  // analysisClientStatus
+  nopaque.socket.on("send_analysis_status_server", (response) => {
+    console.log("Recieved new analysis status from server:", response.status);
+    setAnalysisStatus(response.status);
+  });
+
   // recieves results on "corpus_analysis_query" via socket.io
   nopaque.socket.on("corpus_analysis_query", recieveResults);
 
-- 
GitLab