TAO/examples/Event_Comm/Consumer_Input_Handler.cpp

   1 #include "Consumer_Input_Handler.h"
   2 #include "Consumer_Handler.h"
   3
   4 #include "tao/ORB_Core.h"
   5
   6 Consumer_Input_Handler::Consumer_Input_Handler ()
   7   : receiver_handler_ (0),
   8     consumer_initiated_shutdown_ (0)
   9 {
  10   // No-Op.
  11 }
  12
  13 int
  14 Consumer_Input_Handler::consumer_initiated_shutdown ()
  15 {
  16   return this->consumer_initiated_shutdown_;
  17 }
  18
  19 void
  20 Consumer_Input_Handler::consumer_initiated_shutdown (int c)
  21 {
  22   this->consumer_initiated_shutdown_ = c;
  23 }
  24
  25 int
  26 Consumer_Input_Handler::close ()
  27 {
  28   ACE_DEBUG ((LM_DEBUG,
  29               "closing down Consumer::Input_Handler\n"));
  30
  31   Event_Comm::Consumer *receiver =
  32     this->receiver_handler_->receiver ();
  33   Event_Comm::Notifier *notifier =
  34     this->receiver_handler_->notifier ();
  35
  36   if (this->consumer_initiated_shutdown ())
  37     {
  38       // Only try to unsubscribe if the Consumer initiated the
  39       // shutdown.  Otherwise, the Notifier initiated it and it has
  40       // probably gone away by now!
  41       try
  42         {
  43           // Gracefully shutdown the Receiver by removing it from the
  44           // Notifier's internal map.
  45
  46           if (notifier != 0)
  47             notifier->unsubscribe (receiver,
  48                                    "");
  49         }
  50       catch (const CORBA::Exception& ex)
  51         {
  52           ex._tao_print_exception ("Consumer_Input_Handler::handle_close\n");
  53         }
  54     }
  55
  56   // Make sure to cleanup the STDIN handler.
  57   if (ACE_Event_Handler::remove_stdin_handler
  58       (TAO_ORB_Core_instance ()->reactor (),
  59        TAO_ORB_Core_instance ()->thr_mgr ()) == -1)
  60     ACE_ERROR ((LM_ERROR,
  61                 "%p\n",
  62                 "remove_stdin_handler"));
  63
  64   return 0;
  65 }
  66
  67 int Consumer_Input_Handler::initialize (Consumer_Handler *ch)
  68 {
  69   receiver_handler_ = ch;
  70
  71   if (ACE_Event_Handler::register_stdin_handler
  72       (this,
  73        TAO_ORB_Core_instance ()->reactor (),
  74        TAO_ORB_Core_instance ()->thr_mgr ()) == -1)
  75     ACE_ERROR_RETURN ((LM_ERROR,
  76                        "%p\n",
  77                        "register_stdin_handler"),
  78                       -1);
  79   return 0;
  80 }
  81
  82 int
  83 Consumer_Input_Handler::handle_input (ACE_HANDLE h)
  84 {
  85   char buf[BUFSIZ];
  86
  87   // Read up to BUFSIZ worth of data from ACE_HANDLE h.
  88   ssize_t n = ACE_OS::read (h, buf, sizeof buf - 1);
  89
  90   if (n > 0)
  91     {
  92       // Null terminate the buffer, replacing the '\n' with '\0'.
  93       if (buf[n - 1] == '\n')
  94         buf[n - 1] = '\0';
  95       else
  96         buf[n] = '\0';
  97       ACE_DEBUG ((LM_DEBUG,
  98                   "notifying for event %s\n",
  99                   buf));
 100     }
 101   else
 102     {
 103       // If nothing is read, do nothing.
 104       return 0;
 105     }
 106
 107   Event_Comm::Notifier *notifier =
 108     this->receiver_handler_->notifier ();
 109
 110   ACE_ASSERT (notifier != 0);
 111
 112   if (ACE_OS::strncmp (buf, "quit", 4) == 0)
 113     {
 114       // Consumer wants to shutdown.
 115       this->consumer_initiated_shutdown (1);
 116
 117       // Tell the main event loop to shutdown.
 118        this->receiver_handler_->shutdown ();
 119     }
 120   else
 121     {
 122       try
 123         {
 124           Event_Comm::Event event;
 125
 126           event.tag_ = ACE_OS::strdup (buf);
 127
 128           notifier->push (event);
 129         }
 130       catch (const CORBA::Exception& ex)
 131         {
 132           ex._tao_print_exception ("Unexpected exception\n");
 133         }
 134     }
 135
 136   /* NOTREACHED */
 137   return 0;
 138 }