Changeset 5826 for branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave.Views/3.4/SlaveView.cs

Timestamp:

03/24/11 16:15:23 (13 years ago)

Author:

ascheibe

Message:

#1233 slave ui now receives status information and displays it in doughnut chart

File:

• branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave.Views/3.4/SlaveView.cs (modified) (7 diffs)

branches/HeuristicLab.Hive-3.4/sources/HeuristicLab.Clients.Hive.Slave.Views/3.4/SlaveView.cs

@@ -21 +21 @@
 
 using System;
+using System.Security.Principal;
+using System.ServiceProcess;
 using System.Threading;
 using System.Threading.Tasks;
@@ -47 +49 @@
     public SlaveView() {
       InitializeComponent();
+
+      if (!CheckRunAsAdmin())
+        btnRestartService.Enabled = false;
+      btnRestart.Enabled = false;
+
     }
 
@@ -99 +106 @@
     #region Event Handlers
     void Content_SlaveStatusChanged(object sender, EventArgs<StatusCommons> e) {
-      txtLog.AppendText("got a status changed object from slave\n");
       RenderChart(e.Value);
     }
@@ -114 +120 @@
 
     private void RenderChart(StatusCommons status) {
+      jobChart.Series[0].Points.Clear();
 
       DataPoint pJobs = new DataPoint(status.Jobs.Count, status.Jobs.Count);
@@ -120 +127 @@
       DataPoint pJobsFetched = new DataPoint(status.JobsFetched, status.JobsFetched);
 
+      pJobs.LegendText = "Current jobs: " + status.Jobs.Count;
+      pJobs.Color = System.Drawing.Color.Orange;
+      pJobsAborted.LegendText = "Aborted jobs: " + status.JobsAborted;
+      pJobsAborted.Color = System.Drawing.Color.Red;
+      pJobsDone.LegendText = "Finished jobs: " + status.JobsDone;
+      pJobsDone.Color = System.Drawing.Color.Green;
+      pJobsFetched.LegendText = "Fetched jobs: " + status.JobsFetched;
+      pJobsFetched.Color = System.Drawing.Color.Blue;
+
       jobChart.Series[0].Points.Add(pJobs);
       jobChart.Series[0].Points.Add(pJobsAborted);
@@ -137 +153 @@
 
     private void btnRestart_Click(object sender, System.EventArgs e) {
-      if (Content != null)
+      if (Content != null) {
         Content.RestartCore();
+        btnRestart.Enabled = false;
+        btnSleep.Enabled = true;
+      }
     }
 
     private void btnSleep_Click(object sender, System.EventArgs e) {
-      if (Content != null)
+      if (Content != null) {
         Content.Sleep();
+        btnRestart.Enabled = true;
+        btnSleep.Enabled = false;
+      }
     }
 
@@ -150 +172 @@
       diag.ShowDialog();
     }
+
+    private void btnRestartService_Click(object sender, EventArgs e) {
+      if (CheckRunAsAdmin()) {
+        RestartService();
+      } else {
+        MessageBox.Show("You need to run this application as Administrator to restart the service");
+      }
+    }
+
+    private bool CheckRunAsAdmin() {
+      bool isRunAsAdmin = false;
+      WindowsIdentity user = WindowsIdentity.GetCurrent();
+      WindowsPrincipal principal = new WindowsPrincipal(user);
+
+      try {
+        isRunAsAdmin = principal.IsInRole(WindowsBuiltInRole.Administrator);
+      }
+      catch { }
+      return isRunAsAdmin;
+    }
+
+    private void RestartService() {
+      string serviceName = "HeuristicLab.Clients.Hive.SlaveCore.SlaveWindowsService";
+      TimeSpan timeout = TimeSpan.FromMilliseconds(5000);
+
+      ServiceController service = new ServiceController(serviceName);
+      try {
+        if (service.Status == ServiceControllerStatus.Running) {
+          service.Stop();
+          service.WaitForStatus(ServiceControllerStatus.Stopped, timeout);
+        }
+
+        service.Start();
+        service.WaitForStatus(ServiceControllerStatus.Running, timeout);
+      }
+      catch (InvalidOperationException ex) {
+        MessageBox.Show("Error starting service: Hive Slave Service not found!\n" + ex.ToString());
+      }
+      catch (Exception ex) {
+        MessageBox.Show("Error starting service, exception is: \n" + ex.ToString());
+      }
+    }
   }
 }