TAO/orbsvcs/examples/Notify/MC/monitor/monitor.cpp

   1 #include "ace/Get_Opt.h"
   2 #include "ace/OS_NS_ctype.h"
   3
   4 #include "orbsvcs/Notify/MonitorControl/NotificationServiceMCC.h"
   5
   6 static const ACE_TCHAR* monitor_ior = 0;
   7 static const char* shutdown_cmd = "shutdown";
   8 static const char* rm_consumer = "remove_consumer";
   9 static const char* rm_supplier = "remove_supplier";
  10 static const char* rm_consumeradmin = "remove_consumeradmin";
  11 static const char* rm_supplieradmin = "remove_supplieradmin";
  12
  13 static int
  14 parse_args (int argc, ACE_TCHAR *argv[])
  15 {
  16   ACE_Get_Opt get_opts (argc, argv, ACE_TEXT ("k:"));
  17   int c;
  18
  19   while ((c = get_opts ()) != -1)
  20     {
  21       switch (c)
  22         {
  23         case 'k':
  24           monitor_ior = get_opts.opt_arg ();
  25           break;
  26         case '?':
  27         default:
  28           ACE_ERROR_RETURN ((LM_ERROR,
  29                              "usage: %s "
  30                              "-k <ior> "
  31                              "\n",
  32                              argv [0]),
  33                             -1);
  34         }
  35     }
  36
  37   return 0;
  38 }
  39
  40 extern "C" int
  41 sorter (const void* a, const void* b)
  42 {
  43   const char* left = *(reinterpret_cast<const char* const*> (a));
  44   const char* right = *(reinterpret_cast<const char* const*> (b));
  45   return ACE_OS::strcmp (left, right);
  46 }
  47
  48 bool
  49 process_command (CosNotification::NotificationServiceMonitorControl_ptr nsm,
  50                  char* buf,
  51                  const char* command)
  52 {
  53   if (ACE_OS::strstr (buf, command) == buf)
  54     {
  55       const char* name = buf + ACE_OS::strlen (command);
  56       bool space = false;
  57       size_t i = 0;
  58
  59       while (ACE_OS::ace_isspace (name[i]))
  60         {
  61           space = true;
  62           ++i;
  63         }
  64
  65       if (space)
  66         {
  67           try
  68             {
  69               const char* start = name + i;
  70
  71               if (ACE_OS::strcmp (command, shutdown_cmd) == 0)
  72                 {
  73                   nsm->shutdown_event_channel (start);
  74                 }
  75               else if (ACE_OS::strcmp (command, rm_consumer) == 0)
  76                 {
  77                   nsm->remove_consumer (start);
  78                 }
  79               else if (ACE_OS::strcmp (command, rm_supplier) == 0)
  80                 {
  81                   nsm->remove_supplier (start);
  82                 }
  83               else if (ACE_OS::strcmp (command, rm_consumeradmin) == 0)
  84                 {
  85                   nsm->remove_consumeradmin (start);
  86                 }
  87               else if (ACE_OS::strcmp (command, rm_supplieradmin) == 0)
  88                 {
  89                   nsm->remove_supplieradmin (start);
  90                 }
  91             }
  92           catch (const CORBA::Exception& ex)
  93             {
  94               ACE_OS::strcat (buf, ": ");
  95               ex._tao_print_exception (buf);
  96             }
  97
  98           return true;
  99         }
 100     }
 101
 102   return false;
 103 }
 104
 105 int
 106 ACE_TMAIN (int argc, ACE_TCHAR* argv[])
 107 {
 108   int status = 0;
 109
 110   try
 111     {
 112       CORBA::ORB_var orb = CORBA::ORB_init (argc, argv);
 113
 114       if (parse_args (argc, argv) != 0)
 115         {
 116           return 1;
 117         }
 118
 119       CORBA::Object_var obj = orb->string_to_object (
 120         ACE_TEXT_ALWAYS_CHAR (monitor_ior));
 121       CosNotification::NotificationServiceMonitorControl_var nsm =
 122         CosNotification::NotificationServiceMonitorControl::_narrow (obj.in ());
 123
 124       if (CORBA::is_nil (nsm.in ()))
 125         {
 126           ACE_ERROR_RETURN ((LM_ERROR,
 127                              "Unable to locate the "
 128                              "Notification Service Monitor\n"),
 129                             1);
 130         }
 131
 132       bool done = false;
 133       static const size_t lsize = 1024;
 134       char prev[lsize];
 135
 136       while (!done)
 137         {
 138           ACE_OS::printf ("NotifyService> ");
 139           ACE_OS::fflush (stdout);
 140
 141           char line[lsize] = "";
 142           char* rl = ACE_OS::fgets (line, lsize, stdin);
 143
 144           if (rl != 0)
 145             {
 146               int len = static_cast<int> (ACE_OS::strlen (line));
 147
 148               for (int i = 0; i < len && ACE_OS::ace_isspace (rl[0]); ++i)
 149                 {
 150                   rl++;
 151                 }
 152
 153               for (int i = len - 1; i >= 0 && ACE_OS::ace_isspace (line[i]); --i)
 154                 {
 155                   line[i] = '\0';
 156                 }
 157
 158               if (ACE_OS::strlen (rl) == 0 || ACE_OS::strcmp (rl, "!!") == 0)
 159                 {
 160                   ACE_OS::strcpy (line, prev);
 161                   rl = line;
 162                 }
 163
 164               ACE_OS::strcpy (prev, line);
 165             }
 166
 167
 168           if (rl == 0)
 169             {
 170               done = true;
 171               ACE_DEBUG ((LM_DEBUG, "\n"));
 172             }
 173           else if (ACE_OS::strlen (rl) == 0)
 174             {
 175             }
 176           else if (ACE_OS::strcmp (rl, "quit") == 0)
 177             {
 178               done = true;
 179             }
 180           else if (ACE_OS::strcmp (rl, "help") == 0)
 181             {
 182               ACE_DEBUG ((LM_DEBUG,
 183                           "names                - Get a list of "
 184                           "currently available statistic names.\n"
 185                           "quit                 - Exit the monitor.\n"
 186                           "remove_consumer      - Remove a consumer "
 187                           "with the given name.\n"
 188                           "remove_supplier      - Remove a supplier "
 189                           "with the given name.\n"
 190                           "remove_consumeradmin - Remove a consumer "
 191                           "admin with the given name.\n"
 192                           "remove_supplieradmin - Remove a supplier "
 193                           "admin with the given name.\n"
 194                           "shutdown             - Shut down an "
 195                           "event channel with the given name.\n"
 196                           "<statistic name>     - Return the "
 197                           "information for the specified statistic.\n"));
 198             }
 199           else if (ACE_OS::strcmp (rl, "names") == 0)
 200             {
 201               try
 202                 {
 203                   Monitor::NameList_var names = nsm->get_statistic_names ();
 204                   CORBA::ULong length = names->length ();
 205                   ACE_DEBUG ((LM_DEBUG, "Statistic names\n"));
 206
 207                   // It's much easier to read once it's sorted
 208                   const char** narray = 0;
 209                   ACE_NEW_THROW_EX (narray,
 210                                     const char* [length],
 211                                     CORBA::NO_MEMORY ());
 212
 213                   for (CORBA::ULong i = 0; i < length; ++i)
 214                     {
 215                       narray[i] = names[i].in ();
 216                     }
 217
 218                   ACE_OS::qsort (narray,
 219                                  length,
 220                                  sizeof (const char*),
 221                                  sorter);
 222
 223                   for (CORBA::ULong i = 0; i < length; ++i)
 224                     {
 225                       ACE_DEBUG ((LM_DEBUG, "  %s\n", narray[i]));
 226                     }
 227
 228                   delete [] narray;
 229                 }
 230               catch (const CORBA::Exception& ex)
 231                 {
 232                   ex._tao_print_exception ("names: ");
 233                 }
 234             }
 235           else
 236             {
 237               if (process_command (nsm.in (), rl, shutdown_cmd))
 238                 {
 239                   continue;
 240                 }
 241               else if (process_command (nsm.in (), rl, rm_consumer))
 242                 {
 243                   continue;
 244                 }
 245               else if (process_command (nsm.in (), rl, rm_supplier))
 246                 {
 247                   continue;
 248                 }
 249               else if (process_command (nsm.in (), rl, rm_consumeradmin))
 250                 {
 251                   continue;
 252                 }
 253               else if (process_command (nsm.in (), rl, rm_supplieradmin))
 254                 {
 255                   continue;
 256                 }
 257
 258               try
 259                 {
 260                   Monitor::Data_var data =
 261                     nsm->get_statistic (rl);
 262                   ACE_DEBUG ((LM_DEBUG, "%s => ", rl));
 263
 264                   if (data->data_union._d () == Monitor::DATA_NUMERIC)
 265                     {
 266                       ACE_DEBUG ((LM_DEBUG,
 267                                   "Last: %g Average: %g\n",
 268                                   data->data_union.num().dlist[0].value,
 269                                   data->data_union.num().average));
 270                     }
 271                   else
 272                     {
 273                       Monitor::NameList list = data->data_union.list ();
 274                       CORBA::ULong const length = list.length ();
 275
 276                       for (CORBA::ULong i = 0; i < length; ++i)
 277                         {
 278                           ACE_DEBUG ((LM_DEBUG, "%s ", list[i].in ()));
 279                         }
 280
 281                       ACE_DEBUG ((LM_DEBUG, "\n"));
 282                     }
 283                 }
 284               catch (const CORBA::Exception& ex)
 285                 {
 286                   ACE_OS::strcat (rl, ": ");
 287                   ex._tao_print_exception (rl);
 288                 }
 289             }
 290         }
 291
 292       orb->destroy ();
 293     }
 294   catch (const CORBA::UserException& ex)
 295     {
 296       ex._tao_print_exception ("Notification Service Monitor: ");
 297       ++status;
 298     }
 299   catch (const CORBA::Exception& ex)
 300     {
 301       ex._tao_print_exception ("Notification Service Monitor: ");
 302       ++status;
 303     }
 304   catch (...)
 305     {
 306       ACE_ERROR ((LM_ERROR,
 307                   "Notification Service Monitor: "
 308                   "unexpected exception type\n"));
 309       status++;
 310     }
 311
 312   return status;
 313 }