Skip to content
Snippets Groups Projects
Commit b52b4940 authored by Joshua Colp's avatar Joshua Colp Committed by Gerrit Code Review
Browse files

Merge "StatsD: Add sample rate compatibility"

parents 65102d19 05addf3d
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Showing
with 425 additions and 248 deletions
apps/app_statsd.c
+ 425
− 248
View file @ b52b4940
/*
* Asterisk -- An open source telephony toolkit.
*
* Copyright (C) 2015, Digium, Inc.
*
* Tyler Cambron <tcambron@digium.com>
*
* See http://www.asterisk.org for more information about
* the Asterisk project. Please do not directly contact
* any of the maintainers of this project for assistance;
* the project provides a web site, mailing lists and IRC
* channels for your use.
*
* This program is free software, distributed under the terms of
* the GNU General Public License Version 2. See the LICENSE file
* at the top of the source tree.
*/
/*** MODULEINFO
<depend>res_statsd</depend>
<defaultenabled>no</defaultenabled>
<support_level>extended</support_level>
***/
#include "asterisk.h"
ASTERISK_REGISTER_FILE()
#include <math.h>
#include "asterisk/module.h"
#include "asterisk/logger.h"
#include "asterisk/app.h"
#include "asterisk/pbx.h"
#include "asterisk/strings.h"
#include "asterisk/statsd.h"
/*** DOCUMENTATION
<application name="StatsD" language="en_US">
<synopsis>
Allow statistics to be passed to the StatsD server from the dialplan.
</synopsis>
<syntax>
<parameter name="metric_type" required="true">
<para>The metric type to be sent to StatsD.</para>
</parameter>
<parameter name="statistic_name" required="true">
<para>The name of the variable to be sent to StatsD.</para>
</parameter>
<parameter name="value" required="true">
<para>The value of the variable to be sent to StatsD.</para>
</parameter>
</syntax>
<description>
<para>This dialplan application sends statistics to the StatsD
server specified inside of <literal>statsd.conf</literal>.</para>
</description>
</application>
***/
static const char app[] = "StatsD";
/*!
* \brief Check to ensure the value is within the allowed range.
*
* \param value The value of the statistic to be sent to StatsD.
* \param metric The metric type to be sent to StatsD.
*
* This function checks to see if the value given to the StatsD daialplan
* application is within the allowed range as specified by StatsD. A counter
* is the only metric type allowed to be initialized as a negative number.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int value_in_range(const char *value, const char *metric)
{
double numerical_value = strtod(value, NULL);
if (!strcmp(metric, "c")) {
if (numerical_value < pow(-2, 63) || numerical_value > pow(2, 63)) {
ast_log(AST_LOG_WARNING, "Value %lf out of range!\n", numerical_value);
return 1;
}
} else {
if (numerical_value < 0 || numerical_value > pow(2, 64)) {
ast_log(AST_LOG_WARNING, "Value %lf out of range!\n", numerical_value);
return 1;
}
}
return 0;
}
/*!
* \brief Check to ensure the metric type is a valid metric type.
*
* \param metric The metric type to be sent to StatsD.
*
* This function checks to see if the metric type given to the StatsD dialplan
* is a valid metric type. Metric types are determined by StatsD.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_metric(const char *metric)
{
const char *valid_metrics[] = {"g","s","ms","c"};
int i;
if (ast_strlen_zero(metric)) {
ast_log(AST_LOG_ERROR, "Missing metric type argument.\n");
return 1;
}
for (i = 0; i < ARRAY_LEN(valid_metrics); i++) {
if (!strcmp(valid_metrics[i], metric)) {
return 0;
}
}
ast_log(AST_LOG_ERROR, "Invalid metric type %s.\n", metric);
return 1;
}
/*!
* \brief Check to ensure the statistic name is valid.
*
* \param name The variable name to be sent to StatsD.
*
* This function checks to see if the statistic name given to the StatsD
* dialplan application is valid by ensuring that the name does not have any
* invalid characters.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_name(const char *name)
{
if (ast_strlen_zero(name) || (strstr(name, "|") != NULL)) {
ast_log(AST_LOG_ERROR, "Statistic name %s is missing or contains a pipe (|)"
" character.\n", name);
return 1;
}
return 0;
}
/*!
* \brief Check to ensure the value is valid.
*
* \param value The value of the statistic to be sent to StatsD.
* \param metric The metric type to be sent to StatsD.
*
* This function checks to see if the value given to the StatsD daialplan
* application is valid by testing if it is numeric. A plus or minus is only
* allowed at the beginning of the value if it is a counter or a gauge.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_value(const char *value, const char *metric)
{
const char *actual_value;
if (ast_strlen_zero(value)) {
ast_log(AST_LOG_ERROR, "Missing value argument.\n");
return 1;
}
if (!strcmp(metric, "g") || !strcmp(metric, "c")) {
if ((value[0] == '+') || (value[0] == '-')) {
actual_value = &value[1];
if (ast_strlen_zero(actual_value)) {
ast_log(AST_LOG_ERROR, "Value argument %s only contains a sign"
" operator.\n", value);
return 1;
}
} else {
actual_value = &value[0];
}
} else {
actual_value = &value[0];
}
if (!isdigit(*actual_value)) {
ast_log(AST_LOG_ERROR, "Value of %s is not a valid number!\n", actual_value);
return 1;
}
if (value_in_range(actual_value, metric)) {
return 1;
}
return 0;
}
static int statsd_exec(struct ast_channel *chan, const char *data)
{
char *stats;
double numerical_value;
AST_DECLARE_APP_ARGS(args,
AST_APP_ARG(metric_type);
AST_APP_ARG(statistic_name);
AST_APP_ARG(value);
);
if (!data) {
ast_log(AST_LOG_ERROR, "No parameters were provided. Correct format is "
"StatsD(metric_type,statistic_name,value). All parameters are required.\n");
return 1;
}
stats = ast_strdupa(data);
AST_STANDARD_APP_ARGS(args, stats);
/* If any of the validations fail, emit a warning message. */
if (validate_metric(args.metric_type) || validate_name(args.statistic_name)
|| validate_value(args.value, args.metric_type)) {
ast_log(AST_LOG_WARNING, "Invalid parameters provided. Correct format is "
"StatsD(metric_type,statistic_name,value). All parameters are required.\n");
return 1;
}
/*
* Conversion to a double is safe here since the value would have been validated as a
* number in validate_value().
*/
numerical_value = strtod(args.value, NULL);
ast_statsd_log(args.statistic_name, args.metric_type, numerical_value);
return 0;
}
static int unload_module(void)
{
return ast_unregister_application(app);
}
static int load_module(void)
{
return ast_register_application_xml(app, statsd_exec);
}
AST_MODULE_INFO_STANDARD_EXTENDED(ASTERISK_GPL_KEY, "StatsD Dialplan Application");
/*
* Asterisk -- An open source telephony toolkit.
*
* Copyright (C) 2015, Digium, Inc.
*
* Tyler Cambron <tcambron@digium.com>
*
* See http://www.asterisk.org for more information about
* the Asterisk project. Please do not directly contact
* any of the maintainers of this project for assistance;
* the project provides a web site, mailing lists and IRC
* channels for your use.
*
* This program is free software, distributed under the terms of
* the GNU General Public License Version 2. See the LICENSE file
* at the top of the source tree.
*/
/*** MODULEINFO
<depend>res_statsd</depend>
<defaultenabled>no</defaultenabled>
<support_level>extended</support_level>
***/
#include "asterisk.h"
ASTERISK_REGISTER_FILE()
#include <math.h>
#include "asterisk/module.h"
#include "asterisk/logger.h"
#include "asterisk/app.h"
#include "asterisk/pbx.h"
#include "asterisk/strings.h"
#include "asterisk/statsd.h"
/*** DOCUMENTATION
<application name="StatsD" language="en_US">
<synopsis>
Allow statistics to be passed to the StatsD server from the dialplan.
</synopsis>
<syntax>
<parameter name="metric_type" required="true">
<para>The metric type to be sent to StatsD. Valid metric types
are 'g' for gauge, 'c' for counter, 'ms' for timer, and 's' for
sets.</para>
</parameter>
<parameter name="statistic_name" required="true">
<para>The name of the variable to be sent to StatsD. Statistic
names cannot contain the pipe (|) character.</para>
</parameter>
<parameter name="value" required="true">
<para>The value of the variable to be sent to StatsD. Values
must be numeric. Values for gauge and counter metrics can be
sent with a '+' or '-' to update a value after the value has
been initialized. Only counters can be initialized as negative.
Sets can send a string as the value parameter, but the string
cannot contain the pipe character.</para>
</parameter>
<parameter name="sample_rate">
<para>The value of the sample rate to be sent to StatsD. Sample
rates less than or equal to 0 will never be sent and sample rates
greater than or equal to 1 will always be sent. Any rate
between 1 and 0 will be compared to a randomly generated value,
and if it is greater than the random value, it will be sent.</para>
</parameter>
</syntax>
<description>
<para>This dialplan application sends statistics to the StatsD
server specified inside of <literal>statsd.conf</literal>.</para>
</description>
</application>
***/
static const char app[] = "StatsD";
/*!
* \brief Check to ensure the value is within the allowed range.
*
* \param value The value of the statistic to be sent to StatsD.
*
* This function checks to see if the value given to the StatsD daialplan
* application is within the allowed range of [-2^63, 2^63] as specified by StatsD.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int value_in_range(const char *value) {
double numerical_value = strtod(value, NULL);
if (numerical_value < pow(-2, 63) || numerical_value > pow(2, 63)) {
ast_log(AST_LOG_WARNING, "Value %lf out of range!\n", numerical_value);
return 1;
}
return 0;
}
/*!
* \brief Check to ensure the value is within the allowed range.
*
* \param value The value of the statistic to be sent to StatsD.
*
* This function checks to see if the value given to the StatsD daialplan
* application is within the allowed range of [0, 2^64] as specified by StatsD.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int non_neg_value_range(const char *value) {
double numerical_value = strtod(value, NULL);
if (numerical_value < 0 || numerical_value > pow(2, 64)) {
ast_log(AST_LOG_WARNING, "Value %lf out of range!\n", numerical_value);
return 1;
}
return 0;
}
/*!
* \brief Check to ensure the metric type is a valid metric type.
*
* \param metric The metric type to be sent to StatsD.
*
* This function checks to see if the metric type given to the StatsD dialplan
* is a valid metric type. Metric types are determined by StatsD.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_metric(const char *metric)
{
const char *valid_metrics[] = {"g","s","ms","c"};
int i;
if (ast_strlen_zero(metric)) {
ast_log(AST_LOG_ERROR, "Missing metric type argument.\n");
return 1;
}
for (i = 0; i < ARRAY_LEN(valid_metrics); i++) {
if (!strcmp(valid_metrics[i], metric)) {
return 0;
}
}
ast_log(AST_LOG_ERROR, "Invalid metric type %s.\n", metric);
return 1;
}
/*!
* \brief Check to ensure that a numeric value is valid.
*
* \param numeric_value The numeric value to be sent to StatsD.
*
* This function checks to see if a number to be sent to StatsD is actually
* a valid number. One decimal is allowed.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_numeric(const char *numeric_value) {
const char *num;
int decimal_counter = 0;
num = numeric_value;
while (*num) {
if (!isdigit(*num++)) {
if (strstr(numeric_value, ".") != NULL && decimal_counter == 0) {
decimal_counter++;
continue;
}
ast_log(AST_LOG_ERROR, "%s is not a number!\n", numeric_value);
return 1;
}
}
return 0;
}
/*!
* \brief Determines the actual value of a number by looking for a leading + or -.
*
* \param raw_value The entire numeric string to be sent to StatsD.
*
* This function checks to see if the numeric string contains valid characters
* and then isolates the actual number to be sent for validation. Returns the
* result of the numeric validation.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int determine_actual_value(const char *raw_value) {
const char *actual_value;
if ((raw_value[0] == '+') || (raw_value[0] == '-')) {
actual_value = &raw_value[1];
if (ast_strlen_zero(actual_value)) {
ast_log(AST_LOG_ERROR, "Value argument %s only contains a sign"
" operator.\n", raw_value);
return 1;
}
} else {
actual_value = &raw_value[0];
}
return validate_numeric(actual_value);
}
/*!
* \brief Check to ensure the statistic name is valid.
*
* \param name The variable name to be sent to StatsD.
*
* This function checks to see if the statistic name given to the StatsD
* dialplan application is valid by ensuring that the name does not have any
* invalid characters.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_name(const char *name)
{
if (ast_strlen_zero(name) || (strstr(name, "|") != NULL)) {
ast_log(AST_LOG_ERROR, "Statistic name %s is missing or contains a pipe (|)"
" character.\n", name);
return 1;
}
return 0;
}
/*!
* \brief Calls the appropriate functions to validate a gauge metric.
*
* \param statistic_name The statistic name to be sent to StatsD.
* \param value The value to be sent to StatsD.
*
* This function calls other validating functions to correctly validate each
* input based on allowable input for a gauge metric.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_metric_type_gauge(const char *statistic_name, const char *value) {
if (ast_strlen_zero(value)) {
ast_log(AST_LOG_ERROR, "Missing value argument.\n");
return 1;
}
if (validate_name(statistic_name) || determine_actual_value(value)
|| value_in_range(value)) {
return 1;
}
return 0;
}
/*!
* \brief Calls the appropriate functions to validate a counter metric.
*
* \param statistic_name The statistic name to be sent to StatsD.
* \param value The value to be sent to StatsD.
*
* This function calls other validating functions to correctly validate each
* input based on allowable input for a counter metric.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_metric_type_counter(const char *statistic_name, const char *value) {
if (ast_strlen_zero(value)) {
ast_log(AST_LOG_ERROR, "Missing value argument.\n");
return 1;
}
if (validate_name(statistic_name) || determine_actual_value(value)
|| value_in_range(value)) {
return 1;
}
return 0;
}
/*!
* \brief Calls the appropriate functions to validate a timer metric.
*
* \param statistic_name The statistic name to be sent to StatsD.
* \param value The value to be sent to StatsD.
*
* This function calls other validating functions to correctly validate each
* input based on allowable input for a timer metric.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_metric_type_timer(const char *statistic_name, const char *value) {
if (ast_strlen_zero(value)) {
ast_log(AST_LOG_ERROR, "Missing value argument.\n");
return 1;
}
if (validate_name(statistic_name) || validate_numeric(value)
|| non_neg_value_range(value)) {
return 1;
}
return 0;
}
/*!
* \brief Calls the appropriate functions to validate a set metric.
*
* \param statistic_name The statistic name to be sent to StatsD.
* \param value The value to be sent to StatsD.
*
* This function calls other validating functions to correctly validate each
* input based on allowable input for a set metric.
*
* \retval zero on success.
* \retval 1 on error.
*/
static int validate_metric_type_set(const char *statistic_name, const char *value) {
if (ast_strlen_zero(value)) {
ast_log(AST_LOG_ERROR, "Missing value argument.\n");
return 1;
}
if (validate_name(statistic_name)) {
return 1;
}
if (strstr(value, "|") != NULL) {
ast_log(AST_LOG_ERROR, "Pipe (|) character is not allowed for value %s"
" in a set metric.\n", value);
return 1;
}
return 0;
}
static int statsd_exec(struct ast_channel *chan, const char *data)
{
char *stats;
double numerical_rate = 1.0;
AST_DECLARE_APP_ARGS(args,
AST_APP_ARG(metric_type);
AST_APP_ARG(statistic_name);
AST_APP_ARG(value);
AST_APP_ARG(sample_rate);
);
if (!data) {
ast_log(AST_LOG_ERROR, "No parameters were provided. Correct format is "
"StatsD(metric_type,statistic_name,value[,sample_rate]). Sample rate is the "
"only optional parameter.\n");
return 1;
}
stats = ast_strdupa(data);
AST_STANDARD_APP_ARGS(args, stats);
if (validate_metric(args.metric_type)) {
return 1;
}
if (!strcmp(args.metric_type, "g")) {
if (validate_metric_type_gauge(args.statistic_name, args.value)) {
ast_log(AST_LOG_ERROR, "Invalid input for a gauge metric.\n");
return 1;
}
}
else if (!strcmp(args.metric_type, "c")) {
if (validate_metric_type_counter(args.statistic_name, args.value)) {
ast_log(AST_LOG_ERROR, "Invalid input for a counter metric.\n");
return 1;
}
}
else if (!strcmp(args.metric_type, "ms")) {
if (validate_metric_type_timer(args.statistic_name, args.value)) {
ast_log(AST_LOG_ERROR, "Invalid input for a timer metric.\n");
return 1;
}
}
else if (!strcmp(args.metric_type, "s")) {
if (validate_metric_type_set(args.statistic_name, args.value)) {
ast_log(AST_LOG_ERROR, "Invalid input for a set metric.\n");
return 1;
}
}
if (args.sample_rate) {
if (validate_numeric(args.sample_rate)) {
return 1;
}
numerical_rate = strtod(args.sample_rate, NULL);
}
ast_statsd_log_string(args.statistic_name, args.metric_type, args.value,
numerical_rate);
return 0;
}
static int unload_module(void)
{
return ast_unregister_application(app);
}
static int load_module(void)
{
return ast_register_application_xml(app, statsd_exec);
}
AST_MODULE_INFO_STANDARD_EXTENDED(ASTERISK_GPL_KEY, "StatsD Dialplan Application");
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment