summaryrefslogtreecommitdiff
path: root/TAO/tests/Big_Request_Muxing/Client_Task.cpp
blob: 22a6f5d9f3b9147e340a62c53f314a9597566d82 (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
//
// $Id$
//

#include "Client_Task.h"
#include "tao/AnyTypeCode/Any.h"

ACE_RCSID(Big_Request_Muxing, Client_Task, "$Id$")

Client_Task::Client_Task (ACE_Thread_Manager *thr_mgr,
                          Test::Payload_Receiver_ptr payload_receiver,
                          CORBA::Long event_count,
                          CORBA::ULong event_size,
                          CORBA::ORB_ptr orb,
                          Messaging::SyncScope sync_scope,
                          const ACE_CString & ident)
  : ACE_Task_Base (thr_mgr)
  , payload_receiver_ (Test::Payload_Receiver::_duplicate (payload_receiver))
  , event_count_ (event_count)
  , event_size_ (event_size)
  , orb_ (CORBA::ORB::_duplicate (orb))
  , sync_scope_ (sync_scope)
  , done_(false)
  , id_ (ident)
{
}

bool
Client_Task::done(void) const
{
  return done_;
}

void
Client_Task::do_invocations(Test::Payload& payload)
{
  ACE_DEBUG((LM_DEBUG, "(%P|%t)Client_Task %s sending %d payloads.\n",
             this->id_.c_str(), this->event_count_));

  for (int i = 0; i != this->event_count_; ++i)
    {
      this->payload_receiver_->more_data (payload);
    }
}

void
Client_Task::do_sync_none_invocations(Test::Payload& payload)
{
  ACE_DEBUG((LM_DEBUG, "(%P|%t)Client_Task %s sending %d SYNC_NONE payloads.\n",
             this->id_.c_str(), this->event_count_));

  for (int i = 0; i != this->event_count_; ++i)
    {
      this->payload_receiver_->sync_none_more_data (payload);
    }
}

int
Client_Task::svc (void)
{
  ACE_DEBUG ((LM_DEBUG,"(%P|%t) Client_Task %s started\n",this->id_.c_str()));
  Test::Payload payload (this->event_size_);
  payload.length (this->event_size_);

  for (CORBA::ULong j = 0; j != payload.length (); ++j)
    payload[j] = (j % 256);

  try
    {
      this->validate_connection ();

      CORBA::Object_var object =
        this->orb_->resolve_initial_references ("PolicyCurrent");
      CORBA::PolicyCurrent_var policy_current =
        CORBA::PolicyCurrent::_narrow (object.in ());

      CORBA::Any scope_as_any;
      scope_as_any <<= this->sync_scope_;


      CORBA::PolicyList policy_list (1);
      policy_list.length (1);
      policy_list[0] =
        this->orb_->create_policy (Messaging::SYNC_SCOPE_POLICY_TYPE,
                                   scope_as_any);

      policy_current->set_policy_overrides (policy_list,
                                            CORBA::ADD_OVERRIDE);

      if (this->sync_scope_ == Messaging::SYNC_NONE)
        this->do_sync_none_invocations(payload);
      else
        this->do_invocations(payload);

    }
  catch (const CORBA::Exception& ex)
    {
      ACE_DEBUG((LM_DEBUG, "(%P|%t)Client_Task %s: ",
                 this->id_.c_str()));
      ex._tao_print_exception ("");
      done_ = true;
      return -1;
    }
  ACE_DEBUG((LM_DEBUG, "(%P|%t)Client_Task %s finished.\n", this->id_.c_str()));
  done_ = true;
  return 0;
}

void
Client_Task::validate_connection (void)
{
  try
    {
      Test::Payload payload(0);
      for (int i = 0; i != 100; ++i)
        {
          (void) this->payload_receiver_->more_data (payload);
        }
    }
  catch (const CORBA::Exception&){}
}