[asterisk-commits] dlee: branch dlee/stasis-demo r385966 - in /team/dlee/stasis-demo: configs/ i...

SVN commits to the Asterisk project asterisk-commits at lists.digium.com
Wed Apr 17 10:17:02 CDT 2013


Author: dlee
Date: Wed Apr 17 10:16:58 2013
New Revision: 385966

URL: http://svnview.digium.com/svn/asterisk?view=rev&rev=385966
Log:
Simple example of how to use Stasis.

Adds res_statsd, so Asterisk can publish stats to a statsd server.
And a quick-and-dirty module listening to the channel topic and
publishing basic channel stats.

Added:
    team/dlee/stasis-demo/configs/statsd.conf.sample   (with props)
    team/dlee/stasis-demo/include/asterisk/statsd.h   (with props)
    team/dlee/stasis-demo/res/res_chan_stats.c   (with props)
    team/dlee/stasis-demo/res/res_statsd.c   (with props)

Added: team/dlee/stasis-demo/configs/statsd.conf.sample
URL: http://svnview.digium.com/svn/asterisk/team/dlee/stasis-demo/configs/statsd.conf.sample?view=auto&rev=385966
==============================================================================
--- team/dlee/stasis-demo/configs/statsd.conf.sample (added)
+++ team/dlee/stasis-demo/configs/statsd.conf.sample Wed Apr 17 10:16:58 2013
@@ -1,0 +1,6 @@
+[general]
+;enabled = yes			; When set to no, statsd support is disabled
+;server = localhost:8125	; server[:port] of statsd server to use
+;prefix = 			; Prefix to prepend to all metrics
+;add_newline = no		; Append a newline to every event
+	       			; This is useful if you want to fake out a server using netcat (nc -lu 8125)

Propchange: team/dlee/stasis-demo/configs/statsd.conf.sample
------------------------------------------------------------------------------
    svn:eol-style = native

Propchange: team/dlee/stasis-demo/configs/statsd.conf.sample
------------------------------------------------------------------------------
    svn:keywords = Author Date Id Revision

Propchange: team/dlee/stasis-demo/configs/statsd.conf.sample
------------------------------------------------------------------------------
    svn:mime-type = text/plain

Added: team/dlee/stasis-demo/include/asterisk/statsd.h
URL: http://svnview.digium.com/svn/asterisk/team/dlee/stasis-demo/include/asterisk/statsd.h?view=auto&rev=385966
==============================================================================
--- team/dlee/stasis-demo/include/asterisk/statsd.h (added)
+++ team/dlee/stasis-demo/include/asterisk/statsd.h Wed Apr 17 10:16:58 2013
@@ -1,0 +1,88 @@
+/*
+ * Asterisk -- An open source telephony toolkit.
+ *
+ * Copyright (C) 2013, Digium, Inc.
+ *
+ * David M. Lee, II <dlee at 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.
+ */
+
+#ifndef _ASTERISK_STATSD_H
+#define _ASTERISK_STATSD_H
+
+/*!
+ * \brief Support for publishing to a statsd server.
+ *
+ * \author David M. Lee, II <dlee at digium.com>
+ * \since 12
+ */
+
+#include "asterisk/optional_api.h"
+
+/*! Types of measurements */
+enum ast_statsd_type {
+	/*! An instantaneous measurement of a value. */
+	AST_STATSD_GUAGE,
+	/*! A change in a value. */
+	AST_STATSD_COUNTER,
+	/*! Measure of milliseconds. */
+	AST_STATSD_TIMER,
+	/*! Distribution of values over time. */
+	AST_STATSD_HISTOGRAM,
+	/*! Events over time. Sorta like increment-only counters. */
+	AST_STATSD_METER,
+};
+
+/*!
+ * \brief Send a stat to the configured statsd server.
+ *
+ * The is the most flexible function for sending a message to the statsd server,
+ * but also the least easy to use. See ast_statsd_log() or
+ * ast_statsd_log_sample() for a slightly more convenient interface.
+ *
+ * \param metric_name String (UTF-8) name of the metric.
+ * \param type_str Type of metric to send.
+ * \param value Value to send.
+ * \param sample_rate Percentage of samples to send.
+ * \since 12
+ */
+AST_OPTIONAL_API(void, ast_statsd_log_full, (const char *metric_name,
+	const char *type_str, intmax_t value, double sample_rate), {});
+
+/*!
+ * \brief Send a stat to the configured statsd server.
+ * \param metric_name String (UTF-8) name of the metric.
+ * \param metric_type Type of metric to send.
+ * \param value Value to send.
+ * \since 12
+ */
+AST_OPTIONAL_API(void, ast_statsd_log, (const char *metric_name,
+	enum ast_statsd_type metric_type, intmax_t value), {});
+
+/*!
+ * \brief Send a random sampling of a stat to the configured statsd server.
+ *
+ * The type of sampled metrics is always \ref AST_STATSD_COUNTER. The given
+ * \a sample_rate should be a percentage between 0.0 and 1.0. If it's <= 0.0,
+ * then no samples will be sent. If it's >= 1.0, then all samples will be sent.
+ *
+ * \param metric_name String (UTF-8) name of the metric.
+ * \param value Value to send.
+ * \param sample_rate Percentage of samples to send.
+ * \since 12
+ */
+AST_OPTIONAL_API(void, ast_statsd_log_sample, (const char *metric_name,
+		intmax_t value, double sample_rate), {});
+
+
+#endif /* _ASTERISK_STATSD_H */
+

Propchange: team/dlee/stasis-demo/include/asterisk/statsd.h
------------------------------------------------------------------------------
    svn:eol-style = native

Propchange: team/dlee/stasis-demo/include/asterisk/statsd.h
------------------------------------------------------------------------------
    svn:keywords = Author Date Id Revision

Propchange: team/dlee/stasis-demo/include/asterisk/statsd.h
------------------------------------------------------------------------------
    svn:mime-type = text/plain

Added: team/dlee/stasis-demo/res/res_chan_stats.c
URL: http://svnview.digium.com/svn/asterisk/team/dlee/stasis-demo/res/res_chan_stats.c?view=auto&rev=385966
==============================================================================
--- team/dlee/stasis-demo/res/res_chan_stats.c (added)
+++ team/dlee/stasis-demo/res/res_chan_stats.c Wed Apr 17 10:16:58 2013
@@ -1,0 +1,104 @@
+/*
+ * Asterisk -- An open source telephony toolkit.
+ *
+ * Copyright (C) 2013, Digium, Inc.
+ *
+ * David M. Lee, II <dlee at 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.
+ */
+
+/*!
+ * \brief Statsd channel stats. Exmaple of how to subscribe to Stasis events.
+ *
+ * This module subscribes to the channel caching topic and issues statsd stats
+ * based on the received messages.
+ *
+ * \author David M. Lee, II <dlee at digium.com>
+ * \since 12
+ */
+
+/*** MODULEINFO
+	<depend>res_statsd</depend>
+	<defaultenabled>no</defaultenabled>
+	<support_level>extended</support_level>
+ ***/
+
+#include "asterisk.h"
+
+ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+
+#include "asterisk/module.h"
+#include "asterisk/stasis_channels.h"
+#include "asterisk/statsd.h"
+#include "asterisk/time.h"
+
+static struct stasis_subscription *sub;
+
+static void statsmaker(void *data, struct stasis_subscription *sub,
+	struct stasis_topic *topic, struct stasis_message *message)
+{
+	RAII_VAR(struct ast_str *, metric, NULL, ast_free);
+
+	/* For no good reason, count message types */
+	metric = ast_str_create(80);
+	if (metric) {
+		ast_str_set(&metric, 0, "stasis.message.%s",
+			stasis_message_type_name(stasis_message_type(message)));
+		ast_statsd_log(ast_str_buffer(metric), AST_STATSD_METER, 1);
+	}
+
+	if (stasis_cache_update_type() == stasis_message_type(message)) {
+		struct stasis_cache_update *update = stasis_message_data(message);
+
+		if (ast_channel_snapshot_type() != update->type) {
+			return;
+		}
+		if (!update->old_snapshot && update->new_snapshot) {
+			ast_statsd_log("channels.count", AST_STATSD_COUNTER, 1);
+		} else if (update->old_snapshot && !update->new_snapshot) {
+			struct ast_channel_snapshot *last;
+			int64_t age;
+
+			last = stasis_message_data(update->old_snapshot);
+			age = ast_tvdiff_ms(*stasis_message_timestamp(message),
+				last->creationtime);
+			ast_statsd_log("channels.count", AST_STATSD_COUNTER, -1);
+			ast_statsd_log("channels.calltime", AST_STATSD_TIMER, age);
+		}
+	}
+}
+
+static int load_module(void)
+{
+	sub = stasis_subscribe(
+		stasis_caching_get_topic(ast_channel_topic_all_cached()),
+		statsmaker, NULL);
+	if (!sub) {
+		return AST_MODULE_LOAD_FAILURE;
+	}
+	return AST_MODULE_LOAD_SUCCESS;
+}
+
+static int unload_module(void)
+{
+	stasis_unsubscribe(sub);
+	sub = NULL;
+	return 0;
+}
+
+AST_MODULE_INFO(ASTERISK_GPL_KEY,
+	AST_MODFLAG_DEFAULT,
+	"Example of how to use Stasis",
+	.load = load_module,
+	.unload = unload_module,
+	.nonoptreq = "res_statsd"
+	);

Propchange: team/dlee/stasis-demo/res/res_chan_stats.c
------------------------------------------------------------------------------
    svn:eol-style = native

Propchange: team/dlee/stasis-demo/res/res_chan_stats.c
------------------------------------------------------------------------------
    svn:keywords = Author Date Id Revision

Propchange: team/dlee/stasis-demo/res/res_chan_stats.c
------------------------------------------------------------------------------
    svn:mime-type = text/plain

Added: team/dlee/stasis-demo/res/res_statsd.c
URL: http://svnview.digium.com/svn/asterisk/team/dlee/stasis-demo/res/res_statsd.c?view=auto&rev=385966
==============================================================================
--- team/dlee/stasis-demo/res/res_statsd.c (added)
+++ team/dlee/stasis-demo/res/res_statsd.c Wed Apr 17 10:16:58 2013
@@ -1,0 +1,349 @@
+/*
+ * Asterisk -- An open source telephony toolkit.
+ *
+ * Copyright (C) 2013, Digium, Inc.
+ *
+ * David M. Lee, II <dlee at 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.
+ */
+
+/*!
+ * \brief Support for publishing to a statsd server.
+ *
+ * \author David M. Lee, II <dlee at digium.com>
+ * \since 12
+ */
+
+/*** MODULEINFO
+	<support_level>extended</support_level>
+ ***/
+
+/*** DOCUMENTATION
+	<configInfo name="res_statsd" language="en_US">
+		<synopsis>Statsd client.</synopsis>
+		<configFile name="statsd.conf">
+			<configObject name="global">
+				<synopsis>Global configuration settings</synopsis>
+				<configOption name="enabled">
+					<synopsis>Enable/disable the stasis-http module</synopsis>
+				</configOption>
+				<configOption name="server">
+					<synopsis>Address of the statsd server</synopsis>
+				</configOption>
+				<configOption name="prefix">
+					<synopsis>Prefix to prepend to every metric</synopsis>
+				</configOption>
+				<configOption name="add_newline">
+					<synopsis>Append a newline to every event. This is useful if you want to fake out a server using netcat (nc -lu 8125)</synopsis>
+				</configOption>
+			</configObject>
+		</configFile>
+	</configInfo>
+***/
+
+#include "asterisk.h"
+
+ASTERISK_FILE_VERSION(__FILE__, "$Revision$")
+
+#include "asterisk/statsd.h"
+#include "asterisk/module.h"
+#include "asterisk/netsock2.h"
+#include "asterisk/config_options.h"
+
+#define DEFAULT_STATSD_PORT 8125
+
+#define MAX_PREFIX 40
+
+/*! Socket for sending statd messages */
+static int socket_fd = -1;
+/*! Statsd server address:port. */
+struct ast_sockaddr statsd_server;
+
+/*! \brief Global configuration options for statsd client. */
+struct conf_global_options {
+	/*! Enabled by default, disabled if false. */
+	int enabled;
+	/*! Disabled by default, appends newlines to all messages when enabled. */
+	int add_newline;
+	/*! Statsd server address[:port]. */
+	struct ast_sockaddr statsd_server;
+	/*! Prefix to put on every stat. */
+	char prefix[MAX_PREFIX + 1];
+};
+
+/*! \brief All configuration options for statsd client. */
+struct conf {
+	/*! The general section configuration options. */
+	struct conf_global_options *global;
+};
+
+/*! \brief Locking container for safe configuration access. */
+static AO2_GLOBAL_OBJ_STATIC(confs);
+
+static const char *type_to_str(enum ast_statsd_type type);
+
+void AST_OPTIONAL_API_NAME(ast_statsd_log_full)(const char *metric_name,
+	const char *type_str, intmax_t value, double sample_rate)
+{
+	RAII_VAR(struct conf *, cfg, NULL, ao2_cleanup);
+	RAII_VAR(struct ast_str *, msg, NULL, ast_free);
+	size_t len;
+
+	if (socket_fd == -1) {
+		return;
+	}
+
+	/* Rates <= 0.0 never get logged.
+	 * Rates >= 1.0 always get logged.
+	 * All others leave it to chance.
+	 */
+	if (sample_rate <= 0.0 ||
+		(sample_rate < 1.0 && sample_rate < ast_random_double())) {
+		return;
+	}
+
+	cfg = ao2_global_obj_ref(confs);
+
+	msg = ast_str_create(40);
+	if (!msg) {
+		return;
+	}
+
+	if (!ast_strlen_zero(cfg->global->prefix)) {
+		ast_str_append(&msg, 0, "%s.", cfg->global->prefix);
+	}
+
+	ast_str_append(&msg, 0, "%s:%jd|%s", metric_name, value, type_str);
+
+	if (sample_rate < 1.0) {
+		ast_str_append(&msg, 0, "|@%.2f", sample_rate);
+	}
+
+	if (cfg->global->add_newline) {
+		ast_str_append(&msg, 0, "\n");
+	}
+
+	len = ast_str_strlen(msg);
+
+	ast_sendto(socket_fd, ast_str_buffer(msg), len, 0, &statsd_server);
+}
+
+void AST_OPTIONAL_API_NAME(ast_statsd_log)(const char *metric_name,
+	enum ast_statsd_type metric_type, intmax_t value)
+{
+	ast_statsd_log_full(metric_name, type_to_str(metric_type), value, 1.0);
+}
+
+void AST_OPTIONAL_API_NAME(ast_statsd_log_sample)(const char *metric_name,
+	intmax_t value, double sample_rate)
+{
+	ast_statsd_log_full(metric_name, type_to_str(AST_STATSD_COUNTER), value,
+		sample_rate);
+}
+
+/*! \brief Mapping of the stasis http conf struct's globals to the
+ *         general context in the config file. */
+static struct aco_type global_option = {
+	.type = ACO_GLOBAL,
+	.name = "global",
+	.item_offset = offsetof(struct conf, global),
+	.category = "^general$",
+	.category_match = ACO_WHITELIST
+};
+
+static struct aco_type *global_options[] = ACO_TYPES(&global_option);
+
+/*! \brief Disposes of the stasis http conf object */
+static void conf_destructor(void *obj)
+{
+    struct conf *cfg = obj;
+    ao2_cleanup(cfg->global);
+}
+
+/*! \brief Creates the statis http conf object. */
+static void *conf_alloc(void)
+{
+    struct conf *cfg;
+
+    if (!(cfg = ao2_alloc(sizeof(*cfg), conf_destructor))) {
+        return NULL;
+    }
+
+    if (!(cfg->global = ao2_alloc(sizeof(*cfg->global), NULL))) {
+        ao2_ref(cfg, -1);
+        return NULL;
+    }
+    return cfg;
+}
+
+/*! \brief The conf file that's processed for the module. */
+static struct aco_file conf_file = {
+	/*! The config file name. */
+	.filename = "statsd.conf",
+	/*! The mapping object types to be processed. */
+	.types = ACO_TYPES(&global_option),
+};
+
+CONFIG_INFO_STANDARD(cfg_info, confs, conf_alloc,
+		     .files = ACO_FILES(&conf_file));
+
+/*! \brief Helper function to check if module is enabled. */
+static char is_enabled(void)
+{
+	RAII_VAR(struct conf *, cfg, ao2_global_obj_ref(confs), ao2_cleanup);
+	return cfg->global->enabled;
+}
+
+static int conf_server(struct ast_sockaddr *addr)
+{
+	RAII_VAR(struct conf *, cfg, ao2_global_obj_ref(confs), ao2_cleanup);
+	/* This should really do a name lookup, in which case it may fail.
+	 * For now, it always works.
+	 */
+	*addr = cfg->global->statsd_server;
+	if (ast_sockaddr_port(addr) == 0) {
+		ast_sockaddr_set_port(addr, DEFAULT_STATSD_PORT);
+	}
+	return 0;
+}
+
+static const char *type_to_str(enum ast_statsd_type type)
+{
+	switch (type) {
+	case AST_STATSD_GUAGE:
+		return "g";
+	case AST_STATSD_COUNTER:
+		return "c";
+	case AST_STATSD_TIMER:
+		return "ms";
+	case AST_STATSD_HISTOGRAM:
+		return "h";
+	case AST_STATSD_METER:
+		return "m";
+	}
+
+	ast_assert(0);
+	return "?";
+}
+
+static int statsd_init(void)
+{
+	RAII_VAR(struct conf *, cfg, ao2_global_obj_ref(confs), ao2_cleanup);
+	char *server;
+	int ret;
+
+	ast_assert(is_enabled());
+
+	ast_debug(3, "Starting up statsd client.\n");
+	ret = conf_server(&statsd_server);
+	if (ret != 0) {
+		ast_log(LOG_ERROR, "Failed to find statsd server\n");
+		return AST_MODULE_LOAD_DECLINE;
+	}
+	server = ast_sockaddr_stringify_fmt(&statsd_server,
+		AST_SOCKADDR_STR_DEFAULT);
+	ast_debug(3, "  statsd server = %s.\n", server);
+
+	if (socket_fd == -1) {
+		socket_fd = socket(AF_INET, SOCK_DGRAM, IPPROTO_UDP);
+		if (socket_fd == -1) {
+			perror("Error creating statsd socket");
+			return AST_MODULE_LOAD_FAILURE;
+		}
+	}
+
+	return AST_MODULE_LOAD_SUCCESS;
+}
+
+static void statsd_shutdown(void)
+{
+	ast_debug(3, "Shutting down statsd client.\n");
+	if (socket_fd != -1) {
+		close(socket_fd);
+		socket_fd = -1;
+	}
+}
+
+static int load_module(void)
+{
+	if (aco_info_init(&cfg_info)) {
+		aco_info_destroy(&cfg_info);
+		return AST_MODULE_LOAD_DECLINE;
+	}
+
+	aco_option_register(&cfg_info, "enabled", ACO_EXACT, global_options,
+		"yes", OPT_BOOL_T, 1,
+		FLDSET(struct conf_global_options, enabled));
+
+	aco_option_register(&cfg_info, "add_newline", ACO_EXACT, global_options,
+		"yes", OPT_BOOL_T, 1,
+		FLDSET(struct conf_global_options, add_newline));
+
+	aco_option_register(&cfg_info, "server", ACO_EXACT, global_options,
+		"127.0.0.1", OPT_SOCKADDR_T, 0,
+		FLDSET(struct conf_global_options, statsd_server));
+
+	aco_option_register(&cfg_info, "prefix", ACO_EXACT, global_options,
+		"", OPT_CHAR_ARRAY_T, 0,
+		FLDSET(struct conf_global_options, prefix));
+
+	if (aco_process_config(&cfg_info, 0)) {
+		aco_info_destroy(&cfg_info);
+		return AST_MODULE_LOAD_DECLINE;
+	}
+
+	if (!is_enabled()) {
+		return AST_MODULE_LOAD_SUCCESS;
+	}
+
+	return statsd_init();
+}
+
+static int unload_module(void)
+{
+	statsd_shutdown();
+	aco_info_destroy(&cfg_info);
+	ao2_global_obj_release(confs);
+	return 0;
+}
+
+static int reload_module(void)
+{
+	char was_enabled = is_enabled();
+
+	if (aco_process_config(&cfg_info, 1)) {
+		return AST_MODULE_LOAD_DECLINE;
+	}
+
+	if (!was_enabled && is_enabled()) {
+		return statsd_init();
+	}
+
+	if (was_enabled && !is_enabled()) {
+		statsd_shutdown();
+		return AST_MODULE_LOAD_SUCCESS;
+	}
+
+	return AST_MODULE_LOAD_SUCCESS;
+}
+
+/* The priority of this module is set to be as low as possible, since it could
+ * be used by any other sort of module.
+ */
+AST_MODULE_INFO(ASTERISK_GPL_KEY,
+	AST_MODFLAG_GLOBAL_SYMBOLS | AST_MODFLAG_LOAD_ORDER,
+	"Statsd client support",
+	.load = load_module,
+	.unload = unload_module,
+	.reload = reload_module,
+	.load_pri = 0,
+	);

Propchange: team/dlee/stasis-demo/res/res_statsd.c
------------------------------------------------------------------------------
    svn:eol-style = native

Propchange: team/dlee/stasis-demo/res/res_statsd.c
------------------------------------------------------------------------------
    svn:keywords = Author Date Id Revision

Propchange: team/dlee/stasis-demo/res/res_statsd.c
------------------------------------------------------------------------------
    svn:mime-type = text/plain




More information about the asterisk-commits mailing list