pdns/statnode.hh

   1 /*
   2  * This file is part of PowerDNS or dnsdist.
   3  * Copyright -- PowerDNS.COM B.V. and its contributors
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of version 2 of the GNU General Public License as
   7  * published by the Free Software Foundation.
   8  *
   9  * In addition, for the avoidance of any doubt, permission is granted to
  10  * link this program with OpenSSL and to (re)distribute the binaries
  11  * produced as the result of such linking.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  *
  18  * You should have received a copy of the GNU General Public License
  19  * along with this program; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  21  */
  22 #pragma once
  23 #include "dnsname.hh"
  24 #include <deque>
  25 #include <map>
  26 #include "iputils.hh"
  27
  28 class StatNode
  29 {
  30 public:
  31
  32   struct Stat
  33   {
  34     Stat() : queries(0), noerrors(0), nxdomains(0), servfails(0), drops(0){}
  35     uint64_t queries, noerrors, nxdomains, servfails, drops;
  36
  37     Stat& operator+=(const Stat& rhs) {
  38       queries+=rhs.queries;
  39       noerrors+=rhs.noerrors;
  40       nxdomains+=rhs.nxdomains;
  41       servfails+=rhs.servfails;
  42       drops+=rhs.drops;
  43
  44       for(const remotes_t::value_type& rem : rhs.remotes) {
  45         remotes[rem.first]+=rem.second;
  46       }
  47       return *this;
  48     }
  49     typedef std::map<ComboAddress,int,ComboAddress::addressOnlyLessThan> remotes_t;
  50     remotes_t remotes;
  51   };
  52
  53   Stat s;
  54   std::string name;
  55   std::string fullname;
  56   unsigned int labelsCount{0};
  57
  58   void submit(const DNSName& domain, int rcode, boost::optional<const ComboAddress&> remote);
  59
  60   Stat print(unsigned int depth=0, Stat newstat=Stat(), bool silent=false) const;
  61   typedef boost::function<void(const StatNode*, const Stat& selfstat, const Stat& childstat)> visitor_t;
  62   void visit(visitor_t visitor, Stat& newstat, unsigned int depth=0) const;
  63   bool empty() const
  64   {
  65     return children.empty() && s.remotes.empty();
  66   }
  67   typedef std::map<std::string,StatNode, CIStringCompare> children_t;
  68   children_t children;
  69
  70 private:
  71   void submit(std::vector<string>::const_iterator end, std::vector<string>::const_iterator begin, const std::string& domain, int rcode, boost::optional<const ComboAddress&> remote, unsigned int count);
  72 };