Changeset 2032 for trunk/sources/HeuristicLab.Hive.Engine/3.2/HiveEngine.cs

Timestamp:

06/08/09 18:34:28 (15 years ago)

Author:

gkronber

Message:

Worked on asynchronous result polling in HiveEngine. #545 (Engine which can be executed in the Hive).

File:

• trunk/sources/HeuristicLab.Hive.Engine/3.2/HiveEngine.cs (modified) (5 diffs)

trunk/sources/HeuristicLab.Hive.Engine/3.2/HiveEngine.cs

@@ -33 +33 @@
 using System.Xml;
 using System.IO.Compression;
+using HeuristicLab.Tracing;
 
 namespace HeuristicLab.Hive.Engine {
@@ -44 +45 @@
     private Guid jobId;
     private Job job;
+    private object locker = new object();
+    private volatile bool abortRequested;
+
     public string HiveServerUrl { get; set; }
 
     public HiveEngine() {
       job = new Job();
+      abortRequested = false;
     }
 
@@ -96 +101 @@
         IExecutionEngineFacade executionEngineFacade = ServiceLocator.CreateExecutionEngineFacade(HiveServerUrl);
         ResponseObject<JobResult> response = null;
+        Job restoredJob = null;
         do {
-          response = executionEngineFacade.GetLastResult(jobId, true);
-          if (response.Success && response.StatusMessage == ApplicationConstants.RESPONSE_JOB_RESULT_NOT_YET_HERE) {
-            Thread.Sleep(RESULT_POLLING_INTERVAL_MS);
+          Thread.Sleep(RESULT_POLLING_INTERVAL_MS);
+          lock (locker) {
+            HiveLogger.Debug("HiveEngine: Results-polling - GetLastResult");
+            response = executionEngineFacade.GetLastResult(jobId, false);
+            HiveLogger.Debug("HiveEngine: Results-polling - Server: " + response.StatusMessage + " success: " + response.Success);
+            // loop while 
+            // 1. the user doesn't request an abort 
+            // 2. there is a problem with server communication (success==false)
+            // 3. no result for the job is available yet (response.Obj==null)
+            // 4. the result that we get from the server is a snapshot and not the final result
+            if (abortRequested) return;
+            if (response.Success && response.Obj != null) {
+              HiveLogger.Debug("HiveEngine: Results-polling - Got result!");
+              restoredJob = (Job)PersistenceManager.RestoreFromGZip(response.Obj.Result);
+              HiveLogger.Debug("HiveEngine: Results-polling - IsSnapshotResult: " + restoredJob.Engine.Canceled);
+            }
           }
-        } while (response.Success && response.StatusMessage == ApplicationConstants.RESPONSE_JOB_RESULT_NOT_YET_HERE);
-        if (response.Success) {
-          JobResult jobResult = response.Obj;
-          if (jobResult != null) {
-            job = (Job)PersistenceManager.RestoreFromGZip(jobResult.Result);
-            OnFinished();
-          }
-        } else {
-          Exception ex = new Exception(response.Obj.Exception.Message);
-          ThreadPool.QueueUserWorkItem(delegate(object state) { OnExceptionOccurred(ex); });
-        }
+        } while (restoredJob == null || restoredJob.Engine.Canceled);
+
+        job = restoredJob;
+        OnChanged();
+        OnFinished();
       });
+      HiveLogger.Debug("HiveEngine: Starting results-polling thread");
       t.Start();
     }
@@ -119 +133 @@
       IExecutionEngineFacade executionEngineFacade = ServiceLocator.CreateExecutionEngineFacade(HiveServerUrl);
 
-      // poll until snapshot is ready
       ResponseObject<JobResult> response;
-
-      // request snapshot
-      Response snapShotResponse = executionEngineFacade.RequestSnapshot(jobId);
-      if (snapShotResponse.StatusMessage == ApplicationConstants.RESPONSE_JOB_IS_NOT_BEEING_CALCULATED) {
-        response = executionEngineFacade.GetLastResult(jobId, false);
-      } else {
-        do {
-          response = executionEngineFacade.GetLastResult(jobId, true);
-          if (response.Success && response.StatusMessage == ApplicationConstants.RESPONSE_JOB_RESULT_NOT_YET_HERE) {
+      lock (locker) {
+        HiveLogger.Debug("HiveEngine: Abort - RequestSnapshot");
+        Response snapShotResponse = executionEngineFacade.RequestSnapshot(jobId);
+        if (snapShotResponse.StatusMessage == ApplicationConstants.RESPONSE_JOB_IS_NOT_BEEING_CALCULATED) {
+          // job is finished already
+          HiveLogger.Debug("HiveEngine: Abort - GetLastResult(false)");
+          response = executionEngineFacade.GetLastResult(jobId, false);
+          HiveLogger.Debug("HiveEngine: Abort - Server: " + response.StatusMessage + " success: " + response.Success);
+        } else {
+          // server sent snapshot request to client
+          // poll until snapshot is ready
+          do {
             Thread.Sleep(SNAPSHOT_POLLING_INTERVAL_MS);
-          }
-        } while (response.Success && response.StatusMessage == ApplicationConstants.RESPONSE_JOB_RESULT_NOT_YET_HERE);
+            HiveLogger.Debug("HiveEngine: Abort - GetLastResult(true)");
+            response = executionEngineFacade.GetLastResult(jobId, true);
+            HiveLogger.Debug("HiveEngine: Abort - Server: " + response.StatusMessage + " success: " + response.Success);
+            // loop while
+            // 1. problem with communication with server
+            // 2. job result not yet ready
+          } while (
+            !response.Success ||
+            response.StatusMessage == ApplicationConstants.RESPONSE_JOB_RESULT_NOT_YET_HERE);
+        }
       }
-      if (response.Success) {
-        JobResult jobResult = response.Obj;
-        if (jobResult != null) {
-          job = (Job)PersistenceManager.RestoreFromGZip(jobResult.Result);
-          //PluginManager.ControlManager.ShowControl(job.Engine.CreateView());
-        }
-      } else {
-        Exception ex = new Exception(response.Obj.Exception.Message);
-        ThreadPool.QueueUserWorkItem(delegate(object state) { OnExceptionOccurred(ex); });
+      JobResult jobResult = response.Obj;
+      if (jobResult != null) {
+        HiveLogger.Debug("HiveEngine: Results-polling - Got result!");
+        job = (Job)PersistenceManager.RestoreFromGZip(jobResult.Result);
+        //PluginManager.ControlManager.ShowControl(job.Engine.CreateView());
       }
+      //HiveLogger.Debug("HiveEngine: Results-polling - Exception!");
+      //Exception ex = new Exception(response.Obj.Exception.Message);
+      //ThreadPool.QueueUserWorkItem(delegate(object state) { OnExceptionOccurred(ex); });
     }
 
@@ -155 +178 @@
 
     public void Abort() {
+      abortRequested = true;
+      RequestSnapshot();
       IExecutionEngineFacade executionEngineFacade = ServiceLocator.CreateExecutionEngineFacade(HiveServerUrl);
       executionEngineFacade.AbortJob(jobId);
+      OnChanged();
       OnFinished();
     }
 
     public void Reset() {
+      abortRequested = false;
       job.Engine.Reset();
       jobId = Guid.NewGuid();