kenmirrors
/
freeswitch
mirror of https://github.com/signalwire/freeswitch.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Add events and a shutdown function to mod_limit 

git-svn-id: http://svn.freeswitch.org/svn/freeswitch/trunk@12497 d0543943-73ff-0310-b7d9-9358b9ac24b2
This commit is contained in:
Mathieu Rene 2009-03-07 04:56:07 +00:00
parent bbf213de0a
commit 97412d0e13
1 changed files with 49 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
src/mod/applications/mod_limit/mod_limit.c
View File

@ -37,8 +37,11 @@
#include <switch_odbc.h> #include <switch_odbc.h>
#endif #endif
#define LIMIT_EVENT_USAGE "limit::usage"
SWITCH_MODULE_LOAD_FUNCTION(mod_limit_load); SWITCH_MODULE_LOAD_FUNCTION(mod_limit_load);
SWITCH_MODULE_DEFINITION(mod_limit, mod_limit_load, NULL, NULL); SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_limit_shutdown);
SWITCH_MODULE_DEFINITION(mod_limit, mod_limit_load, mod_limit_shutdown, NULL);
static struct { static struct {
switch_memory_pool_t *pool; switch_memory_pool_t *pool;
@ -291,6 +294,21 @@ static switch_status_t do_config()
return status; return status;
} }
static void limit_fire_event(const char *realm, const char *key, uint32_t usage, uint32_t rate, uint32_t max, uint32_t ratemax)
{
switch_event_t *event;
if (switch_event_create_subclass(&event, SWITCH_EVENT_CUSTOM, LIMIT_EVENT_USAGE) == SWITCH_STATUS_SUCCESS) {
switch_event_add_header_string(event, SWITCH_STACK_BOTTOM, "realm", realm);
switch_event_add_header_string(event, SWITCH_STACK_BOTTOM, "key", key);
switch_event_add_header(event, SWITCH_STACK_BOTTOM, "usage", "%d", usage);
switch_event_add_header(event, SWITCH_STACK_BOTTOM, "rate", "%d", rate);
switch_event_add_header(event, SWITCH_STACK_BOTTOM, "max", "%d", max);
switch_event_add_header(event, SWITCH_STACK_BOTTOM, "ratemax", "%d", ratemax);
switch_event_fire(&event);
}
}
static switch_status_t db_state_handler(switch_core_session_t *session) static switch_status_t db_state_handler(switch_core_session_t *session)
{ {
switch_channel_t *channel = switch_core_session_get_channel(session); switch_channel_t *channel = switch_core_session_get_channel(session);
@ -792,7 +810,8 @@ SWITCH_STANDARD_APP(limit_function)
limit_execute_sql(sql, NULL); limit_execute_sql(sql, NULL);
switch_safe_free(sql); switch_safe_free(sql);
switch_channel_set_variable(channel, "limit_usage", switch_core_session_sprintf(session, "%d", ++got)); switch_channel_set_variable_printf(channel, "limit_usage", "%d", ++got);
limit_fire_event(realm, id, got, 0, max, 0);
done: done:
switch_mutex_unlock(globals.mutex); switch_mutex_unlock(globals.mutex);
@ -954,7 +973,6 @@ SWITCH_STANDARD_APP(limit_hash_function)
if (increment) { if (increment) {
item->total_usage++; item->total_usage++;
switch_core_hash_insert(pvt->hash, hashkey, item); switch_core_hash_insert(pvt->hash, hashkey, item);
if (max == -1) { if (max == -1) {
@ -964,6 +982,8 @@ SWITCH_STANDARD_APP(limit_hash_function)
} else { } else {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "Usage for %s is now %d/%d for the last %d seconds\n", hashkey, item->rate_usage, max, interval); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "Usage for %s is now %d/%d for the last %d seconds\n", hashkey, item->rate_usage, max, interval);
} }
limit_fire_event(realm, id, item->total_usage, item->rate_usage, max, max >=0 ? (uint32_t)max : 0);
} }
/* Save current usage & rate into channel variables so it can be used later in the dialplan, or added to CDR records */ /* Save current usage & rate into channel variables so it can be used later in the dialplan, or added to CDR records */
@ -1036,6 +1056,13 @@ SWITCH_MODULE_LOAD_FUNCTION(mod_limit_load)
gethostname(globals.hostname, sizeof(globals.hostname)); gethostname(globals.hostname, sizeof(globals.hostname));
globals.pool = pool; globals.pool = pool;
if (switch_event_reserve_subclass(LIMIT_EVENT_USAGE) != SWITCH_STATUS_SUCCESS) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Couldnt register event subclass \"%s\"",
LIMIT_EVENT_USAGE);
return SWITCH_STATUS_FALSE;
}
if ((status = do_config() != SWITCH_STATUS_SUCCESS)) { if ((status = do_config() != SWITCH_STATUS_SUCCESS)) {
return status; return status;
} }
@ -1048,6 +1075,8 @@ SWITCH_MODULE_LOAD_FUNCTION(mod_limit_load)
/* connect my internal structure to the blank pointer passed to me */ /* connect my internal structure to the blank pointer passed to me */
*module_interface = switch_loadable_module_create_module_interface(pool, modname); *module_interface = switch_loadable_module_create_module_interface(pool, modname);
SWITCH_ADD_APP(app_interface, "limit", "Limit", LIMIT_DESC, limit_function, LIMIT_USAGE, SAF_SUPPORT_NOMEDIA); SWITCH_ADD_APP(app_interface, "limit", "Limit", LIMIT_DESC, limit_function, LIMIT_USAGE, SAF_SUPPORT_NOMEDIA);
SWITCH_ADD_APP(app_interface, "limit_hash", "Limit (hash)", LIMITHASH_DESC, limit_hash_function, LIMITHASH_USAGE, SAF_SUPPORT_NOMEDIA); SWITCH_ADD_APP(app_interface, "limit_hash", "Limit (hash)", LIMITHASH_DESC, limit_hash_function, LIMITHASH_USAGE, SAF_SUPPORT_NOMEDIA);
@ -1074,6 +1103,23 @@ SWITCH_MODULE_LOAD_FUNCTION(mod_limit_load)
return SWITCH_STATUS_SUCCESS; return SWITCH_STATUS_SUCCESS;
} }
SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_limit_shutdown)
{
switch_event_free_subclass(LIMIT_EVENT_USAGE);
switch_xml_config_cleanup(config_settings);
switch_mutex_destroy(globals.mutex);
switch_mutex_destroy(globals.limit_hash_mutex);
switch_mutex_destroy(globals.db_hash_mutex);
switch_core_hash_destroy(&globals.limit_hash);
switch_core_hash_destroy(&globals.db_hash);
return SWITCH_STATUS_SUCCESS;
}
/* For Emacs: /* For Emacs:
* Local Variables: * Local Variables:
* mode:c * mode:c