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
  30 #ifdef ACE_HAS_CPP14
  31         server_ = std::make_unique<Server> (my_args.argc (), my_args.argv ());
  32 #else
  33         server_.reset (new Server(my_args.argc (), my_args.argv ()));
  34 #endif
  35
  36         ACE_ASSERT (server_);
  37         initializer = true;
  38       }
  39     }
  40
  41     if (initializer)
  42     {
  43       server_->run (false);
  44       this->force_shutdown (); // servant thread is responsible for shutdown
  45     }
  46     else
  47     {
  48       server_->run (true);
  49     }
  50
  51     return 0;
  52   }
  53
  54   bool ready ()
  55   {
  56     if (server_)
  57     {
  58       return server_->init_;
  59     }
  60
  61     return false;
  62   }
  63
  64   void force_shutdown ()
  65   {
  66     if (server_)
  67     {
  68       server_->shutdown ();
  69     }
  70   }
  71
  72 private:
  73   std::string args_;
  74   std::unique_ptr<Server> server_;
  75   TAO_SYNCH_MUTEX mutex_;
  76 };
  77
  78 #endif //_SERVER_TASK_