Changeset 6388 for branches/MPI/HeuristicLab.ExtLibs/HeuristicLab.MPInet/MPIAlgorithmRunner/3.3/Program.cs

Timestamp:

06/08/11 11:07:33 (13 years ago)

Author:

svonolfe

Message:

Worked on the MPIEngine (#1542)

File:

• branches/MPI/HeuristicLab.ExtLibs/HeuristicLab.MPInet/MPIAlgorithmRunner/3.3/Program.cs (modified) (3 diffs)

branches/MPI/HeuristicLab.ExtLibs/HeuristicLab.MPInet/MPIAlgorithmRunner/3.3/Program.cs

@@ -29 +29 @@
 using HeuristicLab.Operators.MPISupport;
 using HeuristicLab.Core;
+using Microsoft.Hpc.Scheduler;
+using System.ServiceModel;
+using System.Net;
 
 namespace HeuristicLab.MPIAlgorithmRunner {
@@ -34 +37 @@
     EventWaitHandle waitHandle = new EventWaitHandle(false, EventResetMode.ManualReset);
 
+    private static void SetJobStatus(string address) {
+      Console.WriteLine("Started service... at " + address);
+
+      // Discover the job's context from the environment
+      String headNodeName = System.Environment.GetEnvironmentVariable("CCP_CLUSTER_NAME");
+      int jobId = System.Convert.ToInt32(System.Environment.GetEnvironmentVariable("CCP_JOBID"));
+
+      // Connect to the head node and get the job
+      IScheduler scheduler = new Scheduler();
+      scheduler.Connect(headNodeName);
+      ISchedulerJob job = scheduler.OpenJob(jobId);
+
+      job.SetCustomProperty("address", address);
+
+      job.Commit();
+    }
+
+    private static string GetAddress(ServiceHost service) {
+      IPHostEntry IPHost = Dns.GetHostByName(Dns.GetHostName());
+      string address = "net.tcp://" + IPHost.AddressList[0].ToString() + ":" + service.ChannelDispatchers[0].Listener.Uri.Port + "/AlgorithmBroker";
+
+      return address;
+    }
+
     static void Main(string[] args) {
-      if (args.Length != 3) {
-        Console.WriteLine("Args:" + args.Length);
+      using (new MPI.Environment(ref args)) {
+        int clients = MPI.Communicator.world.Group.Size;
+        int updateInterval = 5000;
 
-        throw new ArgumentException("You must provide 3 arguments - the algorithm file, the result file and the update interval (in ms)");
-      }
-
-      string fileName = args[0];
-      string resultName = args[1];
-      int updateInterval = int.Parse(args[2]);
-
-      using (new MPI.Environment(ref args)) {
         MPI.Communicator communicator = MPI.Communicator.world.Clone() as MPI.Communicator;
 
         if (communicator.Rank == 0) {
-          IAlgorithm alg = XmlParser.Deserialize<IAlgorithm>(fileName);
-          MPITransportWrapper<IAlgorithm> transport = new MPITransportWrapper<IAlgorithm>(alg);
+          ServiceHost service = AlgorithmBroker.StartService(communicator);
+          AlgorithmBroker broker = (service.SingletonInstance as AlgorithmBroker);
 
-          int clients = communicator.Size - 1;
-          Console.WriteLine("Sending alg to " + clients + " clients");
-          for (int i = 0; i < clients; i++) {
-            int client = i + 1;
-            communicator.Send(transport, client, 0);
-          }
+          string address = GetAddress(service);
+          SetJobStatus(address);
 
           ItemList<ResultCollection> results = new ItemList<ResultCollection>();
@@ -87 +103 @@
 
             Console.WriteLine("Update results");
-            lock (resultName) {
-              XmlGenerator.Serialize(results, resultName);
-            }
+            broker.Results = results;
           }
 
+          lock (broker.ExitWaitHandle) {
+            broker.Terminated = true;
+          }
+
+          broker.ExitWaitHandle.WaitOne();
           Console.WriteLine("Finished.");
+          service.Close();
         } else {
           Program p = new Program();