summaryrefslogtreecommitdiffstats
path: root/web/api/exporters/allmetrics.c
blob: 17e5e077ebc79c0560cd8b9946a8d65186c1bd61 (plain)
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
107
108
109
110
111
112
113
114
115
116
117
// SPDX-License-Identifier: GPL-3.0-or-later

#include "allmetrics.h"

struct prometheus_output_options {
    char *name;
    PROMETHEUS_OUTPUT_OPTIONS flag;
} prometheus_output_flags_root[] = {
        { "help",       PROMETHEUS_OUTPUT_HELP       },
        { "types",      PROMETHEUS_OUTPUT_TYPES      },
        { "names",      PROMETHEUS_OUTPUT_NAMES      },
        { "timestamps", PROMETHEUS_OUTPUT_TIMESTAMPS },
        { "variables",  PROMETHEUS_OUTPUT_VARIABLES  },
        { "oldunits",   PROMETHEUS_OUTPUT_OLDUNITS   },
        { "hideunits",  PROMETHEUS_OUTPUT_HIDEUNITS  },
        // terminator
        { NULL, PROMETHEUS_OUTPUT_NONE },
};

inline int web_client_api_request_v1_allmetrics(RRDHOST *host, struct web_client *w, char *url) {
    (void)url;
    int format = ALLMETRICS_SHELL;
    const char *prometheus_server = w->client_ip;
    uint32_t prometheus_backend_options = global_backend_options;
    PROMETHEUS_OUTPUT_OPTIONS prometheus_output_options = PROMETHEUS_OUTPUT_TIMESTAMPS | ((global_backend_options & BACKEND_OPTION_SEND_NAMES)?PROMETHEUS_OUTPUT_NAMES:0);
    const char *prometheus_prefix = global_backend_prefix;

    uint32_t end = w->total_params;
    if (end) {
        uint32_t i = 0;
        do {
            char *name = w->param_name[i].body;
            size_t lname = w->param_name[i].length;
            char *value = w->param_values[i].body;
            size_t lvalue = w->param_values[i].length;

            if(!strncmp(name, "format",lname)) {
                if(!strncmp(value, ALLMETRICS_FORMAT_SHELL,lvalue))
                    format = ALLMETRICS_SHELL;
                else if(!strncmp(value, ALLMETRICS_FORMAT_PROMETHEUS,lvalue))
                    format = ALLMETRICS_PROMETHEUS;
                else if(!strncmp(value, ALLMETRICS_FORMAT_PROMETHEUS_ALL_HOSTS,lvalue))
                    format = ALLMETRICS_PROMETHEUS_ALL_HOSTS;
                else if(!strncmp(value, ALLMETRICS_FORMAT_JSON,lvalue))
                    format = ALLMETRICS_JSON;
                else
                    format = 0;
            }
            else if(!strncmp(name, "server",lname)) {
                prometheus_server = value;
            }
            else if(!strncmp(name, "prefix",lname)) {
                prometheus_prefix = value;
            }
            else if(!strncmp(name, "data",lname) || !strncmp(name, "source",lname) || !strncmp(name, "data source",lname) || !strncmp(name, "data-source",lname) || !strncmp(name, "data_source",lname) || !strncmp(name, "datasource",lname)) {
                prometheus_backend_options = backend_parse_data_source(value, prometheus_backend_options);
            }
            else {
                int i;
                for(i = 0; prometheus_output_flags_root[i].name ; i++) {
                    if(!strncmp(name, prometheus_output_flags_root[i].name,lname)) {
                        if(!strncmp(value, "yes",lvalue) || !strncmp(value, "1",lvalue) || !strncmp(value, "true",lvalue))
                            prometheus_output_options |= prometheus_output_flags_root[i].flag;
                        else
                            prometheus_output_options &= ~prometheus_output_flags_root[i].flag;

                        break;
                    }
                }
            }
        } while( ++i < end);
    }

    buffer_flush(w->response.data);
    buffer_no_cacheable(w->response.data);

    switch(format) {
        case ALLMETRICS_JSON:
            w->response.data->contenttype = CT_APPLICATION_JSON;
            rrd_stats_api_v1_charts_allmetrics_json(host, w->response.data);
            return 200;

        case ALLMETRICS_SHELL:
            w->response.data->contenttype = CT_TEXT_PLAIN;
            rrd_stats_api_v1_charts_allmetrics_shell(host, w->response.data);
            return 200;

        case ALLMETRICS_PROMETHEUS:
            w->response.data->contenttype = CT_PROMETHEUS;
            rrd_stats_api_v1_charts_allmetrics_prometheus_single_host(
                    host
                    , w->response.data
                    , prometheus_server
                    , prometheus_prefix
                    , prometheus_backend_options
                    , prometheus_output_options
            );
            return 200;

        case ALLMETRICS_PROMETHEUS_ALL_HOSTS:
            w->response.data->contenttype = CT_PROMETHEUS;
            rrd_stats_api_v1_charts_allmetrics_prometheus_all_hosts(
                    host
                    , w->response.data
                    , prometheus_server
                    , prometheus_prefix
                    , prometheus_backend_options
                    , prometheus_output_options
            );
            return 200;

        default:
            w->response.data->contenttype = CT_TEXT_PLAIN;
            buffer_strcat(w->response.data, "Which format? '" ALLMETRICS_FORMAT_SHELL "', '" ALLMETRICS_FORMAT_PROMETHEUS "', '" ALLMETRICS_FORMAT_PROMETHEUS_ALL_HOSTS "' and '" ALLMETRICS_FORMAT_JSON "' are currently supported.");
            return 400;
    }
}