TAO/tests/Oneway_Send_Timeouts/Server_Task.h

   1 #ifndef _SERVER_TASK_
   2 #define _SERVER_TASK_
   3
   4 #include "Server.h"
   5
   6 #include "ace/ARGV.h"
   7 #include "ace/Task.h"
   8
   9 #include <string>
  10
  11 class Server_Task : public ACE_Task_Base
  12 {
  13 public:
  14   Server_Task (const std::string& args)
  15     : args_ (args)
  16   {}
  17   ~Server_Task () { this->force_shutdown (); }
  18
  19   virtual int svc ()
  20   {
  21     bool initializer = false;
  22     {
  23       ACE_GUARD_RETURN (TAO_SYNCH_MUTEX, ace_mon, this->mutex_, -1);
  24       if (!server_)
  25       {
  26         ACE_ARGV my_args (args_.c_str ());
  27
  28         // Initialize Server ORB in new thread
  29         server_ = std::make_unique<Server> (my_args.argc (), my_args.argv ());
  30
  31         ACE_ASSERT (server_);
  32         initializer = true;
  33       }
  34     }
  35
  36     if (initializer)
  37     {
  38       server_->run (false);
  39       this->force_shutdown (); // servant thread is responsible for shutdown
  40     }
  41     else
  42     {
  43       server_->run (true);
  44     }
  45
  46     return 0;
  47   }
  48
  49   bool ready ()
  50   {
  51     if (server_)
  52     {
  53       return server_->init_;
  54     }
  55
  56     return false;
  57   }
  58
  59   void force_shutdown ()
  60   {
  61     if (server_)
  62     {
  63       server_->shutdown ();
  64     }
  65   }
  66
  67 private:
  68   std::string args_;
  69   std::unique_ptr<Server> server_;
  70   TAO_SYNCH_MUTEX mutex_;
  71 };
  72
  73 #endif //_SERVER_TASK_