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

// $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.start (threads) == -1)
    {
        ACE_ERROR_RETURN ((LM_ERROR, "%p\n", "start"), -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. 

              Note that copy() will advance the wr_ptr() for us.  This means
              we don't have to do it ourselves!  If you do advance it, it
              will be way beyond what you want.
            */
        message->copy ((const char *) &data, 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);
}