TAO/examples/Content_Server/AMI_Observer/client.cpp

   1 // -*- C++ -*-
   2 // Ossama Othman <ossama@uci.edu>
   3
   4 #include "ace/Process_Manager.h"
   5 #include "ace/OS_NS_unistd.h"
   6 #include "orbsvcs/CosNamingC.h"
   7 #include "Push_Web_ServerC.h"
   8 #include "Push_Iterator_Handler.h"
   9
  10 // Obtain reference to Iterator_Factory
  11 Web_Server::Iterator_Factory_ptr
  12 get_iterator (CORBA::ORB_ptr orb);
  13
  14 // Perform file requests
  15 void invoke_requests (int argc,
  16                       ACE_TCHAR *argv[],
  17                       int *request_count,
  18                       Web_Server::Iterator_Factory_ptr f);
  19
  20 int
  21 ACE_TMAIN(int argc, ACE_TCHAR *argv[])
  22 {
  23   try
  24     {
  25       if (argc < 2)
  26         ACE_ERROR_RETURN ((LM_ERROR,
  27                            ACE_TEXT ("Usage: client filename ")
  28                            ACE_TEXT ("[filename ...]\n")),
  29                           -1);
  30
  31       // Initialize the ORB.
  32       CORBA::ORB_var orb = CORBA::ORB_init (argc,
  33                                             argv,
  34                                             "Mighty ORB");
  35
  36       // Get the Root POA.
  37       CORBA::Object_var obj =
  38         orb->resolve_initial_references ("RootPOA");
  39
  40       PortableServer::POA_var poa =
  41         PortableServer::POA::_narrow (obj.in ());
  42
  43       // Activate the POA manager.
  44       PortableServer::POAManager_var mgr = poa->the_POAManager ();
  45       mgr->activate ();
  46
  47       // Get an Iterator_Factory reference.
  48       Web_Server::Iterator_Factory_var factory =
  49         ::get_iterator (orb.in ());
  50
  51       if (CORBA::is_nil (factory.in ()))
  52         {
  53           ACE_ERROR_RETURN ((LM_ERROR,
  54                              ACE_TEXT ("Object pointed to by:\n %s\n")
  55                              ACE_TEXT ("is not an Iterator_Factory")
  56                              ACE_TEXT ("object.\n"),
  57                              argv[1]),
  58                             -1);
  59         }
  60
  61       // Variable used to keep track of when file retrieval has
  62       // completed.
  63       int request_count = 0;
  64
  65       // Activate and run the reply handlers.
  66       ::invoke_requests (argc,
  67                          argv,
  68                          &request_count,
  69                          factory.in ());
  70
  71       // 1 millisecond delay to reduce "busy waiting" in ORB event
  72       // loop.
  73       ACE_Time_Value tv (0, 1000);
  74
  75       // Run the ORB event loop.
  76       while (request_count > 0)
  77         {
  78           CORBA::Boolean more_work;
  79
  80           more_work = orb->work_pending ();
  81
  82           if (more_work)
  83             {
  84               orb->perform_work ();
  85             }
  86           else
  87             ACE_OS::sleep (tv);
  88         }
  89
  90       orb->shutdown (0);
  91
  92       //orb->destroy ();
  93     }
  94   catch (const Web_Server::Error_Result& exc)
  95     {
  96       ACE_ERROR_RETURN ((LM_ERROR,
  97                          ACE_TEXT ("Caught Web Server exception ")
  98                          ACE_TEXT ("status %d\n"),
  99                          exc.status),
 100                         -1);
 101     }
 102   catch (const CORBA::Exception& ex)
 103     {
 104       ex._tao_print_exception (ACE_TEXT ("Caught unexpected exception:"));
 105
 106       return -1;
 107     }
 108
 109   // Wait for all spawned viewers to exit.
 110   ACE_Process_Manager::instance ()->wait ();
 111
 112   return 0;
 113 }
 114
 115 Web_Server::Iterator_Factory_ptr
 116 get_iterator (CORBA::ORB_ptr o)
 117 {
 118   CORBA::ORB_var orb = CORBA::ORB::_duplicate (o);
 119
 120   // Get a reference to the Name Service.
 121   CORBA::Object_var obj =
 122     orb->resolve_initial_references ("NameService");
 123
 124   // Narrow to a Naming Context
 125   CosNaming::NamingContext_var nc =
 126     CosNaming::NamingContext::_narrow (obj.in ());
 127
 128   if (CORBA::is_nil (obj.in ()))
 129     {
 130       ACE_ERROR ((LM_ERROR,
 131                   ACE_TEXT ("Nil reference to Name Service\n")));
 132       return Web_Server::Iterator_Factory::_nil ();
 133     }
 134
 135   // Create a name.
 136   CosNaming::Name name;
 137   name.length (1);
 138   name[0].id = CORBA::string_dup ("Push_Iterator_Factory");
 139   name[0].kind = CORBA::string_dup ("");
 140
 141   obj = nc->resolve (name);
 142
 143   Web_Server::Iterator_Factory_ptr factory =
 144     Web_Server::Iterator_Factory::_narrow (obj.in ());
 145
 146   return factory;
 147 }
 148
 149 void invoke_requests (int argc,
 150                       ACE_TCHAR *argv[],
 151                       int *request_count,
 152                       Web_Server::Iterator_Factory_ptr f)
 153 {
 154   Web_Server::Iterator_Factory_var factory =
 155     Web_Server::Iterator_Factory::_duplicate (f);
 156
 157   // Activate and run the reply handlers.
 158   for (int i = 0;
 159        i < argc - 1;  // Don't include the program name.
 160        ++i)
 161     {
 162       Push_Iterator_Handler *handler = 0;
 163       ACE_NEW_THROW_EX (handler,
 164                         Push_Iterator_Handler,
 165                         CORBA::NO_MEMORY ());
 166
 167       // Transfer ownership to the POA.
 168       PortableServer::ServantBase_var tmp (handler);
 169
 170       // This ends up being an AMI call, so it won't block.
 171       handler->run (request_count,
 172                     ACE_TEXT_ALWAYS_CHAR(argv[i + 1]),
 173                     factory.in ());
 174     }
 175 }