TAO/tests/CSD_Strategy_Tests/TP_Common/ClientTask.cpp

   1 #include "ClientTask.h"
   2 #include "ClientEngine.h"
   3 #include "AppShutdown.h"
   4 #include "ace/SString.h"
   5
   6
   7 ClientTask::ClientTask(bool shutdown_after_done)
   8 : shutdown_after_done_(shutdown_after_done),
   9   failure_count_(0)
  10 {
  11 }
  12
  13
  14 ClientTask::~ClientTask()
  15 {
  16 }
  17
  18
  19 void
  20 ClientTask::add_engine(ClientEngine* engine)
  21 {
  22   // Pass in false so that _add_ref() is called.
  23   ClientEngine_Handle engine_handle(engine,false);
  24   this->engines_.push_back(engine_handle);
  25 }
  26
  27
  28 int
  29 ClientTask::open(void*)
  30 {
  31   size_t num_threads = this->engines_.size();
  32
  33   if (num_threads == 0)
  34     {
  35       ACE_ERROR_RETURN((LM_ERROR,
  36                         "(%P|%t) ClientTask cannot activate 0 threads.\n"),
  37                        -1);
  38     }
  39
  40   if (this->activate(THR_NEW_LWP | THR_JOINABLE, num_threads) != 0)
  41     {
  42       // Assumes that when activate returns non-zero return code that
  43       // no threads were activated.
  44       ACE_ERROR_RETURN((LM_ERROR,
  45                         "(%P|%t) ClientTask failed to activate "
  46                         "the %d client threads.\n", num_threads),
  47                        -1);
  48     }
  49
  50   return 0;
  51 }
  52
  53
  54 int
  55 ClientTask::svc()
  56 {
  57   ClientEngine_Handle engine;
  58
  59   {
  60     ACE_GUARD_RETURN (TAO_SYNCH_MUTEX, guard, this->lock_, 0);
  61     this->engines_.get(engine, this->engines_.size() - 1);
  62     this->engines_.pop_back();
  63   }
  64
  65   try
  66   {
  67     bool exec_ret = engine->execute();
  68     if (exec_ret == false)
  69       {
  70         ACE_GUARD_RETURN (TAO_SYNCH_MUTEX, guard, this->lock_, 0);
  71         this->failure_count_ ++;
  72       }
  73   }
  74   catch (const CORBA::Exception& ex)
  75   {
  76     ex._tao_print_exception (
  77       "ClientTask::svc Caught exception from execute():");
  78
  79     ACE_GUARD_RETURN (TAO_SYNCH_MUTEX, guard, this->lock_, 0);
  80     this->failure_count_ ++;
  81   }
  82   catch (...)
  83   {
  84     ACE_ERROR((LM_ERROR,
  85                "(%P|%t) ClientTask::svc caught unknown (...) exception "\
  86                "in execute() " ));
  87     ACE_GUARD_RETURN (TAO_SYNCH_MUTEX, guard, this->lock_, 0);
  88     this->failure_count_ ++;
  89   }
  90
  91   if(this->shutdown_after_done_)
  92     {
  93       // This is used to shutdown orb for a client application
  94       // with an orb running.
  95       TheAppShutdown->client_done ();
  96     }
  97
  98   return 0;
  99 }
 100
 101 int
 102 ClientTask::close(u_long)
 103 {
 104   return 0;
 105 }
 106
 107 unsigned
 108 ClientTask::failure_count () const
 109 {
 110   return this->failure_count_;
 111 }