diff options
author | Nir Simionovich <nirs@greenfieldtech.net> | 2017-10-16 17:46:02 -0400 |
---|---|---|
committer | Nir Simionovich <nirs@greenfieldtech.net> | 2017-10-20 10:08:05 -0400 |
commit | 4559cd0e284ee3cb672d9c7c935ab99b3bb16489 (patch) | |
tree | 5654e35e5a6b6c1db91d475d892cc5b588b9a0ed /cdr | |
parent | 00d1c7ddd28557aa845c3522956852a60310df96 (diff) |
This patch adds a beanstalk CDR backend.
Beanstalkd is a simple to use job queue. It provides a means to
create multiple job queues called "tubes". Each tube can store
multiple jobs, with varying priorities with the queue. Queue
processing is available via a simple TCP socket or via well defined
libraries, avaialble at
https://github.com/kr/beanstalkd/wiki/client-libraries
This module is based upon the beanstalk-client library, available
for download at: https://github.com/deepfryed/beanstalk-client
Change-Id: I5fe4089a34ab3b39230786d9bbfddafa56715f48
Diffstat (limited to 'cdr')
-rw-r--r-- | cdr/cdr_beanstalkd.c | 270 |
1 files changed, 270 insertions, 0 deletions
diff --git a/cdr/cdr_beanstalkd.c b/cdr/cdr_beanstalkd.c new file mode 100644 index 000000000..4812c91aa --- /dev/null +++ b/cdr/cdr_beanstalkd.c @@ -0,0 +1,270 @@ +/* + * Asterisk -- An open source telephony toolkit. + * + * Copyright (C) 2017 + * + * Nir Simionovich <nirs@greenfieldtech.net> + * + * 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. + */ + +/*! + * \file + * \brief Asterisk Beanstalkd CDR records. + * + * This module requires the beanstalk-client library, avaialble from + * https://github.com/deepfryed/beanstalk-client + * + * See also + * \arg \ref AstCDR + * \ingroup cdr_drivers + */ + +/*! \li \ref cdr_beanstalkd.c uses the configuration file \ref cdr_beanstalkd.conf + * \addtogroup configuration_file Configuration Files + */ + +/*! + * \page cdr_beanstalkd.conf cdr_beanstalkd.conf + * \verbinclude cdr_beanstalkd.conf.sample + */ + +/*** MODULEINFO + <depend>beanstalk</depend> + <support_level>extended</support_level> + ***/ + +#include "asterisk.h" + +#include <time.h> +#include <stdio.h> + +#include "beanstalk.h" +#include "asterisk/channel.h" +#include "asterisk/cdr.h" +#include "asterisk/module.h" +#include "asterisk/utils.h" +#include "asterisk/manager.h" +#include "asterisk/config.h" +#include "asterisk/pbx.h" +#include "asterisk/json.h" + +#define DATE_FORMAT "%Y-%m-%d %T" +#define CONF_FILE "cdr_beanstalkd.conf" +#define BEANSTALK_JOB_SIZE 4096 +#define BEANSTALK_JOB_PRIORITY 99 +#define BEANSTALK_JOB_TTR 60 +#define BEANSTALK_JOB_DELAY 0 +#define DEFAULT_BEANSTALK_HOST "127.0.0.1" +#define DEFAULT_BEANSTALK_PORT 11300 +#define DEFAULT_BEANSTALK_TUBE "asterisk-cdr" + +static const char name[] = "cdr_beanstalkd"; + +static int enablecdr = 0; +static char *bs_host; +static int bs_port; +static char *bs_tube; +static int priority; + +AST_RWLOCK_DEFINE_STATIC(config_lock); + +static int beanstalk_put(struct ast_cdr *cdr); + +static int load_config(int reload) { + char *cat = NULL; + struct ast_config *cfg; + struct ast_variable *v; + struct ast_flags config_flags = {reload ? CONFIG_FLAG_FILEUNCHANGED : 0}; + int newenablecdr = 0; + + cfg = ast_config_load(CONF_FILE, config_flags); + if (cfg == CONFIG_STATUS_FILEUNCHANGED) { + return 0; + } + + if (cfg == CONFIG_STATUS_FILEINVALID) { + ast_log(LOG_ERROR, "Config file '%s' could not be parsed\n", CONF_FILE); + return -1; + } + + if (!cfg) { + /* Standard configuration */ + ast_log(LOG_WARNING, "Failed to load configuration file. Module not activated.\n"); + if (enablecdr) { + ast_cdr_backend_suspend(name); + } + enablecdr = 0; + return -1; + } + + if (reload) { + ast_rwlock_wrlock(&config_lock); + ast_free(bs_host); + ast_free(bs_tube); + } + + /* Bootstrap the default configuration */ + bs_host = ast_strdup(DEFAULT_BEANSTALK_HOST); + bs_port = DEFAULT_BEANSTALK_PORT; + bs_tube = ast_strdup(DEFAULT_BEANSTALK_TUBE); + priority = BEANSTALK_JOB_PRIORITY; + + while ((cat = ast_category_browse(cfg, cat))) { + if (!strcasecmp(cat, "general")) { + v = ast_variable_browse(cfg, cat); + while (v) { + + if (!strcasecmp(v->name, "enabled")) { + newenablecdr = ast_true(v->value); + } else if (!strcasecmp(v->name, "host")) { + ast_free(bs_host); + bs_host = ast_strdup(v->value); + } else if (!strcasecmp(v->name, "port")) { + bs_port = atoi(v->value); + } else if (!strcasecmp(v->name, "tube")) { + ast_free(bs_tube); + bs_tube = ast_strdup(v->value); + } else if (!strcasecmp(v->name, "priority")) { + priority = atoi(v->value); + } + v = v->next; + + } + } + } + + if (reload) { + ast_rwlock_unlock(&config_lock); + } + + ast_config_destroy(cfg); + + if (!newenablecdr) { + ast_cdr_backend_suspend(name); + } else if (newenablecdr) { + ast_cdr_backend_unsuspend(name); + ast_log(LOG_NOTICE, "Added beanstalkd server %s at port %d with tube %s", bs_host, bs_port, bs_tube); + } + enablecdr = newenablecdr; + + return 0; +} + +static int beanstalk_put(struct ast_cdr *cdr) { + struct ast_tm timeresult; + char strAnswerTime[80] = ""; + char strStartTime[80]; + char strEndTime[80]; + char *cdr_buffer; + int bs_id; + int bs_socket; + struct ast_json *t_cdr_json; + + if (!enablecdr) { + return 0; + } + + ast_rwlock_rdlock(&config_lock); + bs_socket = bs_connect(bs_host, bs_port); + + if (bs_use(bs_socket, bs_tube) != BS_STATUS_OK) { + ast_log(LOG_ERROR, "Connection to Beanstalk tube %s @ %s:%d had failed", bs_tube, bs_host, bs_port); + ast_rwlock_unlock(&config_lock); + return 0; + } + + ast_localtime(&cdr->start, &timeresult, NULL); + ast_strftime(strStartTime, sizeof(strStartTime), DATE_FORMAT, &timeresult); + + if (cdr->answer.tv_sec) { + ast_localtime(&cdr->answer, &timeresult, NULL); + ast_strftime(strAnswerTime, sizeof(strAnswerTime), DATE_FORMAT, &timeresult); + } + + ast_localtime(&cdr->end, &timeresult, NULL); + ast_strftime(strEndTime, sizeof(strEndTime), DATE_FORMAT, &timeresult); + + ast_rwlock_unlock(&config_lock); + + t_cdr_json = ast_json_pack("{s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:s, s:i, s:i, s:s, s:s, s:s, s:s}", + "AccountCode", S_OR(cdr->accountcode, ""), + "Source", S_OR(cdr->src, ""), + "Destination", S_OR(cdr->dst, ""), + "DestinationContext", S_OR(cdr->dcontext, ""), + "CallerID", S_OR(cdr->clid, ""), + "Channel", S_OR(cdr->channel, ""), + "DestinationChannel", S_OR(cdr->dstchannel, ""), + "LastApplication", S_OR(cdr->lastapp, ""), + "LastData", S_OR(cdr->lastdata, ""), + "StartTime", S_OR(strStartTime, ""), + "AnswerTime", S_OR(strAnswerTime, ""), + "EndTime", S_OR(strEndTime, ""), + "Duration", cdr->duration, + "Billsec", cdr->billsec, + "Disposition", S_OR(ast_cdr_disp2str(cdr->disposition), ""), + "AMAFlags", S_OR(ast_channel_amaflags2string(cdr->amaflags), ""), + "UniqueID", S_OR(cdr->uniqueid, ""), + "UserField", S_OR(cdr->userfield, "")); + + cdr_buffer = ast_json_dump_string(t_cdr_json); + + ast_json_unref(t_cdr_json); + + bs_id = bs_put(bs_socket, priority, BEANSTALK_JOB_DELAY, BEANSTALK_JOB_TTR, cdr_buffer, strlen(cdr_buffer)); + + if (bs_id > 0) { + ast_log(LOG_DEBUG, "Successfully created job %d with %s\n", bs_id, cdr_buffer); + } else { + ast_log(LOG_ERROR, "CDR job creation failed for %s\n", cdr_buffer); + } + + bs_disconnect(bs_socket); + ast_json_free(cdr_buffer); + return 0; +} + +static int unload_module(void) { + if (ast_cdr_unregister(name)) { + return -1; + } + + ast_free(bs_host); + ast_free(bs_tube); + + return 0; +} + +static int load_module(void) { + if (ast_cdr_register(name, "Asterisk CDR Beanstalkd Backend", beanstalk_put)) { + return AST_MODULE_LOAD_DECLINE; + } + + if (load_config(0)) { + ast_cdr_unregister(name); + return AST_MODULE_LOAD_DECLINE; + } + + return AST_MODULE_LOAD_SUCCESS; +} + +static int reload(void) { + return load_config(1); +} + +AST_MODULE_INFO(ASTERISK_GPL_KEY, AST_MODFLAG_LOAD_ORDER, "Asterisk Beanstalkd CDR Backend", +.support_level = AST_MODULE_SUPPORT_EXTENDED, +.load = load_module, +.unload = unload_module, +.reload = reload, +.load_pri = AST_MODPRI_CDR_DRIVER, +); + |