kenmirrors/freeswitch
1
0
Fork 0
mirror of https://github.com/signalwire/freeswitch.git synced 2025-04-16 08:49:01 +00:00
Code Issues Projects Releases Wiki Activity

use new cached odbc api

Browse Source 
git-svn-id: http://svn.freeswitch.org/svn/freeswitch/trunk@15680 d0543943-73ff-0310-b7d9-9358b9ac24b2
This commit is contained in:
Rupa Schomaker 2009-11-26 03:54:04 +00:00
parent 36689432e0
commit ae2ed8f5d8
1 changed files with 42 additions and 62 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

104
src/mod/applications/mod_cidlookup/mod_cidlookup.c
View File

@ -55,12 +55,12 @@ static struct {
int cache_expire; int cache_expire;
char *odbc_dsn; char *odbc_dsn;
char *odbc_user;
char *odbc_pass;
char *sql; char *sql;
char *citystate_sql; char *citystate_sql;
switch_mutex_t *db_mutex;
switch_memory_pool_t *pool; switch_memory_pool_t *pool;
switch_odbc_handle_t *master_odbc;
} globals; } globals;
struct http_data { struct http_data {
@ -84,71 +84,62 @@ struct callback_obj {
}; };
typedef struct callback_obj callback_t; typedef struct callback_obj callback_t;
static switch_event_node_t *reload_xml_event = NULL; static switch_event_node_t *reload_xml_event = NULL;
static switch_cache_db_handle_t *cidlookup_get_db_handle(void)
{
switch_cache_db_connection_options_t options = { {0} };
switch_cache_db_handle_t *dbh = NULL;
if (globals.odbc_dsn && globals.odbc_user && globals.odbc_pass) {
options.odbc_options.dsn = globals.odbc_dsn;
options.odbc_options.user = globals.odbc_user;
options.odbc_options.pass = globals.odbc_pass;
if (switch_cache_db_get_db_handle(&dbh, SCDB_TYPE_ODBC, &options) != SWITCH_STATUS_SUCCESS) dbh = NULL;
}
return dbh;
}
static switch_status_t config_callback_dsn(switch_xml_config_item_t *data, const char *newvalue, switch_config_callback_type_t callback_type, switch_bool_t changed) static switch_status_t config_callback_dsn(switch_xml_config_item_t *data, const char *newvalue, switch_config_callback_type_t callback_type, switch_bool_t changed)
{ {
switch_status_t status = SWITCH_STATUS_SUCCESS; switch_status_t status = SWITCH_STATUS_SUCCESS;
char *odbc_user = NULL;
char *odbc_pass = NULL;
char *odbc_dsn = NULL;
switch_odbc_handle_t *odbc = NULL; switch_cache_db_handle_t *dbh = NULL;
if (!switch_odbc_available()) { if (!switch_odbc_available()) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "ODBC is not compiled in. Do not configure odbc-dsn parameter!\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "ODBC is not compiled in. Do not configure odbc-dsn parameter!\n");
return SWITCH_STATUS_FALSE; return SWITCH_STATUS_FALSE;
} }
if (globals.db_mutex) {
switch_mutex_lock(globals.db_mutex);
}
if ((callback_type == CONFIG_LOAD || callback_type == CONFIG_RELOAD) && changed) { if ((callback_type == CONFIG_LOAD || callback_type == CONFIG_RELOAD) && changed) {
if(zstr(newvalue)) { if(zstr(newvalue)) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "No local database defined.\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "No local database defined.\n");
} else { } else {
odbc_dsn = strdup(newvalue); switch_safe_free(globals.odbc_dsn);
if ((odbc_user = strchr(odbc_dsn, ':'))) { globals.odbc_dsn = strdup(newvalue);
*odbc_user++ = '\0'; if ((globals.odbc_user = strchr(globals.odbc_dsn, ':'))) {
if ((odbc_pass = strchr(odbc_user, ':'))) { *globals.odbc_user++ = '\0';
*odbc_pass++ = '\0'; if ((globals.odbc_pass = strchr(globals.odbc_user, ':'))) {
*globals.odbc_pass++ = '\0';
} }
} }
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "Connecting to dsn: %s\n", globals.odbc_dsn); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_INFO, "Connecting to dsn: %s\n", globals.odbc_dsn);
/* setup dsn */ if (!(dbh = cidlookup_get_db_handle())) {
if (!(odbc = switch_odbc_handle_new(odbc_dsn, odbc_user, odbc_pass))) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_CRIT, "Cannot Open ODBC Database!\n");
switch_goto_status(SWITCH_STATUS_FALSE, done);
}
if (switch_odbc_handle_connect(odbc) != SWITCH_ODBC_SUCCESS) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_CRIT, "Cannot Open ODBC Database!\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_CRIT, "Cannot Open ODBC Database!\n");
switch_goto_status(SWITCH_STATUS_FALSE, done); switch_goto_status(SWITCH_STATUS_FALSE, done);
} }
} }
/* ok, we have a new connection, tear down old one */
if (globals.master_odbc) {
switch_odbc_handle_disconnect(globals.master_odbc);
switch_odbc_handle_destroy(&globals.master_odbc);
}
/* and swap in new connection */
globals.master_odbc = odbc;
} }
switch_goto_status(SWITCH_STATUS_SUCCESS, done); switch_goto_status(SWITCH_STATUS_SUCCESS, done);
done: done:
if (globals.db_mutex) { switch_cache_db_release_db_handle(&dbh);
switch_mutex_unlock(globals.db_mutex);
}
switch_safe_free(odbc_dsn);
return status; return status;
} }
@ -180,20 +171,23 @@ static void event_handler(switch_event_t *event)
do_config(SWITCH_TRUE); do_config(SWITCH_TRUE);
} }
static switch_bool_t cidlookup_execute_sql_callback(char *sql, switch_core_db_callback_func_t callback, void *pdata) static switch_bool_t cidlookup_execute_sql_callback(char *sql, switch_core_db_callback_func_t callback, callback_t *cbt, char **err)
{ {
switch_bool_t retval = SWITCH_FALSE; switch_bool_t retval = SWITCH_FALSE;
switch_cache_db_handle_t *dbh = NULL;
switch_mutex_lock(globals.db_mutex); if (globals.odbc_dsn && (dbh = cidlookup_get_db_handle())) {
if (globals.odbc_dsn) { if (switch_cache_db_execute_sql_callback(dbh, sql, callback, (void *)cbt, err)
if (switch_odbc_handle_callback_exec(globals.master_odbc, sql, callback, pdata, NULL)
== SWITCH_ODBC_FAIL) { == SWITCH_ODBC_FAIL) {
retval = SWITCH_FALSE; retval = SWITCH_FALSE;
} else { } else {
retval = SWITCH_TRUE; retval = SWITCH_TRUE;
} }
} else {
*err = switch_core_sprintf(cbt->pool, "Unable to get ODBC handle. dsn: %s, dbh is %s\n", globals.odbc_dsn, dbh ? "not null" : "null");
} }
switch_mutex_unlock(globals.db_mutex);
switch_cache_db_release_db_handle(&dbh);
return retval; return retval;
} }
@ -492,16 +486,17 @@ done:
static char *do_db_lookup(switch_memory_pool_t *pool, switch_event_t *event, const char *num, const char *sql) { static char *do_db_lookup(switch_memory_pool_t *pool, switch_event_t *event, const char *num, const char *sql) {
char *name = NULL; char *name = NULL;
char *newsql = NULL; char *newsql = NULL;
char *err = NULL;
callback_t cbt = { 0 }; callback_t cbt = { 0 };
cbt.pool = pool; cbt.pool = pool;
if (globals.master_odbc) { if (globals.odbc_dsn) {
newsql = switch_event_expand_headers(event, sql); newsql = switch_event_expand_headers(event, sql);
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "SQL: %s\n", newsql); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_DEBUG, "SQL: %s\n", newsql);
if (cidlookup_execute_sql_callback(newsql, cidlookup_callback, &cbt)) { if (cidlookup_execute_sql_callback(newsql, cidlookup_callback, &cbt, &err)) {
name = cbt.name; name = cbt.name;
} else { } else {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Unable to lookup cid\n"); switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "Unable to lookup cid: %s\n", err ? err : "(null)");
} }
} }
if (newsql != globals.sql) { if (newsql != globals.sql) {
@ -524,7 +519,7 @@ static cid_data_t *do_lookup(switch_memory_pool_t *pool, switch_event_t *event,
switch_event_add_header_string(event, SWITCH_STACK_BOTTOM, "caller_id_number", number); switch_event_add_header_string(event, SWITCH_STACK_BOTTOM, "caller_id_number", number);
/* database always wins */ /* database always wins */
if (switch_odbc_available() && globals.master_odbc && globals.sql) { if (switch_odbc_available() && globals.odbc_dsn && globals.sql) {
name = do_db_lookup(pool, event, number, globals.sql); name = do_db_lookup(pool, event, number, globals.sql);
if (name) { if (name) {
cid->name = name; cid->name = name;
@ -572,7 +567,7 @@ done:
if (!cid->area && if (!cid->area &&
!skipcitystate && !skipcitystate &&
strlen(number) == 11 && number[0] == '1' && strlen(number) == 11 && number[0] == '1' &&
switch_odbc_available() && globals.master_odbc && globals.citystate_sql) { switch_odbc_available() && globals.odbc_dsn && globals.citystate_sql) {
/* yes, this is really area */ /* yes, this is really area */
name = do_db_lookup(pool, event, number, globals.citystate_sql); name = do_db_lookup(pool, event, number, globals.citystate_sql);
@ -771,12 +766,6 @@ SWITCH_MODULE_LOAD_FUNCTION(mod_cidlookup_load)
globals.pool = pool; globals.pool = pool;
if (switch_odbc_available() && !globals.db_mutex) {
if (switch_mutex_init(&globals.db_mutex, SWITCH_MUTEX_UNNESTED, globals.pool) != SWITCH_STATUS_SUCCESS) {
switch_log_printf(SWITCH_CHANNEL_LOG, SWITCH_LOG_ERROR, "failed to initialize db_mutex\n");
}
}
do_config(SWITCH_FALSE); do_config(SWITCH_FALSE);
if ((switch_event_bind_removable(modname, SWITCH_EVENT_RELOADXML, NULL, event_handler, NULL, &reload_xml_event) != SWITCH_STATUS_SUCCESS)) { if ((switch_event_bind_removable(modname, SWITCH_EVENT_RELOADXML, NULL, event_handler, NULL, &reload_xml_event) != SWITCH_STATUS_SUCCESS)) {
@ -797,15 +786,6 @@ SWITCH_MODULE_LOAD_FUNCTION(mod_cidlookup_load)
Macro expands to: switch_status_t mod_cidlookup_shutdown() */ Macro expands to: switch_status_t mod_cidlookup_shutdown() */
SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_cidlookup_shutdown) SWITCH_MODULE_SHUTDOWN_FUNCTION(mod_cidlookup_shutdown)
{ {
/* Cleanup dynamically allocated config settings */
if (globals.db_mutex) {
switch_mutex_destroy(globals.db_mutex);
}
if (globals.master_odbc) {
switch_odbc_handle_disconnect(globals.master_odbc);
switch_odbc_handle_destroy(&globals.master_odbc);
}
switch_event_unbind(&reload_xml_event); switch_event_unbind(&reload_xml_event);
return SWITCH_STATUS_SUCCESS; return SWITCH_STATUS_SUCCESS;