TAO/orbsvcs/tests/ImplRepo/nestea_i.cpp

   1 #include "nestea_i.h"
   2 #include "tao/debug.h"
   3 #include "ace/ACE.h"
   4 #include "ace/FILE_Addr.h"
   5 #include "ace/FILE_Connector.h"
   6 #include "ace/FILE_IO.h"
   7 #include "ace/OS_NS_stdio.h"
   8 #include "ace/OS_NS_string.h"
   9
  10 const size_t MAX_UINT32_STR_LEN = 11;  // Largest UINT32 is 8589934591 + NUL is 11 characters
  11
  12
  13 Nestea_i::Nestea_i (CORBA::ORB_ptr orb, const ACE_TCHAR *filename)
  14 : cans_ (0)
  15 {
  16   orb_ = CORBA::ORB::_duplicate(orb);
  17
  18   this->data_filename_ = ACE::strnew (filename);
  19
  20   // @@ This should probably be called from somewhere else
  21   this->load_data ();
  22 }
  23
  24
  25 Nestea_i::~Nestea_i ()
  26 {
  27   delete [] this->data_filename_;
  28 }
  29
  30
  31 // Add <cans> number of cans to the bookshelf.
  32
  33 void
  34 Nestea_i::drink (CORBA::Long cans)
  35 {
  36   if (TAO_debug_level)
  37     ACE_DEBUG ((LM_DEBUG, "Nestea_i::drink %d cans\n", cans));
  38
  39   this->cans_ += cans;
  40
  41   this->save_data ();
  42 }
  43
  44
  45 // Removes <cans> number of cans from the bookshelf.
  46
  47 void
  48 Nestea_i::crush (CORBA::Long cans)
  49 {
  50   if (TAO_debug_level)
  51     ACE_DEBUG ((LM_DEBUG, "Nestea_i::crush %d cans\n", cans));
  52
  53   if (static_cast<ACE_UINT32> (cans) > this->cans_)
  54     this->cans_ = 0;
  55   else
  56     this->cans_ -= cans;
  57
  58   this->save_data ();
  59 }
  60
  61
  62 // Returns the number of cans in the bookshelf.
  63
  64 CORBA::Long
  65 Nestea_i::bookshelf_size ()
  66 {
  67   if (TAO_debug_level)
  68     ACE_DEBUG ((LM_DEBUG, "Nestea_i::bookshelf_size\n"));
  69
  70   return this->cans_;
  71 }
  72
  73 // Returns comments about your collection.
  74
  75 char *
  76 Nestea_i::get_praise ()
  77 {
  78   if (TAO_debug_level)
  79     ACE_DEBUG ((LM_DEBUG, "Nestea_i::get_praise\n"));
  80
  81   if (this->cans_ > 500)
  82     return CORBA::string_dup ("Man, that is one excellent Nestea Collection!");
  83   else if (this->cans_ > 250)
  84     return CORBA::string_dup ("We are getting into the big leagues now!");
  85   else if (this->cans_ > 100)
  86     return CORBA::string_dup ("Things are looking up!");
  87   else if (this->cans_ > 0)
  88     return CORBA::string_dup ("Well, it is a start. Drink more Nestea!");
  89   else
  90     return CORBA::string_dup ("No cans, no praise.");
  91 }
  92
  93 void
  94 Nestea_i::shutdown ()
  95 {
  96   if (TAO_debug_level)
  97     ACE_DEBUG ((LM_DEBUG, "Nestea_i::shutdown\n"));
  98
  99   orb_->shutdown(0);
 100 }
 101
 102 // Saves bookshelf data to a file.
 103
 104 int
 105 Nestea_i::save_data ()
 106 {
 107   ACE_FILE_IO file;
 108   ACE_FILE_Connector connector;
 109
 110   if (connector.connect (file,
 111                          ACE_FILE_Addr (this->data_filename_),
 112                          0,
 113                          ACE_Addr::sap_any) == -1)
 114     ACE_ERROR_RETURN ((LM_ERROR,
 115                        "%p\n to %s",
 116                        "connect",
 117                        this->data_filename_),
 118                       -1);
 119
 120   char str[MAX_UINT32_STR_LEN];
 121
 122   ACE_OS::sprintf (str, "%d", this->cans_);
 123
 124   return file.send_n (str, ACE_OS::strlen (str) + 1);
 125 }
 126
 127
 128 // Loads bookshelf data from a file.
 129
 130 int
 131 Nestea_i::load_data ()
 132 {
 133   ACE_FILE_IO file;
 134   ACE_FILE_Connector connector;
 135
 136   if (connector.connect (file,
 137                          ACE_FILE_Addr (this->data_filename_),
 138                          0,
 139                          ACE_Addr::sap_any) == -1)
 140     ACE_ERROR_RETURN ((LM_ERROR,
 141                        "%p\n to %s",
 142                        "connect",
 143                        this->data_filename_),
 144                       -1);
 145
 146   char str[MAX_UINT32_STR_LEN];
 147
 148   int len = file.recv (str, MAX_UINT32_STR_LEN);
 149   str[len] = 0;
 150
 151   if (len > 0)
 152     this->cans_ = ACE_OS::atoi (str);
 153   else
 154     this->cans_ = 0;
 155   return 0;
 156 }