--- mqtt/src/pubmqtt.c 2011/11/28 10:17:12 1.1.2.9 +++ mqtt/src/pubmqtt.c 2011/11/28 22:31:19 1.1.2.13 @@ -1,6 +1,9 @@ #include "global.h" +extern const char sql_schema[]; + + /* * mqtt_db_log() Log database connection message * @@ -36,18 +39,25 @@ mqtt_rtlm_open(sl_config *cfg) if (!cfg) return NULL; + sqlite3_config(SQLITE_CONFIG_SERIALIZED); + str = (const char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("name")); if (!str) { mqtt_rtlm_log("Error:: Unknown database name ...\n"); return NULL; } - if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READWRITE, NULL)) { + if (sqlite3_open_v2(str, &sql, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE, NULL)) { MQTT_RTLM_LOG(sql); sqlite3_close(sql); return NULL; } + if (sqlite3_exec(sql, sql_schema, NULL, NULL, NULL)) { + MQTT_RTLM_LOG(sql); + sqlite3_close(sql); + return NULL; + } return sql; } @@ -302,36 +312,66 @@ mqtt_rtlm_delete_topic(sl_config *cfg, sqlite3 *sql, c * @cfg = loaded config * @sql = SQL handle * @topic = topic - * return: -1 error, 0 not found or >0 topics rows + * @retain = retain 0 get only dynamic, >0 get only retained and -1 no matter + * return: NULL error or not found and !=NULL allocated subscribe topics */ -int -mqtt_rtlm_read_topic(sl_config *cfg, sqlite3 *sql, const char *topic) +mqtt_subscr_t * +mqtt_rtlm_read_topic(sl_config *cfg, sqlite3 *sql, const char *topic, char retain) { - int colz, ret = 0; - char *str, szStmt[BUFSIZ] = { 0 }; + int rowz = 0; + char *str, szStr[STRSIZ], szStmt[BUFSIZ] = { 0 }; sqlite3_stmt *stmt; - register int i; + register int j; + mqtt_subscr_t *s = NULL; - if (!cfg || !sql) - return -1; + if (!cfg || !sql || !topic) + return NULL; + switch (retain) { + case -1: + memset(szStr, 0, sizeof szStr); + break; + case 0: + snprintf(szStr, sizeof szStr, "AND Retain = 0"); + break; + default: + snprintf(szStr, sizeof szStr, "AND Retain > 0"); + break; + } + str = (char*) cfg_GetAttribute(cfg, CFG("mqtt_pub"), CFG("tbl_topics")); if (!str) { mqtt_rtlm_log("Error:: not found topics table name"); - return -1; + return NULL; } - snprintf(szStmt, sizeof szStmt, "SELECT Retain, Topic, Value FROM %s WHERE Topic LIKE '%s';", - str, topic); + snprintf(szStmt, sizeof szStmt, "SELECT Retain, Topic, Value FROM %s WHERE Topic LIKE '%s' %s;", + str, topic, szStr); if (sqlite3_prepare_v2(sql, szStmt, strlen(szStmt), &stmt, NULL)) { MQTT_RTLM_LOG(sql); - return -1; + return NULL; } - while (sqlite3_step(stmt) == SQLITE_ROW && (colz = sqlite3_column_count(stmt))) { - for (i = 0; i < colz; i++) { - } + + /* calculate count of rows and allocate subscribe items */ + while (sqlite3_step(stmt) == SQLITE_ROW) + rowz++; + if (!(s = malloc((rowz + 1) * sizeof(mqtt_subscr_t)))) { + mqtt_rtlm_log("Error:: System #%d - %s", errno, strerror(errno)); + goto end; + } else + memset(s, 0, (rowz + 1) * sizeof(mqtt_subscr_t)); + sqlite3_reset(stmt); + + /* fill with data */ + for (j = 0; j < rowz && sqlite3_step(stmt) == SQLITE_ROW; j++) { + s[j].sub_ret = (char) sqlite3_column_int(stmt, 0); + s[j].sub_topic._base = strdup(sqlite3_column_text(stmt, 1)); + s[j].sub_topic._size = strlen(s[j].sub_topic._base); + s[j].sub_value._base = strdup(sqlite3_column_text(stmt, 2)); + s[j].sub_value._size = strlen(s[j].sub_value._base); } +end: sqlite3_finalize(stmt); - return ret; + return s; }