TAO/orbsvcs/tests/Notify/lib/Driver.h

   1 /* -*- C++ -*- */
   2 /**
   3  *  @file Driver.h
   4  *
   5  *  @author Pradeep Gore <pradeep@oomworks.com>
   6  */
   7
   8 #ifndef TAO_Notify_Tests_DRIVER_H
   9 #define TAO_Notify_Tests_DRIVER_H
  10 #include /**/ "ace/pre.h"
  11
  12 #include "notify_test_export.h"
  13
  14 #if !defined (ACE_LACKS_PRAGMA_ONCE)
  15 # pragma once
  16 #endif /* ACE_LACKS_PRAGMA_ONCE */
  17
  18 #include "ace/Service_Config.h"
  19 #include "ace/Task.h"
  20
  21 #include "tao/ORB.h"
  22
  23 #include "Driver_Base.h"
  24
  25 class TAO_Notify_Tests_Command_Builder;
  26 class TAO_Notify_Tests_Activation_Manager;
  27
  28 /**
  29  * @class TAO_Notify_Tests_Worker
  30  *
  31  * @brief A Task to execute commands asynchronously.
  32  */
  33 class TAO_Notify_Tests_Worker : public ACE_Task_Base
  34 {
  35 public:
  36   TAO_Notify_Tests_Worker ();
  37   // ctor
  38
  39   /// Set the command builder.
  40   void command_builder (TAO_Notify_Tests_Command_Builder* cmd_builder);
  41
  42   virtual int svc ();
  43   // The thread entry point.
  44
  45   void shutdown ();
  46
  47 private:
  48   /// The command builder
  49   TAO_Notify_Tests_Command_Builder* cmd_builder_;
  50 };
  51
  52
  53 /**
  54  * Run a server thread
  55  *
  56  * Use the ACE_Task_Base class to run server threads
  57  */
  58 class TAO_Notify_Tests_ORB_Run_Worker : public ACE_Task_Base
  59 {
  60 public:
  61   TAO_Notify_Tests_ORB_Run_Worker ();
  62   // ctor
  63
  64   void orb (CORBA::ORB_ptr orb);
  65
  66   /// Srt the run period.
  67   void run_period (ACE_Time_Value run_period);
  68
  69   virtual int svc ();
  70   // The thread entry point.
  71
  72 private:
  73   /// The orb
  74   CORBA::ORB_var orb_;
  75
  76   /// ORB Run Period.
  77   ACE_Time_Value run_period_;
  78 };
  79
  80
  81 /**
  82  * @class TAO_Notify_Tests_Driver
  83  *
  84  * @brief A default Application Starter.
  85  */
  86 class TAO_NOTIFY_TEST_Export TAO_Notify_Tests_Driver : public TAO_Notify_Tests_Driver_Base
  87 {
  88 public:
  89   /// Constructor
  90   TAO_Notify_Tests_Driver ();
  91
  92   /// Destructor
  93   ~TAO_Notify_Tests_Driver ();
  94
  95   /// Init
  96   int init (int argc, ACE_TCHAR *argv[]);
  97
  98   /// Execute the commands.
  99   void run ();
 100
 101   /// Shutdown
 102   virtual void shutdown ();
 103
 104 protected:
 105   /// Parse command line parameters.
 106   int parse_args (int argc, ACE_TCHAR *argv[]);
 107
 108   /// The command builder
 109   TAO_Notify_Tests_Command_Builder* cmd_builder_;
 110
 111   /// Manage activation of periodic suppliers and consumers.
 112   TAO_Notify_Tests_Activation_Manager* activation_manager_;
 113
 114   /// Thread in which to run commands.
 115   TAO_Notify_Tests_Worker worker_;
 116
 117   /// Thread in which to run the orb.
 118   TAO_Notify_Tests_ORB_Run_Worker orb_run_worker_;
 119
 120   /// The ORB we run.
 121   CORBA::ORB_var orb_;
 122
 123   /// ORB Run Period.
 124   ACE_Time_Value run_period_;
 125
 126   /// The file for output
 127   ACE_CString ior_output_file_;
 128
 129   /// The file for input
 130   ACE_CString ior_input_file_;
 131
 132   /// Set to skip the check for multiple priority levels.
 133   int skip_priority_levels_check_;
 134 };
 135
 136 #include /**/ "ace/post.h"
 137 #endif /* TAO_Notify_Tests_DRIVER_H */