--- mqtt/src/pubmqtt.c 2011/11/28 10:17:12 1.1.2.9 +++ mqtt/src/pubmqtt.c 2011/11/28 13:19:39 1.1.2.10 @@ -302,36 +302,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 = *sqlite3_column_text(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; }