[svn-commits] tilghman: branch tilghman/realtime_update2 r672 - /team/tilghman/realtime_upd...
SVN commits to the Digium repositories
svn-commits at lists.digium.com
Wed Oct 8 16:30:53 CDT 2008
Author: tilghman
Date: Wed Oct 8 16:30:52 2008
New Revision: 672
URL: http://svn.digium.com/view/asterisk-addons?view=rev&rev=672
Log:
Updated driver to have update2 method
Modified:
team/tilghman/realtime_update2/res/res_config_mysql.c
Modified: team/tilghman/realtime_update2/res/res_config_mysql.c
URL: http://svn.digium.com/view/asterisk-addons/team/tilghman/realtime_update2/res/res_config_mysql.c?view=diff&rev=672&r1=671&r2=672
==============================================================================
--- team/tilghman/realtime_update2/res/res_config_mysql.c (original)
+++ team/tilghman/realtime_update2/res/res_config_mysql.c Wed Oct 8 16:30:52 2008
@@ -109,6 +109,10 @@
AST_CLI_DEFINE(handle_cli_realtime_mysql_cache, "Shows cached tables within the MySQL realtime driver"),
};
+AST_THREADSTORAGE(sql_buf);
+AST_THREADSTORAGE(where_buf);
+AST_THREADSTORAGE(escapebuf_buf);
+
static int internal_require(const char *database, const char *table, ...)
{
va_list ap;
@@ -486,14 +490,7 @@
return -1;
}
- /* Check that the column exists in the table */
- AST_LIST_TRAVERSE(&table->columns, column, list) {
- if (strcmp(column->name, keyfield) == 0) {
- break;
- }
- }
-
- if (!column) {
+ if (!(column = find_column(table, keyfield))) {
ast_log(LOG_ERROR, "PostgreSQL RealTime: Updating on column '%s', but that column does not exist within the table '%s'!\n", keyfield, tablename);
ast_mutex_unlock(&table->lock);
ast_free(sql);
@@ -589,6 +586,122 @@
ast_free(sql);
ast_free(buf);
+
+ numrows = mysql_affected_rows(&dbwrite.handle);
+ ast_mutex_unlock(&dbwrite.lock);
+
+ ast_debug(1, "MySQL RealTime: Updated %llu rows on table: %s\n", numrows, tablename);
+
+ /* From http://dev.mysql.com/doc/mysql/en/mysql-affected-rows.html
+ * An integer greater than zero indicates the number of rows affected
+ * Zero indicates that no records were updated
+ * -1 indicates that the query returned an error (although, if the query failed, it should have been caught above.)
+ */
+
+ return (int)numrows;
+}
+
+#define ESCAPE_STRING(buf, var) \
+ do { \
+ size_t size; \
+ if ((size = strlen(var)) * 2 + 1 > (buf)->len) { \
+ ast_str_make_space(&(buf), size * 2 + 1); \
+ } \
+ mysql_real_escape_string(&dbwrite.handle, (buf)->str, var, size); \
+ } while (0)
+
+static int update2_mysql(const char *database, const char *tablename, va_list ap)
+{
+ my_ulonglong numrows;
+ int first = 1;
+ const char *newparam, *newval;
+ struct ast_str *sql = ast_str_thread_get(&sql_buf, 100), *buf = ast_str_thread_get(&sql_buf, 100);
+ struct ast_str *where = ast_str_thread_get(&where_buf, 100);
+ struct tables *table;
+ struct columns *column = NULL;
+
+ if (!tablename) {
+ ast_log(LOG_WARNING, "MySQL RealTime: No table specified.\n");
+ return -1;
+ }
+
+ if (!(table = find_table(tablename))) {
+ ast_log(LOG_ERROR, "Table '%s' does not exist!!\n", tablename);
+ return -1;
+ }
+
+ if (!sql || !buf || !where) {
+ return -1;
+ }
+
+ ast_str_set(&sql, 0, "UPDATE %s SET", tablename);
+ ast_str_set(&where, 0, "WHERE");
+
+ /* Must connect to the server before anything else, as the escape function requires the mysql handle. */
+ ast_mutex_lock(&dbwrite.lock);
+ if (!mysql_reconnect(&dbwrite)) {
+ ast_mutex_unlock(&table->lock);
+ ast_mutex_unlock(&dbwrite.lock);
+ return -1;
+ }
+
+ while ((newparam = va_arg(ap, const char *))) {
+ if (!(column = find_column(table, newparam))) {
+ ast_log(LOG_ERROR, "Updating on column '%s', but that column does not exist within the table '%s'!\n", newparam, tablename);
+ ast_mutex_unlock(&table->lock);
+ return -1;
+ }
+ if (!(newval = va_arg(ap, const char *))) {
+ ast_log(LOG_ERROR, "Invalid arguments: no value specified for column '%s' on '%s@%s'\n", newparam, tablename, database);
+ ast_mutex_unlock(&table->lock);
+ return -1;
+ }
+ ESCAPE_STRING(buf, newval);
+ ast_str_append(&where, 0, "%s %s='%s'", first ? "" : " AND", newparam, buf->str);
+
+ /* If the column length isn't long enough, give a chance to lengthen it. */
+ if (strncmp(column->type, "char", 4) == 0 || strncmp(column->type, "varchar", 7) == 0) {
+ internal_require(database, tablename, newparam, RQ_CHAR, buf->used);
+ }
+ }
+
+ first = 1;
+ while ((newparam = va_arg(ap, const char *))) {
+ if (!(newval = va_arg(ap, const char *))) {
+ ast_log(LOG_ERROR, "Invalid arguments: no value specified for column '%s' on '%s@%s'\n", newparam, tablename, database);
+ ast_mutex_unlock(&table->lock);
+ return -1;
+ }
+
+ /* If the column is not within the table, then skip it */
+ if (!(column = find_column(table, newparam))) {
+ ast_log(LOG_WARNING, "Attempted to update column '%s' in table '%s', but column does not exist!\n", newparam, tablename);
+ continue;
+ }
+
+ ESCAPE_STRING(buf, newval);
+ ast_str_append(&sql, 0, "%s %s = '%s'", first ? "" : ",", newparam, buf->str);
+
+ /* If the column length isn't long enough, give a chance to lengthen it. */
+ if (strncmp(column->type, "char", 4) == 0 || strncmp(column->type, "varchar", 7) == 0) {
+ internal_require(database, tablename, newparam, RQ_CHAR, buf->used);
+ }
+ }
+ va_end(ap);
+ ast_mutex_unlock(&table->lock);
+
+ ast_str_append(&sql, 0, " %s", where->str);
+
+ ast_debug(1, "MySQL RealTime: Update SQL: %s\n", sql->str);
+
+ /* Execution. */
+ if (mysql_real_query(&dbwrite.handle, sql->str, sql->used)) {
+ ast_log(LOG_WARNING, "MySQL RealTime: Failed to query database. Check debug for more info.\n");
+ ast_debug(1, "MySQL RealTime: Query: %s\n", sql->str);
+ ast_debug(1, "MySQL RealTime: Query Failed because: %s\n", mysql_error(&dbwrite.handle));
+ ast_mutex_unlock(&dbwrite.lock);
+ return -1;
+ }
numrows = mysql_affected_rows(&dbwrite.handle);
ast_mutex_unlock(&dbwrite.lock);
@@ -1205,6 +1318,7 @@
.store_func = store_mysql,
.destroy_func = destroy_mysql,
.update_func = update_mysql,
+ .update2_func = update2_mysql,
.require_func = require_mysql,
.unload_func = unload_mysql,
};
More information about the svn-commits
mailing list