1 files changed, 67 insertions, 0 deletions
diff --git a/org.eclipse.osee.ote.core/src/org/eclipse/osee/ote/WaitForCompletion.java b/org.eclipse.osee.ote.core/src/org/eclipse/osee/ote/WaitForCompletion.java
new file mode 100644
index 000000000..883d1208f
--- /dev/null
+++ b/org.eclipse.osee.ote.core/src/org/eclipse/osee/ote/WaitForCompletion.java
@@ -0,0 +1,67 @@
+package org.eclipse.osee.ote;
+
+import java.io.IOException;
+
+import org.eclipse.core.runtime.SubProgressMonitor;
+import org.eclipse.osee.ote.message.event.send.OteEventMessageCallable;
+import org.eclipse.osee.ote.message.event.send.OteEventMessageFuture;
+import org.eclipse.osee.ote.remote.messages.JobStatus;
+import org.eclipse.osee.ote.remote.messages.SerializedConfigurationAndResponse;
+import org.eclipse.osee.ote.remote.messages.SerializedOTEJobStatus;
+
+class WaitForCompletion implements OteEventMessageCallable<SerializedConfigurationAndResponse, SerializedOTEJobStatus> {
+
+   private SubProgressMonitor monitor;
+   private int lastUnitsWorked = 0;
+   private boolean firstTime = true;
+   private JobStatus status;
+
+   WaitForCompletion(SubProgressMonitor monitor) {
+      this.monitor = monitor;
+   }
+
+   @Override
+   public void call(SerializedConfigurationAndResponse transmitted, SerializedOTEJobStatus recieved, OteEventMessageFuture<?, ?> future) {
+      
+      try {
+         this.status = recieved.getObject();
+         reportStatus(status, future);
+      } catch (IOException e) {
+         e.printStackTrace();
+      } catch (ClassNotFoundException e) {
+         e.printStackTrace();
+      }
+   }
+
+   private synchronized void reportStatus(JobStatus status, OteEventMessageFuture<?, ?> future) {
+      if(monitor != null){
+         if(monitor.isCanceled() || status.isJobComplete()){
+            monitor.done();
+            future.complete();
+            return;
+         }
+         if(firstTime){
+            monitor.beginTask("Configure Test Server", status.getTotalUnitsOfWork());
+            firstTime = false;
+         } else {
+            monitor.worked(status.getUnitsWorked() - lastUnitsWorked);
+            lastUnitsWorked = status.getUnitsWorked();
+         }
+      } else {
+         if(status.isJobComplete()){
+            future.complete();
+            return;
+         }
+      }
+   }
+
+   @Override
+   public void timeout(SerializedConfigurationAndResponse transmitted) {
+      System.out.println("timed out");
+   }
+   
+   public JobStatus getStatus(){
+      return this.status;
+   }
+   
+}