starbitdev/zblock
1
0
Fork 0
mirror of https://github.com/WCBROW01/zblock.git synced 2025-12-11 20:18:07 -05:00
Code Issues Packages Projects Releases Wiki Activity

Abstract out the last remaining SQL queries in timer_retrieve_feeds()

Browse Source
This commit is contained in:
Mina Brown
2024-11-25 00:36:40 -05:00
parent 23ed4a7d96
commit 69be63aa2e
3 changed files with 87 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

65
feed_info.c
View File

@ -23,10 +23,24 @@ static const char *ZBLOCK_FEED_INFO_ERRORS[] = {
"Invalid arguments provided",
"An error was encountered with the feed database",
"The feed already exists",
"The feed does not exist"
"The feed does not exist",
"Finished getting feed info",
"Out of memory"
};
static_assert(sizeof(ZBLOCK_FEED_INFO_ERRORS) / sizeof(*ZBLOCK_FEED_INFO_ERRORS) == ZBLOCK_FEED_INFO_ERRORCOUNT, "Not all feed info errors implemented");
// free all information associated with a minimal feed info struct (does not assume the struct was allocated using malloc)
void zblock_feed_info_minimal_free(zblock_feed_info_minimal *feed_info) {
free(feed_info->last_pubDate);
free(feed_info->url);
}
// free all information associated with a feed info struct (does not assume the struct was allocated using malloc)
void zblock_feed_info_free(zblock_feed_info *feed_info) {
free(feed_info->title);
zblock_feed_info_minimal_free((zblock_feed_info_minimal *) feed_info);
}
// returns a string about the result of a feed_info function
const char *zblock_feed_info_strerror(zblock_feed_info_err error) {
return error < 0 || error >= ZBLOCK_FEED_INFO_ERRORCOUNT ? "Unspecified error" : ZBLOCK_FEED_INFO_ERRORS[error];
@ -43,6 +57,49 @@ time_t pubDate_to_time_t(char *s) {
return timegm(&tm);
}
// Begin retrieval of feed info objects.
zblock_feed_info_err zblock_feed_info_retrieve_list_begin(PGconn *conn) {
if (!conn) return ZBLOCK_FEED_INFO_INVALID_ARGS;
if (!PQsendQueryParams(
conn, "SELECT url, last_pubDate, channel_id from feeds",
0, NULL, NULL, NULL, NULL, 1
)) {
return ZBLOCK_FEED_INFO_DBERROR;
}
PQsetSingleRowMode(conn);
return ZBLOCK_FEED_INFO_OK;
}
// Retrieve the next feed list object.
// On error, no more objects can be retrieved and the returned object is invalid.
zblock_feed_info_err zblock_feed_info_retrieve_list_item(PGconn *conn, zblock_feed_info_minimal *feed_info) {
if (!conn || !feed_info) return ZBLOCK_FEED_INFO_INVALID_ARGS;
PGresult *res = PQgetResult(conn);
if (PQresultStatus(res) != PGRES_SINGLE_TUPLE) {
if (PQresultStatus(res) != PGRES_TUPLES_OK) {
log_error("Unable to retrieve feeds: %s", PQresultErrorMessage(res));
PQclear(res);
return ZBLOCK_FEED_INFO_DBERROR;
} else {
PQclear(res);
return ZBLOCK_FEED_INFO_FINISHED;
}
}
feed_info->url = strdup(PQgetvalue(res, 0, 0));
feed_info->last_pubDate = strdup(PQgetvalue(res, 0, 1));
if (!feed_info->url || !feed_info->last_pubDate) {
PQclear(res);
return ZBLOCK_FEED_INFO_NOMEM;
}
feed_info->channel_id = be64toh(*(uint64_t *) PQgetvalue(res, 0, 2));
PQclear(res);
return ZBLOCK_FEED_INFO_OK;
}
// check if the feed currently exists. the result is in the exists pointer.
zblock_feed_info_err zblock_feed_info_exists(PGconn *conn, const char *url, u64snowflake channel_id, int *exists) {
if (!conn || !exists) return ZBLOCK_FEED_INFO_INVALID_ARGS;
@ -149,11 +206,7 @@ zblock_feed_info_err zblock_feed_info_update(PGconn *conn, zblock_feed_info_mini
);
zblock_feed_info_err result = ZBLOCK_FEED_INFO_OK;
if (PQresultStatus(update_res) != PGRES_COMMAND_OK) {
log_error("Failed to update feed: %s", PQresultErrorMessage(update_res));
result = ZBLOCK_FEED_INFO_DBERROR;
}
if (PQresultStatus(update_res) != PGRES_COMMAND_OK) result = ZBLOCK_FEED_INFO_DBERROR;
PQclear(update_res);
return result;
}