summaryrefslogtreecommitdiff
path: root/docs/tutorials/011/message_queue.cpp
blob: 8bd1e0cc13ae4ad219708e4ee8c246a9258627ea (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

// $Id$

/*
   Most of this is the same as the previous tutorial, so I'll just point out
   the differences. 
 */
#include "task.h"
#include "block.h"
#include "data.h"

int run_test (int iterations, int threads)
{
  Task task;

  if (task.open (threads) == -1)
  {
    ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "open"), -1);
  }

ACE_OS::sleep (ACE_Time_Value (1));

  int i;
  for (i = 0; i < iterations; ++i)
  {
    /*
       Construct a Data object that we'll put into the Queue. 
     */
    Data data (i);

    /*
       Create a block large enough for our Data object as well as a text
       message. 
     */
    Block *message = new Block (sizeof (data) + 128);

    /*
       As before, put a text message into the block. 
     */
  ACE_OS::sprintf (message->wr_ptr (), "This is message %d.", i);
    message->wr_ptr (strlen (message->rd_ptr ()));

    *(message->wr_ptr ()) = 0;  // Null-terminate the string we just wrote

    message->wr_ptr (1);        // Move beyond the NULL

    /*
       To copy arbitrary data into a message block, we use the copy() method.
       Since it wants a 'const char*', we have to cast our Data pointer. 
     */
    message->copy ((const char *) &data, sizeof (data));
    message->wr_ptr (sizeof (data));

    if (task.putq (message) == -1)
    {
      break;
    }
  }

  Block *message = new Block ();
message->msg_type (ACE_Message_Block::MB_HANGUP);
  task.putq (message);

  task.wait ();

  return (0);
}

int main (int argc, char *argv[])
{
  int iterations = argc > 1 ? atoi (argv[1]) : 4;
  int threads = argc > 2 ? atoi (argv[2]) : 2;

  (void) run_test (iterations, threads);

  ACE_DEBUG ((LM_DEBUG, "(%P|%t) Application exiting\n"));

  return (0);
}