- Implemented DB_REPLACE capability

modules/db_mysql/db_mod.c

@@ -60,6 +60,7 @@ static cmd_export_t cmds[] = {
 	{"db_insert",      (cmd_function)db_insert,      2, 0, 0},
 	{"db_delete",      (cmd_function)db_delete,      2, 0, 0},
 	{"db_update",      (cmd_function)db_update,      2, 0, 0},
+	{"db_replace",     (cmd_function)db_replace,     2, 0, 0},
 	{0, 0, 0, 0, 0}
 };
 

modules/db_mysql/dbase.c

@@ -612,3 +612,46 @@ int db_update(db_con_t* _h, db_key_t* _k, db_op_t* _o, db_val_t* _v,
 	LOG(L_ERR, "db_update: Error in snprintf\n");
 	return -1;
 }
+
+
+/*
+ * Just like insert, but replace the row if it exists
+ */
+int db_replace(db_con_t* handle, db_key_t* keys, db_val_t* vals, int n)
+{
+	int off, ret;
+
+	if (!handle || !keys || !vals) {
+		LOG(L_ERR, "db_replace: Invalid parameter value\n");
+		return -1;
+	}
+
+	ret = snprintf(sql_buf, SQL_BUF_LEN, "replace %s (", CON_TABLE(handle));
+	if (ret < 0 || ret >= SQL_BUF_LEN) goto error;
+	off = ret;
+
+	ret = print_columns(sql_buf + off, SQL_BUF_LEN - off, keys, n);
+	if (ret < 0) return -1;
+	off += ret;
+
+	ret = snprintf(sql_buf + off, SQL_BUF_LEN - off, ") values (");
+	if (ret < 0 || ret >= (SQL_BUF_LEN - off)) goto error;
+	off += ret;
+
+	ret = print_values(CON_CONNECTION(handle), sql_buf + off, SQL_BUF_LEN - off, vals, n);
+	if (ret < 0) return -1;
+	off += ret;
+
+	*(sql_buf + off++) = ')';
+	*(sql_buf + off) = '\0';
+
+	if (submit_query(handle, sql_buf) < 0) {
+	        LOG(L_ERR, "db_replace: Error while submitting query\n");
+		return -2;
+	}
+	return 0;
+
+ error:
+	LOG(L_ERR, "db_replace: Error in snprintf\n");
+	return -1;
+}

modules/db_mysql/dbase.h

@@ -89,6 +89,12 @@ int db_update(db_con_t* _h, db_key_t* _k, db_op_t* _o, db_val_t* _v,
 	      db_key_t* _uk, db_val_t* _uv, int _n, int _un);
 
 
+/*
+ * Just like insert, but replace the row if it exists
+ */
+int db_replace(db_con_t* handle, db_key_t* keys, db_val_t* vals, int n);
+
+
 /*
  * Store name of table that will be used by
  * subsequent database functions