TAO/examples/Simple/chat/Broadcaster_i.cpp

   1 //=============================================================================
   2 /**
   3  *  @file    Broadcaster_i.cpp
   4  *
   5  * Implementation of the Broadcaster_i class. This class is the servant
   6  * object for the chat server.
   7  *
   8  *  @author Pradeep Gore <pradeep@cs.wustl.edu>
   9  */
  10 //=============================================================================
  11
  12 #include "Broadcaster_i.h"
  13
  14 bool
  15 Broadcaster_i::Receiver_Data::operator == (const Broadcaster_i::Receiver_Data &receiver_data) const
  16 {
  17   // The <_is_equivalent> function checks if the _var and _ptr objects
  18   // are the same.  NOTE: this call might not behave well on other
  19   // ORBs since <_is_equivalent> isn't guaranteed to differentiate
  20   // object references.
  21   return
  22     this->receiver_->_is_equivalent (receiver_data.receiver_.in ())
  23     && this->nickname_ == receiver_data.nickname_;
  24 }
  25
  26 void
  27 Broadcaster_i::add (Receiver_ptr receiver,
  28                     const char *nickname)
  29 {
  30   Broadcaster_i::Receiver_Data receiver_data;
  31
  32   // Store the client information.
  33   receiver_data.receiver_ = Receiver::_duplicate (receiver);
  34   receiver_data.nickname_ = nickname;
  35
  36   // Insert the Receiver reference to the set
  37   if (receiver_set_.insert (receiver_data) == -1)
  38     throw Broadcaster::CannotAdd ("failed to add to the receiver set\n");
  39
  40   // Tell everyone which person just joined the chat.
  41   ACE_CString broadcast_string =
  42     ACE_CString ("**** ")
  43     + ACE_CString (nickname)
  44     + ACE_CString (" has joined the chat ****\n");
  45
  46   try
  47     {
  48       this->broadcast (broadcast_string.fast_rep ());
  49     }
  50   catch (const CORBA::Exception& ex)
  51     {
  52       ex._tao_print_exception ("Broadcaster_i::broadcast failed.\t\n");
  53     }
  54 }
  55
  56 void
  57 Broadcaster_i::remove (Receiver_ptr receiver)
  58 {
  59   Broadcaster_i::Receiver_Data receiver_data_to_remove;
  60
  61   // Go through the list of <Receiver_Data> to find which registered client
  62   // wants to be removed.
  63   for (RECEIVER_SET_ITERATOR iter = this->receiver_set_.begin ();
  64        iter != this->receiver_set_.end ();
  65        iter++)
  66     {
  67       // The <_is_equivalent> function checks if the _var and _ptr objects
  68       // are the same.  NOTE: this call might not behave well on other
  69       // ORBs since <_is_equivalent> isn't guaranteed to differentiate
  70       // object references.
  71       if ((*iter).receiver_.in ()->_is_equivalent (receiver) == 1)
  72         {
  73           receiver_data_to_remove = *iter;
  74           break;
  75         }
  76     }
  77
  78   // Remove the reference from our list.
  79   if (this->receiver_set_.remove (receiver_data_to_remove) == -1)
  80     throw Broadcaster::CannotRemove ("failed to remove from receiver set\n");
  81
  82   // Tell everyone, which person left the chat.
  83   ACE_CString broadcast_string = "**** "
  84     + receiver_data_to_remove.nickname_
  85     + " left the chat"
  86     + " ****\n";
  87
  88   this->broadcast (broadcast_string.fast_rep ());
  89 }
  90
  91 void
  92 Broadcaster_i::say (Receiver_ptr receiver,
  93                     const char *text)
  94 {
  95   try
  96     {
  97       ACE_CString sender_nickname ("Sender Unknown");
  98
  99       // Find the nickname for this receiver.
 100
 101       for (RECEIVER_SET_ITERATOR iter = this->receiver_set_.begin ();
 102            iter != this->receiver_set_.end ();
 103            iter++)
 104         {
 105            // The <_is_equivalent> function checks if the _var and
 106            // _ptr objects are the same.  NOTE: this call might not
 107            // behave well on other ORBs since <_is_equivalent> isn't
 108            // guaranteed to differentiate object references.
 109           if ((*iter).receiver_.in ()->_is_equivalent (receiver) == 1)
 110             sender_nickname = (*iter).nickname_;
 111         }
 112
 113       // Broadcast the message to all registered clients
 114       ACE_CString broadcast_string ("[" + sender_nickname + "] " + text);
 115
 116       this->broadcast (broadcast_string.fast_rep ());
 117     }
 118   catch (const CORBA::Exception& ex)
 119     {
 120       ex._tao_print_exception ("Broadcaster_i::say\t\n");
 121     }
 122 }
 123
 124 void
 125 Broadcaster_i::broadcast (const char *text)
 126 {
 127   // Broadcast the message to all registered clients.
 128
 129   for (RECEIVER_SET_ITERATOR iter = this->receiver_set_.begin ();
 130        iter != this->receiver_set_.end ();
 131        iter++)
 132     {
 133       try
 134         {
 135           (*iter).receiver_->message (text);
 136         }
 137       catch (const CORBA::Exception& ex)
 138         {
 139           ex._tao_print_exception ("Failed to send a message\n");
 140         }
 141     }
 142 }
 143