bludgeon/Daemon.cs

   1
   2 using System;
   3 using System.Diagnostics;
   4 using System.Threading;
   5
   6 using Beagle.Util;
   7
   8 namespace Bludgeon {
   9
  10         public class Daemon {
  11
  12                 private Daemon () { }
  13
  14                 // Returns the daemon's version
  15                 static public string Ping ()
  16                 {
  17                         Beagle.RequestMessage request;
  18                         request = new Beagle.DaemonInformationRequest ();
  19
  20                         Beagle.ResponseMessage response = null;
  21
  22                         while (response == null) {
  23                                 try {
  24                                         response = request.Send ();
  25                                 } catch {
  26                                         Log.Spew ("ERROR: Daemon is not responding");
  27                                         Thread.Sleep (1000);
  28                                 }
  29                         }
  30
  31                         Beagle.DaemonInformationResponse info;
  32                         info = (Beagle.DaemonInformationResponse) response;
  33
  34                         return info.Version;
  35                 }
  36
  37                 static public void Start ()
  38                 {
  39                         Log.Spew ("Starting daemon");
  40
  41                         string beagled;
  42                         beagled = Environment.GetEnvironmentVariable ("BEAGLED_COMMAND");
  43
  44                         string args;
  45                         args = "--debug --bg --allow-backend files";
  46
  47                         Process p;
  48                         p = new Process ();
  49                         p.StartInfo.UseShellExecute = false;
  50
  51                         p.StartInfo.FileName = beagled;
  52                         p.StartInfo.Arguments = args;
  53
  54                         p.StartInfo.EnvironmentVariables ["BEAGLE_HOME"] = Beagle.Util.PathFinder.HomeDir;
  55                         p.StartInfo.EnvironmentVariables ["BEAGLE_EXERCISE_THE_DOG"] = "1";
  56
  57                         Thread.Sleep (2000); // wait 2s to let the daemon get started
  58
  59                         p.Start ();
  60
  61                         string version;
  62                         version = Ping (); // Then try to ping the daemon
  63
  64                         Log.Spew ("Successfully started daemon (version={0})", version);
  65                 }
  66
  67                 static public string GetStatus ()
  68                 {
  69                         Beagle.RequestMessage request;
  70                         request = new Beagle.DaemonInformationRequest ();
  71
  72                         Beagle.ResponseMessage response = null;
  73
  74                         int failure_count = 0;
  75                         while (response == null) {
  76                                 try {
  77                                         response = request.Send ();
  78                                 } catch {
  79                                         ++failure_count;
  80                                         if (failure_count > 10)
  81                                                 Log.Spew ("Daemon is not responding");
  82                                         Thread.Sleep (1000);
  83                                 }
  84                         }
  85
  86                         Beagle.DaemonInformationResponse info;
  87                         info = (Beagle.DaemonInformationResponse) response;
  88
  89                         return info.HumanReadableStatus;
  90                 }
  91
  92                 static public void WaitUntilIdle ()
  93                 {
  94                         Stopwatch sw = new Stopwatch ();
  95                         sw.Start ();
  96
  97                         bool first = true;
  98
  99                         while (true) {
 100                                 string status;
 101                                 status = GetStatus ();
 102                                 if (status.IndexOf ("Waiting on empty queue") != -1)
 103                                         break;
 104
 105                                 if (first)
 106                                         Log.Spew ("Waiting for daemon to become idle");
 107                                 first = false;
 108
 109                                 Thread.Sleep (1000);
 110                         }
 111                         sw.Stop ();
 112
 113                         if (! first)
 114                                 Log.Spew ("Waited for {0}", sw);
 115                 }
 116
 117                 static public void OptimizeIndexes ()
 118                 {
 119                         Beagle.RequestMessage request;
 120                         request = new Beagle.OptimizeIndexesRequest ();
 121
 122                         Log.Spew ("Optimizing Indexes");
 123
 124                         try {
 125                                 request.Send ();
 126                         } catch {
 127                                 Log.Failure ("Optimize request failed");
 128                         }
 129                 }
 130
 131                 static public void Shutdown ()
 132                 {
 133                         Beagle.RequestMessage request;
 134                         request = new Beagle.ShutdownRequest ();
 135
 136                         Log.Spew ("Shutting down daemon");
 137
 138                         try {
 139                                 request.Send ();
 140                         } catch {
 141                                 Log.Failure ("beagled shutdown failed");
 142                         }
 143                 }
 144         }
 145 }
 146