summaryrefslogtreecommitdiff
path: root/cpp/src/tests/storePerftools/common/TestOptions.cpp
blob: 39e3434a6c7293b8706b3b6a1b029f6a87ddf971 (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
/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
 * KIND, either express or implied.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */

/**
 * \file TestOptions.cpp
 */

#include "TestOptions.h"

namespace tests {
namespace storePerftools {
namespace common {

// static declarations
uint32_t TestOptions::s_defaultNumMsgs = 1024;
uint32_t TestOptions::s_defaultMsgSize = 1024;
uint16_t TestOptions::s_defaultNumQueues = 1;
uint16_t TestOptions::s_defaultEnqThreadsPerQueue = 1;
uint16_t TestOptions::s_defaultDeqThreadsPerQueue = 1;

TestOptions::TestOptions(const std::string& name) :
        qpid::Options(name),
        m_numMsgs(s_defaultNumMsgs),
        m_msgSize(s_defaultMsgSize),
        m_numQueues(s_defaultNumQueues),
        m_numEnqThreadsPerQueue(s_defaultEnqThreadsPerQueue),
        m_numDeqThreadsPerQueue(s_defaultDeqThreadsPerQueue)
{
    doAddOptions();
}

TestOptions::TestOptions(const uint32_t numMsgs,
                         const uint32_t msgSize,
                         const uint16_t numQueues,
                         const uint16_t numEnqThreadsPerQueue,
                         const uint16_t numDeqThreadsPerQueue,
                         const std::string& name) :
        qpid::Options(name),
        m_numMsgs(numMsgs),
        m_msgSize(msgSize),
        m_numQueues(numQueues),
        m_numEnqThreadsPerQueue(numEnqThreadsPerQueue),
        m_numDeqThreadsPerQueue(numDeqThreadsPerQueue)
{
    doAddOptions();
}

TestOptions::~TestOptions()
{}

void
TestOptions::printVals(std::ostream& os) const
{
    os << "TEST OPTIONS:" << std::endl;
    os << "                       Number of queues [-q, --num-queues]: " << m_numQueues << std::endl;
    os << "       Number of producers per queue [-p, --num-producers]: " << m_numEnqThreadsPerQueue << std::endl;
    os << "       Number of consumers per queue [-c, --num-consumers]: " << m_numDeqThreadsPerQueue << std::endl;
    os << "  Number of messages to send per producer [-m, --num-msgs]: " << m_numMsgs << std::endl;
    os << "             Size of each message (bytes) [-s, --msg-size]: " << m_msgSize << std::endl;
}

void
TestOptions::validate()
{
    if (((m_numEnqThreadsPerQueue * m_numMsgs) % m_numDeqThreadsPerQueue) != 0) {
        throw qpid::Exception("Parameter Error: (num-producers * num-msgs) must be a multiple of num-consumers.");
    }
}

void
TestOptions::doAddOptions()
{
    addOptions()
            ("num-queues,q", qpid::optValue(m_numQueues, "N"),
                    "Number of queues")
            ("num-producers,p", qpid::optValue(m_numEnqThreadsPerQueue, "N"),
                    "Number of producers per queue")
            ("num-consumers,c", qpid::optValue(m_numDeqThreadsPerQueue, "N"),
                    "Number of consumers per queue")
            ("num-msgs,m", qpid::optValue(m_numMsgs, "N"),
                    "Number of messages to send per producer")
            ("msg-size,s", qpid::optValue(m_msgSize, "N"),
                    "Size of each message (bytes)")
    ;
}

}}} // namespace tests::storePerftools::common