1: /*
2: * out_rrd.c RRD Output
3: *
4: * Copyright (c) 2001-2005 Thomas Graf <tgraf@suug.ch>
5: *
6: * Permission is hereby granted, free of charge, to any person obtaining a
7: * copy of this software and associated documentation files (the "Software"),
8: * to deal in the Software without restriction, including without limitation
9: * the rights to use, copy, modify, merge, publish, distribute, sublicense,
10: * and/or sell copies of the Software, and to permit persons to whom the
11: * Software is furnished to do so, subject to the following conditions:
12: *
13: * The above copyright notice and this permission notice shall be included
14: * in all copies or substantial portions of the Software.
15: *
16: * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
17: * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18: * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19: * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20: * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21: * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
22: * DEALINGS IN THE SOFTWARE.
23: */
24:
25: #include <bmon/bmon.h>
26: #include <bmon/node.h>
27: #include <bmon/output.h>
28: #include <bmon/graph.h>
29: #include <bmon/input.h>
30: #include <bmon/utils.h>
31: #include <inttypes.h>
32:
33: #define MAX_RRA 128
34:
35: static const char *c_path = "./";
36: static char *c_step = "1";
37: static char *c_heartbeat = "2";
38: static char *c_rra[MAX_RRA];
39: static int c_rra_index;
40: static int c_update_interval = 1;
41:
42: static void create_rrd(char *file, item_t *it)
43: {
44: char *argv[256];
45: char nows[32];
46: int i, m, ul;
47:
48: memset(argv, 0, sizeof(argv));
49:
50: snprintf(nows, sizeof(nows), "%lld", (unsigned long long) (time(0) - 1));
51:
52: argv[0] = "create";
53: argv[1] = file;
54: argv[2] = "--start";
55: argv[3] = nows;
56: argv[4] = "--step";
57: argv[5] = c_step;
58:
59: i = 6;
60:
61: for (m = 0; m < ATTR_HASH_MAX; m++) {
62: stat_attr_t *a;
63: for (a = it->i_attrs[m]; a; a = a->a_next) {
64: char ds[128];
65: if (i >= (253 - c_rra_index))
66: goto overflow;
67: snprintf(ds, sizeof(ds), "DS:%s_rx:%s:%s:U:U",
68: type2name(a->a_type),
69: a->a_flags & ATTR_FLAG_IS_RATE ? "GAUGE" : "COUNTER",
70: c_heartbeat);
71: argv[i++] = strdup(ds);
72:
73: if (i >= (253 - c_rra_index))
74: goto overflow;
75: snprintf(ds, sizeof(ds), "DS:%s_tx:%s:%s:U:U",
76: type2name(a->a_type),
77: a->a_flags & ATTR_FLAG_IS_RATE ? "GAUGE" : "COUNTER",
78: c_heartbeat);
79: argv[i++] = strdup(ds);
80: }
81: }
82:
83: ul = i;
84:
85: if (i + c_rra_index >= 255)
86: goto overflow;
87:
88: for (m = 0; m < c_rra_index; m++)
89: argv[i++] = c_rra[m];
90:
91: optind = 0; /* What a nice undocumented precondition */
92: opterr = 0;
93:
94: if (rrd_create(i, argv) < 0)
95: fprintf(stderr, "rrd_create failed: %s\n", rrd_get_error());
96:
97: for (i = 6; i < ul && argv[i]; i++)
98: free(argv[i]);
99:
100: return;
101:
102: overflow:
103: quit("Argument overflow, blame the RRD API\n");
104: }
105:
106: static void update_rrd(char *file, item_t *it)
107: {
108: char *argv[6];
109: char template[256];
110: char data[1024];
111: int m;
112:
113: memset(argv, 0, sizeof(argv));
114:
115: argv[0] = "update";
116: argv[1] = file;
117: argv[2] = "--template";
118:
119: memset(template, 0, sizeof(template));
120:
121: for (m = 0; m < ATTR_HASH_MAX; m++) {
122: stat_attr_t *a;
123: for (a = it->i_attrs[m]; a; a = a->a_next) {
124: if (template[0])
125: strncat(template, ":",
126: sizeof(template) - strlen(template) - 1);
127: strncat(template, type2name(a->a_type),
128: sizeof(template) - strlen(template) - 1);
129: strncat(template, "_rx",
130: sizeof(template) - strlen(template) - 1);
131: strncat(template, ":",
132: sizeof(template) - strlen(template) - 1);
133: strncat(template, type2name(a->a_type),
134: sizeof(template) - strlen(template) - 1);
135: strncat(template, "_tx",
136: sizeof(template) - strlen(template) - 1);
137: }
138: }
139:
140: argv[3] = template;
141:
142: snprintf(data, sizeof(data), "%lld",
143: rtiming.rt_last_read.tv_sec);
144:
145: for (m = 0; m < ATTR_HASH_MAX; m++) {
146: stat_attr_t *a;
147: for (a = it->i_attrs[m]; a; a = a->a_next) {
148: char valuepair[64];
149: snprintf(valuepair, sizeof(valuepair),
150: "%s%lld:%lld",
151: data[0] ? ":" : "", attr_get_rx(a),
152: attr_get_tx(a));
153:
154: strncat(data, valuepair,
155: sizeof(data) - strlen(data) - 1);
156: }
157: }
158:
159: argv[4] = data;
160:
161: for (m = 0; m < 5; m++)
162: printf("%s ", argv[m]);
163: printf("\n");
164:
165: optind = 0; /* What a nice undocumented precondition */
166: opterr = 0;
167:
168: rrd_update(5, argv);
169: }
170:
171: static void write_per_item(item_t *it, void *arg)
172: {
173: node_t *node = (node_t *) arg;
174: char file[FILENAME_MAX];
175:
176: snprintf(file, sizeof(file), "%s/%s_%d.rrd", c_path, node->n_name, it->i_index);
177:
178: if (access(file, W_OK) != 0)
179: create_rrd(file, it);
180:
181: if (access(file, W_OK) == 0)
182: update_rrd(file, it);
183: }
184:
185: static void write_per_node(node_t *node, void *arg)
186: {
187: foreach_item(node, write_per_item, (void *) node);
188: }
189:
190: void rrd_draw(void)
191: {
192: static int rem = 1;
193:
194: if (--rem)
195: return;
196: else
197: rem = c_update_interval;
198:
199: foreach_node(write_per_node, NULL);
200: }
201:
202: static void print_module_help(void)
203: {
204: printf(
205: "RRD - RRD Output\n" \
206: "\n" \
207: " Writes updated to RRD databases. Databases are created if needed\n" \
208: " and non-existent\n" \
209: " Author: Thomas Graf <tgraf@suug.ch>\n" \
210: "\n" \
211: " Options:\n" \
212: " path=PATH Output directory\n" \
213: " step=SECS Interval RRD expects updates (default: 1 second)\n" \
214: " heartbeat=SECS Maximum interval until RRD throws away an\n" \
215: " update (default: 2 seconds)\n" \
216: " interval=NUM Update interval in read interval cycles\n" \
217: " rra=RRA_DEF RRA definition (default: RRA:AVERAGE:0.5:1:86400)\n");
218: }
219:
220: static void rrd_set_opts(tv_t *attrs)
221: {
222: static int first_rra = 1;
223:
224: while (attrs) {
225: if (!strcasecmp(attrs->type, "path") && attrs->value)
226: c_path = attrs->value;
227: else if (!strcasecmp(attrs->type, "step") && attrs->value)
228: c_step = attrs->value;
229: else if (!strcasecmp(attrs->type, "heartbeat") && attrs->value)
230: c_heartbeat = attrs->value;
231: else if (!strcasecmp(attrs->type, "interval") && attrs->value)
232: c_update_interval = strtol(attrs->value, NULL, 0);
233: else if (!strcasecmp(attrs->type, "rra") && attrs->value) {
234: if (first_rra) {
235: c_rra_index = 0;
236: first_rra = 0;
237: }
238: if (c_rra_index < MAX_RRA)
239: c_rra[c_rra_index++] = attrs->value;
240: } else if (!strcasecmp(attrs->type, "help")) {
241: print_module_help();
242: exit(0);
243: }
244:
245: attrs = attrs->next;
246: }
247: }
248:
249: static int rrd_probe(void)
250: {
251: return 1;
252: }
253:
254: static struct output_module rrd_ops = {
255: .om_name = "rrd",
256: .om_draw = rrd_draw,
257: .om_set_opts = rrd_set_opts,
258: .om_probe = rrd_probe,
259: };
260:
261: static void __init save_init(void)
262: {
263: c_rra[0] = "RRA:AVERAGE:0.5:1:86400";
264: c_rra_index = 1;
265: register_secondary_output_module(&rrd_ops);
266: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>