~drizzle-trunk/drizzle/development

2131.7.1 by Andrew Hutchings
Add protocol counters table
1
/* - mode: c; c-basic-offset: 2; indent-tabs-mode: nil; -*-
2
 *  vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
3
 *
4
 *  Copyright (C) 2011 Andrew Hutchings
5
 *
6
 *  This program is free software; you can redistribute it and/or modify
7
 *  it under the terms of the GNU General Public License as published by
8
 *  the Free Software Foundation; either version 2 of the License, or
9
 *  (at your option) any later version.
10
 *
11
 *  This program is distributed in the hope that it will be useful,
12
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
13
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14
 *  GNU General Public License for more details.
15
 *
16
 *  You should have received a copy of the GNU General Public License
17
 *  along with this program; if not, write to the Free Software
18
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
19
 */
20
2173.2.1 by Monty Taylor
Fixes incorrect usage of include
21
#include <config.h>
22
#include <plugin/protocol_dictionary/dictionary.h>
23
#include <drizzled/plugin/listen.h>
2131.7.1 by Andrew Hutchings
Add protocol counters table
24
25
using namespace drizzled;
26
27
ProtocolTool::ProtocolTool() :
28
  plugin::TableFunction("DATA_DICTIONARY", "PROTOCOL_COUNTERS")
29
{
30
  add_field("PROTOCOL");
31
  add_field("COUNTER");
32
  add_field("VALUE", plugin::TableFunction::NUMBER, 0, false);
33
}
34
35
ProtocolTool::Generator::Generator(Field **arg) :
36
  plugin::TableFunction::Generator(arg)
37
{
38
  protocol_it= plugin::Listen::getListenProtocols().begin();
39
  protocol= *protocol_it;
2131.7.5 by Andrew Hutchings
Fix compile error in Solaris
40
  counter_it= protocol->getListenCounters().begin();
2131.7.1 by Andrew Hutchings
Add protocol counters table
41
}
42
43
bool ProtocolTool::Generator::populate()
44
{
45
  if (protocol_it == plugin::Listen::getListenProtocols().end())
46
    return false;
47
48
  protocol= *protocol_it;
49
2131.7.5 by Andrew Hutchings
Fix compile error in Solaris
50
  while (counter_it == protocol->getListenCounters().end())
2131.7.1 by Andrew Hutchings
Add protocol counters table
51
  {
52
    protocol_it++;
53
    if (protocol_it == plugin::Listen::getListenProtocols().end())
54
      return false;
55
    protocol= *protocol_it;
2131.7.5 by Andrew Hutchings
Fix compile error in Solaris
56
    counter_it= protocol->getListenCounters().begin();
2131.7.1 by Andrew Hutchings
Add protocol counters table
57
  }
58
59
  fill();
60
  counter_it++;
61
  return true;
62
}
63
64
void ProtocolTool::Generator::fill()
65
{
66
  protocol= *protocol_it;
67
  counter= *counter_it;
68
69
  push(protocol->getName());
70
  push(*counter->first);
71
  push((uint64_t) *counter->second);
72
}
73
74
static ProtocolTool *protocols;
75
76
static int init(drizzled::module::Context &context)
77
{
2246.3.1 by Olaf van der Spek
Remove std::nothrow from new()
78
  protocols= new ProtocolTool;
79
	context.add(protocols);
2131.7.1 by Andrew Hutchings
Add protocol counters table
80
  return 0;
81
}
82
83
DRIZZLE_DECLARE_PLUGIN
84
{
85
  DRIZZLE_VERSION_ID,
86
  "protocol_dictionary",
87
  "1.0",
88
  "Andrew Hutchings",
89
  "Provides dictionary for protocol counters.",
90
  PLUGIN_LICENSE_GPL,
91
  init,     /* Plugin Init */
92
  NULL,               /* depends */
93
  NULL                /* config options   */
94
}
95
DRIZZLE_DECLARE_PLUGIN_END;
96