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
104
105
106
|
/*
* Copyright (c) 2002-2013 Balabit
* Copyright (c) 1998-2013 Balázs Scheidler
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation; either
* version 2.1 of the License, or (at your option) any later version.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
*
* As an additional exemption you are allowed to compile & link against the
* OpenSSL libraries as published by the OpenSSL project. See the file
* COPYING for details.
*
*/
#include "stats/stats-csv.h"
#include "stats/stats-registry.h"
#include "stats/stats-cluster.h"
#include "utf8utils.h"
#include <string.h>
static gboolean
has_csv_special_character(const gchar *var)
{
if (strchr(var, ';'))
return TRUE;
if (strchr(var, '\"'))
return TRUE;
return FALSE;
}
static gchar *
stats_format_csv_escapevar(const gchar *var)
{
gchar *escaped_result;
if (var[0] && has_csv_special_character(var))
{
gchar *result;
/* quote */
result = convert_unsafe_utf8_to_escaped_binary(var, -1, "\"");
escaped_result = g_strdup_printf("\"%s\"", result);
g_free(result);
}
else
{
escaped_result = convert_unsafe_utf8_to_escaped_binary(var, -1, NULL);
}
return escaped_result;
}
static void
stats_format_csv(StatsCluster *sc, gint type, StatsCounterItem *counter, gpointer user_data)
{
gpointer *args = (gpointer *) user_data;
StatsCSVRecordFunc process_record = (StatsCSVRecordFunc) args[0];
gpointer process_record_arg = args[1];
gchar *s_id, *s_instance, *tag_name;
gchar buf[32];
gchar state;
GString *csv = g_string_sized_new(512);
s_id = stats_format_csv_escapevar(sc->key.legacy.id);
s_instance = stats_format_csv_escapevar(sc->key.legacy.instance);
if (sc->dynamic)
state = 'd';
else if (stats_cluster_is_orphaned(sc))
state = 'o';
else
state = 'a';
tag_name = stats_format_csv_escapevar(stats_cluster_get_type_name(sc, type));
g_string_printf(csv, "%s;%s;%s;%c;%s;%"G_GSIZE_FORMAT"\n",
stats_cluster_get_component_name(sc, buf, sizeof(buf)),
s_id, s_instance, state, tag_name, stats_counter_get(&sc->counter_group.counters[type]));
process_record(csv->str, process_record_arg);
g_string_free(csv, TRUE);
g_free(tag_name);
g_free(s_id);
g_free(s_instance);
}
void
stats_generate_csv(StatsCSVRecordFunc process_record, gpointer user_data, gboolean *cancelled)
{
GString *csv = g_string_sized_new(512);
g_string_printf(csv, "%s;%s;%s;%s;%s;%s\n", "SourceName", "SourceId", "SourceInstance", "State", "Type",
"Number");
process_record(csv->str, user_data);
g_string_free(csv, TRUE);
gpointer format_csv_args[] = {process_record, user_data};
stats_lock();
stats_foreach_legacy_counter(stats_format_csv, format_csv_args, cancelled);
stats_unlock();
}
|
