summaryrefslogtreecommitdiff
path: root/TAO/examples/Callback_Quoter/Supplier_i.cpp
blob: 9cf9e0218eff5636cb2ac94bb05af49ebd6ab4a9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
// $Id$

// ===========================================================
//
// = LIBRARY
//    TAO/examples/Callback_Quoter
//
// = FILENAME
//    Supplier_i.cpp
//
// = DESCRIPTION
//   Implementation of the Supplier_i class. This class is the servant
//   object for the callback quoter server.
//
// = AUTHOR
//    Kirthika Parameswaran <kirthika@cs.wustl.edu>
//
// ===========================================================

#include "Supplier_i.h"

Supplier_i::Supplier_i (void)
{
  // No-op
}

Supplier_i::~Supplier_i (void)
{
  // No-op
}

// Register a distributed callback handler that is invoked when the
// given stock reaches the desired threshold value.

void
Supplier_i::register_callback (const char *stock_name,
			       CORBA::Long threshold_value,
			       Callback_Quoter::Consumer_ptr consumer_handler,
			       CORBA::Environment &TAO_TRY_ENV)
{
  // Store the client information.
  Consumer_Data consumer_data;

  // necessary to make another copy of the consumer_handler using <_duplicate>
  // so that we dont lose the consumer object reference after the method
  // invocation is done.
  consumer_data.consumer_ = Callback_Quoter::Consumer::_duplicate (consumer_handler);
  consumer_data.desired_value_= threshold_value;

  CONSUMERS *consumers = 0;

  if (this->consumer_map_.find (stock_name, consumers) == 0)
    {
      consumers->insert (consumer_data);
    }
  else
    {
      consumers = new CONSUMERS;
      consumers->insert (consumer_data);
      this->consumer_map_.bind (stock_name, consumers);
    }
}

void
Supplier_i::orb (CORBA::ORB_ptr orb)
{
  this->orb_ = orb;
}

// Remove the client handler.

void
Supplier_i::unregister_callback (Callback_Quoter::Consumer_ptr consumer_handler,
                                 CORBA::Environment &TAO_TRY_ENV)
{


  for (CONSUMER_MAP::ITERATOR iter = this->consumer_map_.begin ();
       iter!= this->consumer_map_.end ();
       iter ++)
    {

      // the *iter is nothing but the stockname + unbounded set
      // of consumers+threshold values, i.e a ACE_Hash_Map_Entry.

       CONSUMERS *consumers;
       Consumer_Data consumer_to_remove;
       consumer_to_remove.consumer_ = consumer_handle;

       (*iter).consumers->remove (consumer_to_remove);
       // the remove method will do a find internally using
       // operator == which will check only the consumer pointers.
       // If match found it will be removed from the set.

    }

}

// gets the market status and sends the information to
// the consumer who is interested in it.

void
Supplier_i::market_status (const char *stock_name,
                           CORBA::Long stock_value,
                           CORBA::Environment &TAO_TRY_ENV)
{
  ACE_DEBUG ((LM_DEBUG,
	      "Supplier:: The stockname is %s with price %d\n",
	      stock_name,
              stock_value));

  // Go through the list of <Consumer_Data> to find which registered
  // client wants to be notified.

  CONSUMERS *consumers;
  if (this->consumer_map_.find (stock_name, consumers) == 0)
    {
      for (CONSUMERS::ITERATOR iter = consumers->begin ();
           iter != consumers->end ();
           iter++)
        {
          // Check whether the stockname is equal before proceeding further.
          if (stock_value >= (*iter).desired_value_)
            {
              Callback_Quoter::Info interested_consumer_data;

              interested_consumer_data.stock_name = CORBA::string_dup (stock_name);
              interested_consumer_data.value = stock_value;

              (*iter).consumer_->push (interested_consumer_data);
            }
        }
    }
}

void
Supplier_i::shutdown (CORBA::Environment &)
{
  ACE_DEBUG ((LM_DEBUG,
              "\n%s\n",
              "The Callback Quoter server is shutting down"));

  // Instruct the ORB to shutdown.
  this->orb_->shutdown ();
}

int
Supplier_i::Consumer_Data::operator== (const Consumer_Data &rhs)
{
  // The <_is_equivalent> function checks if the _var and _ptr objects
  // are the same.  NOTE: this call might not behave well on other
  // ORBs since <_is_equivalent> isn't guaranteed to differentiate
  // object references.
  return this->consumer_->_is_equivalent (rhs.consumer_.in ());
}