[2155] update for structured statistics items

Conflicts:
	src/bin/auth/statistics.cc
	src/bin/auth/tests/auth_srv_unittest.cc
	src/bin/auth/tests/statistics_unittest.cc

diff --git a/src/bin/auth/statistics.cc b/src/bin/auth/statistics.cc
index 89c9d08cea4b87c59b40252809768bd5782a9888..069be746a31d7dcfe297d0a1ef243a80daab917b 100644 (file)
--- a/src/bin/auth/statistics.cc
+++ b/src/bin/auth/statistics.cc
@@ -42,16 +42,23 @@ using namespace isc::statistics;
 namespace {
 
 void
-fillNodes(const Counter& counter, const char* const nodename[],
-          const size_t size,
+fillNodes(const Counter& counter, const struct CounterTypeTree type_tree[],
           isc::auth::statistics::Counters::ItemTreeType& trees)
 {
     using namespace isc::data;
 
-    for (size_t i = 0; i < size; ++i) {
-        trees->set (nodename[i],
-                    Element::create(static_cast<long int>(counter.get(i)))
-                    );
+    for (int i = 0; type_tree[i].name != NULL; ++i) {
+        if (type_tree[i].sub_tree != NULL) {
+            isc::auth::statistics::Counters::ItemTreeType sub_tree =
+                Element::createMap();
+            trees->set(type_tree[i].name, sub_tree);
+            fillNodes(counter, type_tree[i].sub_tree, sub_tree);
+        } else {
+            trees->set(type_tree[i].name,
+                       Element::create(static_cast<long int>(
+                           counter.get(type_tree[i].counter_id)))
+                       );
+        }
     }
 }
 
@@ -63,9 +70,7 @@ namespace statistics {
 
 Counters::Counters() :
     // size of server_qr_counter_, zone_qr_counters_: QR_COUNTER_TYPES
-    // size of server_socket_counter_: SOCKET_COUNTER_TYPES
     server_qr_counter_(QR_COUNTER_TYPES),
-    socket_counter_(SOCKET_COUNTER_TYPES),
     zone_qr_counters_(QR_COUNTER_TYPES)
 {}
 
@@ -231,8 +236,7 @@ Counters::get() const {
     item_tree->set("zones", zones);
 
     Counters::ItemTreeType server = Element::createMap();
-    fillNodes(server_qr_counter_, QRCounterItemName, QR_COUNTER_TYPES,
-              server);
+    fillNodes(server_qr_counter_, QRCounterTree, server);
     zones->set("_SERVER_", server);
 
     return (item_tree);

diff --git a/src/bin/auth/statistics.h b/src/bin/auth/statistics.h
index 7de481366c403a2fff448551c2f7df219ceb34ec..59fb9327e46e457272934b5a5638157340b1525b 100644 (file)
--- a/src/bin/auth/statistics.h
+++ b/src/bin/auth/statistics.h
@@ -206,8 +206,6 @@ class Counters : boost::noncopyable {
 private:
     // counter for query/response
     isc::statistics::Counter server_qr_counter_;
-    // counter for socket
-    isc::statistics::Counter socket_counter_;
     // set of counters for zones
     isc::statistics::CounterDictionary zone_qr_counters_;
     void incRequest(const QRAttributes& qrattrs,

diff --git a/src/bin/auth/tests/auth_srv_unittest.cc b/src/bin/auth/tests/auth_srv_unittest.cc
index fe2f0129769f9bc4f7b98e08199e36bdb67f2f5c..98e17430ceeecc885724b293e8877478a1cb2c6a 100644 (file)
--- a/src/bin/auth/tests/auth_srv_unittest.cc
+++ b/src/bin/auth/tests/auth_srv_unittest.cc
@@ -132,46 +132,40 @@ protected:
     // Checks whether all Rcode counters are set to zero
     void checkAllRcodeCountersZero() const {
         const std::map<std::string, ConstElementPtr>
-            stats_map(server.getStatistics()->mapValue());
+            stats_map(server.getStatistics()->get("zones")->get("_SERVER_")->
+                      get("rcode")->mapValue());
 
-        const std::string rcode_prefix("rcode.");
         for (std::map<std::string, ConstElementPtr>::const_iterator
                  i = stats_map.begin(), e = stats_map.end();
              i != e;
              ++i)
         {
-            if (i->first.compare(0, rcode_prefix.size(), rcode_prefix) == 0) {
-                checkRcodeCounter(i->first, i->second->intValue(), 0);
-            }
+            checkRcodeCounter(i->first, i->second->intValue(), 0);
         }
     }
 
     // Checks whether all Rcode counters are set to zero except the given
     // rcode (it is checked to be set to 'value')
     void checkAllRcodeCountersZeroExcept(const Rcode& rcode, int value) const {
-        std::string target_rcode_name = rcode.toText();
+        // rcode 16 is registered as both BADVERS and BADSIG
+        std::string target_rcode_name =
+            rcode.toText() == "BADVERS" ? "BADSIGVERS" : rcode.toText();
         std::transform(target_rcode_name.begin(), target_rcode_name.end(),
                        target_rcode_name.begin(), ::tolower);
-        // rcode 16 is registered as both BADVERS and BADSIG
-        if (target_rcode_name == "badvers") {
-            target_rcode_name = "badsigvers";
-        }
 
         const std::map<std::string, ConstElementPtr>
-            stats_map(server.getStatistics()->mapValue());
+            stats_map(server.getStatistics()->get("zones")->get("_SERVER_")->
+                      get("rcode")->mapValue());
 
-        const std::string rcode_prefix("rcode.");
         for (std::map<std::string, ConstElementPtr>::const_iterator
                  i = stats_map.begin(), e = stats_map.end();
              i != e;
              ++i)
         {
-            if (i->first.compare(0, rcode_prefix.size(), rcode_prefix) == 0) {
-                if (i->first.compare(rcode_prefix + target_rcode_name) == 0) {
-                    checkRcodeCounter(i->first, i->second->intValue(), value);
-                } else {
-                    checkRcodeCounter(i->first, i->second->intValue(), 0);
-                }
+            if (i->first.compare(target_rcode_name) == 0) {
+                checkRcodeCounter(i->first, i->second->intValue(), value);
+            } else {
+                checkRcodeCounter(i->first, i->second->intValue(), 0);
             }
         }
     }
@@ -441,7 +435,7 @@ TEST_F(AuthSrvTest, TSIGCheckFirst) {
     // TSIG should have failed, and so the per opcode counter shouldn't be
     // incremented.
     ConstElementPtr stats = server.getStatistics();
-    expectCounterItem(stats->get("zones")->get("_SERVER_"), "opcode.other", 0);
+    expectCounterItem(stats->get("zones")->get("_SERVER_")->get("opcode"), "other", 0);
 
     checkAllRcodeCountersZeroExcept(Rcode::NOTAUTH(), 1);
 }
@@ -1072,10 +1066,10 @@ TEST_F(AuthSrvTest, queryCounterUDPNormal) {
     // The counters should be initialized to 0.
     ConstElementPtr stats_init = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_init, "request.udp", 0);
-    expectCounterItem(stats_init, "request.tcp", 0);
-    expectCounterItem(stats_init, "opcode.query", 0);
-    expectCounterItem(stats_init, "rcode.refused", 0);
+    expectCounterItem(stats_init->get("request"), "udp", 0);
+    expectCounterItem(stats_init->get("request"), "tcp", 0);
+    expectCounterItem(stats_init->get("opcode"), "query", 0);
+    expectCounterItem(stats_init->get("rcode"), "refused", 0);
     // Create UDP message and process.
     UnitTestUtil::createRequestMessage(request_message, Opcode::QUERY(),
                                        default_qid, Name("example.com"),
@@ -1089,10 +1083,10 @@ TEST_F(AuthSrvTest, queryCounterUDPNormal) {
     //   request.tcp
     ConstElementPtr stats_after = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_after, "request.udp", 1);
-    expectCounterItem(stats_after, "request.tcp", 0);
-    expectCounterItem(stats_after, "opcode.query", 1);
-    expectCounterItem(stats_after, "rcode.refused", 1);
+    expectCounterItem(stats_after->get("request"), "udp", 1);
+    expectCounterItem(stats_after->get("request"), "tcp", 0);
+    expectCounterItem(stats_after->get("opcode"), "query", 1);
+    expectCounterItem(stats_after->get("rcode"), "refused", 1);
 }
 
 // Submit TCP normal query and check query counter
@@ -1100,10 +1094,10 @@ TEST_F(AuthSrvTest, queryCounterTCPNormal) {
     // The counters should be initialized to 0.
     ConstElementPtr stats_init = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_init, "request.udp", 0);
-    expectCounterItem(stats_init, "request.tcp", 0);
-    expectCounterItem(stats_init, "opcode.query", 0);
-    expectCounterItem(stats_init, "rcode.refused", 0);
+    expectCounterItem(stats_init->get("request"), "udp", 0);
+    expectCounterItem(stats_init->get("request"), "tcp", 0);
+    expectCounterItem(stats_init->get("opcode"), "query", 0);
+    expectCounterItem(stats_init->get("rcode"), "refused", 0);
     // Create TCP message and process.
     UnitTestUtil::createRequestMessage(request_message, Opcode::QUERY(),
                                        default_qid, Name("example.com"),
@@ -1117,10 +1111,10 @@ TEST_F(AuthSrvTest, queryCounterTCPNormal) {
     //   request.udp
     ConstElementPtr stats_after = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_after, "request.udp", 0);
-    expectCounterItem(stats_after, "request.tcp", 1);
-    expectCounterItem(stats_after, "opcode.query", 1);
-    expectCounterItem(stats_after, "rcode.refused", 1);
+    expectCounterItem(stats_after->get("request"), "udp", 0);
+    expectCounterItem(stats_after->get("request"), "tcp", 1);
+    expectCounterItem(stats_after->get("opcode"), "query", 1);
+    expectCounterItem(stats_after->get("rcode"), "refused", 1);
 }
 
 // Submit TCP AXFR query and check query counter
@@ -1128,9 +1122,9 @@ TEST_F(AuthSrvTest, queryCounterTCPAXFR) {
     // The counters should be initialized to 0.
     ConstElementPtr stats_init = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_init, "request.udp", 0);
-    expectCounterItem(stats_init, "request.tcp", 0);
-    expectCounterItem(stats_init, "opcode.query", 0);
+    expectCounterItem(stats_init->get("request"), "udp", 0);
+    expectCounterItem(stats_init->get("request"), "tcp", 0);
+    expectCounterItem(stats_init->get("opcode"), "query", 0);
     UnitTestUtil::createRequestMessage(request_message, opcode, default_qid,
                          Name("example.com"), RRClass::IN(), RRType::AXFR());
     createRequestPacket(request_message, IPPROTO_TCP);
@@ -1146,9 +1140,9 @@ TEST_F(AuthSrvTest, queryCounterTCPAXFR) {
     //   request.udp, response
     ConstElementPtr stats_after = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_after, "request.udp", 0);
-    expectCounterItem(stats_after, "request.tcp", 1);
-    expectCounterItem(stats_after, "opcode.query", 1);
+    expectCounterItem(stats_after->get("request"), "udp", 0);
+    expectCounterItem(stats_after->get("request"), "tcp", 1);
+    expectCounterItem(stats_after->get("opcode"), "query", 1);
 }
 
 // Submit TCP IXFR query and check query counter
@@ -1156,9 +1150,9 @@ TEST_F(AuthSrvTest, queryCounterTCPIXFR) {
     // The counters should be initialized to 0.
     ConstElementPtr stats_init = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_init, "request.udp", 0);
-    expectCounterItem(stats_init, "request.tcp", 0);
-    expectCounterItem(stats_init, "opcode.query", 0);
+    expectCounterItem(stats_init->get("request"), "udp", 0);
+    expectCounterItem(stats_init->get("request"), "tcp", 0);
+    expectCounterItem(stats_init->get("opcode"), "query", 0);
     UnitTestUtil::createRequestMessage(request_message, opcode, default_qid,
                          Name("example.com"), RRClass::IN(), RRType::IXFR());
     createRequestPacket(request_message, IPPROTO_TCP);
@@ -1174,19 +1168,31 @@ TEST_F(AuthSrvTest, queryCounterTCPIXFR) {
     //   request.udp, response
     ConstElementPtr stats_after = server.getStatistics()->
         get("zones")->get("_SERVER_");
-    expectCounterItem(stats_after, "request.udp", 0);
-    expectCounterItem(stats_after, "request.tcp", 1);
-    expectCounterItem(stats_after, "opcode.query", 1);
+    expectCounterItem(stats_after->get("request"), "udp", 0);
+    expectCounterItem(stats_after->get("request"), "tcp", 1);
+    expectCounterItem(stats_after->get("opcode"), "query", 1);
 }
 
 TEST_F(AuthSrvTest, queryCounterOpcodes) {
-    // Check for 0..2, 3(=other), 4..5
-    // The counter should be initialized to 0.
-    for (int i = 0; i < 6; ++i) {
-        // The counter should be initialized to 0.
+    int other_expected = 0;
+    for (int i = 0; i < 16; ++i) {
+        std::string item_name;
+        int expected;
+        if (QROpCodeToQRCounterType[i] == QR_OPCODE_OTHER) {
+            item_name = "OTHER";
+            other_expected += i + 1;
+            expected = other_expected;
+        } else {
+            item_name = Opcode(i).toText();
+            expected = i + 1;
+        }
+        std::transform(item_name.begin(), item_name.end(), item_name.begin(),
+                       ::tolower);
+
+        // The counter should be initialized to expected value.
         expectCounterItem(server.getStatistics()->
-                              get("zones")->get("_SERVER_"),
-                          QRCounterItemName[QROpCodeToQRCounterType[i]], 0);
+                              get("zones")->get("_SERVER_")->get("opcode"),
+                          item_name, expected - (i + 1));
 
         // For each possible opcode, create a request message and send it
         UnitTestUtil::createRequestMessage(request_message, Opcode(i),
@@ -1205,41 +1211,8 @@ TEST_F(AuthSrvTest, queryCounterOpcodes) {
 
         // Confirm the counter.
         expectCounterItem(server.getStatistics()->
-                              get("zones")->get("_SERVER_"),
-                          QRCounterItemName[QROpCodeToQRCounterType[i]],
-                          i + 1);
-    }
-    // Check for 6..15
-    // they are treated as the 'other' opcode
-    // the 'other' opcode counter is 4 at this point
-    int expected = 4;
-    for (int i = 6; i < 16; ++i) {
-        // The counter should be initialized to 0.
-        expectCounterItem(server.getStatistics()->
-                              get("zones")->get("_SERVER_"),
-                          QRCounterItemName[QROpCodeToQRCounterType[i]],
-                          expected);
-
-        // For each possible opcode, create a request message and send it
-        UnitTestUtil::createRequestMessage(request_message, Opcode(i),
-                                           default_qid, Name("example.com"),
-                                           RRClass::IN(), RRType::NS());
-        createRequestPacket(request_message, IPPROTO_UDP);
-
-        // "send" the request once
-        parse_message->clear(Message::PARSE);
-        server.processMessage(*io_message, *parse_message,
-                              *response_obuffer,
-                              &dnsserv);
-
-        // the 'other' opcode counter should be incremented
-        ++expected;
-
-        // Confirm the counter.
-        expectCounterItem(server.getStatistics()->
-                              get("zones")->get("_SERVER_"),
-                          QRCounterItemName[QROpCodeToQRCounterType[i]],
-                          expected);
+                              get("zones")->get("_SERVER_")->get("opcode"),
+                          item_name, expected);
     }
 }
 

diff --git a/src/bin/auth/tests/statistics_unittest.cc b/src/bin/auth/tests/statistics_unittest.cc
index 1269611a824b87ca36307a593c750b59bb792e43..ab84307d02421fe06f691f371061997a63ed9084 100644 (file)
--- a/src/bin/auth/tests/statistics_unittest.cc
+++ b/src/bin/auth/tests/statistics_unittest.cc
@@ -53,12 +53,35 @@ protected:
     Counters counters;
 };
 
+void
+flatten(std::map<std::string, int> &flat_map, const std::string &prefix,
+        const isc::data::ConstElementPtr map_element) {
+    std::map<std::string, ConstElementPtr> map = map_element->mapValue();
+    for (std::map<std::string, ConstElementPtr>::const_iterator
+            i = map.begin(), e = map.end();
+            i != e;
+            ++i)
+    {
+        switch (i->second->getType()) {
+            case isc::data::Element::map:
+                flatten(flat_map, i->first + ".", i->second);
+                break;
+            case isc::data::Element::integer:
+                flat_map[prefix + i->first] = i->second->intValue();
+                break;
+            default:
+                EXPECT_FALSE("Element Parse Error");
+        }
+    }
+}
+
 bool
 checkCountersAllZeroExcept(const isc::data::ConstElementPtr counters,
                            const std::set<std::string>& except_for) {
-    std::map<std::string, ConstElementPtr> stats_map = counters->mapValue();
+    std::map<std::string, int> stats_map;
+    flatten(stats_map, "", counters);
 
-    for (std::map<std::string, ConstElementPtr>::const_iterator
+    for (std::map<std::string, int>::const_iterator
             i = stats_map.begin(), e = stats_map.end();
             i != e;
             ++i)
@@ -67,9 +90,9 @@ checkCountersAllZeroExcept(const isc::data::ConstElementPtr counters,
         if (except_for.count(i->first) != 0) {
             expect = 1;
         }
-        EXPECT_EQ(expect, i->second->intValue()) << "Expected counter "
+        EXPECT_EQ(expect, i->second) << "Expected counter "
             << i->first << " = " << expect << ", actual: "
-            << i->second->intValue();
+            << i->second;
     }
 
     return false;
@@ -104,7 +127,7 @@ TEST_F(CountersTest, incrementNormalQuery) {
     expect_nonzero.insert("request.udp");
     expect_nonzero.insert("request.edns0");
     expect_nonzero.insert("request.dnssec_ok");
-    expect_nonzero.insert("response");
+    expect_nonzero.insert("responses");
     expect_nonzero.insert("qrynoauthans");
     expect_nonzero.insert("rcode.refused");
     expect_nonzero.insert("authqryrej");
@@ -112,20 +135,6 @@ TEST_F(CountersTest, incrementNormalQuery) {
                                expect_nonzero);
 }
 
-TEST_F(CountersTest, getStatistics) {
-    std::map<std::string, ConstElementPtr> stats_map =
-        counters.get()->get("zones")->get("_SERVER_")->mapValue();
-    for (std::map<std::string, ConstElementPtr>::const_iterator
-            i = stats_map.begin(), e = stats_map.end();
-            i != e;
-            ++i)
-    {
-        // item type check
-        EXPECT_NO_THROW(i->second->intValue())
-            << "Item " << i->first << " is not IntElement";
-    }
-}
-
 int
 countTreeElements(const struct CounterTypeTree* tree) {
     int count = 0;