diff options
author | grothoff <grothoff@140774ce-b5e7-0310-ab8b-a85725594a96> | 2010-08-25 19:10:18 +0000 |
---|---|---|
committer | grothoff <grothoff@140774ce-b5e7-0310-ab8b-a85725594a96> | 2010-08-25 19:10:18 +0000 |
commit | 8e84b508b993982c85ca4797c5d70dd3d69e6131 (patch) | |
tree | 29ce98471d30a686c101ffff10b3461f7044491c /src/datastore/plugin_datastore_postgres.c | |
parent | 27b53e4e5ec32e34a4dc0f2ace7ac804f5395fcc (diff) |
postgres hackery
git-svn-id: https://gnunet.org/svn/gnunet@12734 140774ce-b5e7-0310-ab8b-a85725594a96
Diffstat (limited to 'src/datastore/plugin_datastore_postgres.c')
-rw-r--r-- | src/datastore/plugin_datastore_postgres.c | 1190 |
1 files changed, 1190 insertions, 0 deletions
diff --git a/src/datastore/plugin_datastore_postgres.c b/src/datastore/plugin_datastore_postgres.c new file mode 100644 index 0000000000..63a51d0956 --- /dev/null +++ b/src/datastore/plugin_datastore_postgres.c @@ -0,0 +1,1190 @@ +/* + This file is part of GNUnet + (C) 2009, 2010 Christian Grothoff (and other contributing authors) + + GNUnet is free software; you can redistribute it and/or modify + it under the terms of the GNU General Public License as published + by the Free Software Foundation; either version 3, or (at your + option) any later version. + + GNUnet is distributed in the hope that it will be useful, but + WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + General Public License for more details. + + You should have received a copy of the GNU General Public License + along with GNUnet; see the file COPYING. If not, write to the + Free Software Foundation, Inc., 59 Temple Place - Suite 330, + Boston, MA 02111-1307, USA. +*/ + +/** + * @file datastore/plugin_datastore_postgres.c + * @brief postgres-based datastore backend + * @author Christian Grothoff + */ + +#include "platform.h" +#include "plugin_datastore.h" +#include <postgresql/libpq-fe.h> + +#define DEBUG_POSTGRES GNUNET_NO + +#define SELECT_IT_LOW_PRIORITY "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (prio = $1 AND oid > $2) " \ + "ORDER BY prio ASC,oid ASC LIMIT 1) "\ + "UNION "\ + "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (prio > $1 AND oid != $2)"\ + "ORDER BY prio ASC,oid ASC LIMIT 1)"\ + "ORDER BY prio ASC,oid ASC LIMIT 1" + +#define SELECT_IT_NON_ANONYMOUS "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (prio = $1 AND oid < $2)"\ + " AND anonLevel=0 ORDER BY prio DESC,oid DESC LIMIT 1) "\ + "UNION "\ + "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (prio < $1 AND oid != $2)"\ + " AND anonLevel=0 ORDER BY prio DESC,oid DESC LIMIT 1) "\ + "ORDER BY prio DESC,oid DESC LIMIT 1" + +#define SELECT_IT_EXPIRATION_TIME "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (expire = $1 AND oid > $2) "\ + "ORDER BY expire ASC,oid ASC LIMIT 1) "\ + "UNION "\ + "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (expire > $1 AND oid != $2) " \ + "ORDER BY expire ASC,oid ASC LIMIT 1)"\ + "ORDER BY expire ASC,oid ASC LIMIT 1" + + +#define SELECT_IT_MIGRATION_ORDER "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (expire = $1 AND oid < $2)"\ + " AND expire > $3 AND type!=3"\ + " ORDER BY expire DESC,oid DESC LIMIT 1) "\ + "UNION "\ + "(SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 "\ + "WHERE (expire < $1 AND oid != $2)" \ + " AND expire > $3 AND type!=3"\ + " ORDER BY expire DESC,oid DESC LIMIT 1)"\ + "ORDER BY expire DESC,oid DESC LIMIT 1" + +/** + * After how many ms "busy" should a DB operation fail for good? + * A low value makes sure that we are more responsive to requests + * (especially PUTs). A high value guarantees a higher success + * rate (SELECTs in iterate can take several seconds despite LIMIT=1). + * + * The default value of 1s should ensure that users do not experience + * huge latencies while at the same time allowing operations to succeed + * with reasonable probability. + */ +#define BUSY_TIMEOUT GNUNET_TIME_UNIT_SECONDS + + +struct NextRequestClosure +{ + struct Plugin *plugin; + PluginIterator iter; + void *iter_cls; + const char *paramValues[5]; + const char *pname; + int paramLengths[5]; + int nparams; // nparams + struct GNUNET_TIME_Absolute now; + GNUNET_HashCode key; + GNUNET_HashCode vhash; + long long count; + long long off; + long long limit_off; + unsigned long long total; + unsigned long long last_expire; + unsigned int last_rowid; // last_rowid + unsigned int last_prio; + enum GNUNET_BLOCK_Type type; + int end_it; +}; + + +/** + * Context for all functions in this plugin. + */ +struct Plugin +{ + /** + * Our execution environment. + */ + struct GNUNET_DATASTORE_PluginEnvironment *env; + + /** + * Native Postgres database handle. + */ + PGconn *dbh; + + /** + * Closure of the 'next_task' (must be freed if 'next_task' is cancelled). + */ + struct NextRequestClosure *next_task_nc; + + /** + * Pending task with scheduler for running the next request. + */ + GNUNET_SCHEDULER_TaskIdentifier next_task; + + unsigned long long payload; + + unsigned int lastSync; + +}; + + +/** + * Check if the result obtained from Postgres has + * the desired status code. If not, log an error, clear the + * result and return GNUNET_SYSERR. + * + * @return GNUNET_OK if the result is acceptable + */ +static int +check_result (struct Plugin *plugin, + PGresult * ret, + int expected_status, + const char *command, const char *args, int line) +{ + if (ret == NULL) + { + GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR | GNUNET_ERROR_TYPE_BULK, + "datastore-postgres", + "Postgres failed to allocate result for `%s:%s' at %d\n", + command, args, line); + return GNUNET_SYSERR; + } + if (PQresultStatus (ret) != expected_status) + { + GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR | GNUNET_ERROR_TYPE_BULK, + "datastore-postgres", + _("`%s:%s' failed at %s:%d with error: %s"), + command, args, __FILE__, line, PQerrorMessage (plugin->dbh)); + PQclear (ret); + return GNUNET_SYSERR; + } + return GNUNET_OK; +} + +/** + * Run simple SQL statement (without results). + */ +static int +pq_exec (struct Plugin *plugin, + const char *sql, int line) +{ + PGresult *ret; + ret = PQexec (plugin->dbh, sql); + if (GNUNET_OK != check_result (plugin, + ret, + PGRES_COMMAND_OK, "PQexec", sql, line)) + return GNUNET_SYSERR; + PQclear (ret); + return GNUNET_OK; +} + +/** + * Prepare SQL statement. + */ +static int +pq_prepare (struct Plugin *plugin, + const char *name, const char *sql, int nparms, int line) +{ + PGresult *ret; + ret = PQprepare (plugin->dbh, name, sql, nparms, NULL); + if (GNUNET_OK != + check_result (plugin, + ret, PGRES_COMMAND_OK, "PQprepare", sql, line)) + return GNUNET_SYSERR; + PQclear (ret); + return GNUNET_OK; +} + +/** + * @brief Get a database handle + * @return GNUNET_OK on success, GNUNET_SYSERR on error + */ +static int +init_connection (struct Plugin *plugin) +{ + char *conninfo; + PGresult *ret; + + /* Open database and precompile statements */ + conninfo = NULL; + GNUNET_CONFIGURATION_get_value_string (plugin->env->cfg, + "datastore-postgres", + "CONFIG", + &conninfo); + plugin->dbh = PQconnectdb (conninfo == NULL ? "" : conninfo); + GNUNET_free_non_null (conninfo); + if (NULL == plugin->dbh) + { + /* FIXME: warn about out-of-memory? */ + return GNUNET_SYSERR; + } + if (PQstatus (plugin->dbh) != CONNECTION_OK) + { + GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR, + "postgres", + _("Unable to initialize Postgres: %s"), + PQerrorMessage (plugin->dbh)); + PQfinish (plugin->dbh); + plugin->dbh = NULL; + return GNUNET_SYSERR; + } + ret = PQexec (plugin->dbh, + "CREATE TABLE gn080 (" + " size INTEGER NOT NULL DEFAULT 0," + " type INTEGER NOT NULL DEFAULT 0," + " prio INTEGER NOT NULL DEFAULT 0," + " anonLevel INTEGER NOT NULL DEFAULT 0," + " expire BIGINT NOT NULL DEFAULT 0," + " hash BYTEA NOT NULL DEFAULT ''," + " vhash BYTEA NOT NULL DEFAULT ''," + " value BYTEA NOT NULL DEFAULT '')" "WITH OIDS"); + if ( (ret == NULL) || + ( (PQresultStatus (ret) != PGRES_COMMAND_OK) && + (0 != strcmp ("42P07", /* duplicate table */ + PQresultErrorField + (ret, + PG_DIAG_SQLSTATE))))) + { + check_result (plugin, + ret, PGRES_COMMAND_OK, "CREATE TABLE", "gn080", __LINE__); + PQfinish (plugin->dbh); + plugin->dbh = NULL; + return GNUNET_SYSERR; + } + if (PQresultStatus (ret) == PGRES_COMMAND_OK) + { + if ((GNUNET_OK != + pq_exec (plugin, "CREATE INDEX idx_hash ON gn080 (hash)", __LINE__)) || + (GNUNET_OK != + pq_exec (plugin, "CREATE INDEX idx_hash_vhash ON gn080 (hash,vhash)", + __LINE__)) + || (GNUNET_OK != + pq_exec (plugin, "CREATE INDEX idx_prio ON gn080 (prio)", __LINE__)) + || (GNUNET_OK != + pq_exec (plugin, "CREATE INDEX idx_expire ON gn080 (expire)", __LINE__)) + || (GNUNET_OK != + pq_exec (plugin, "CREATE INDEX idx_comb3 ON gn080 (prio,anonLevel)", + __LINE__)) + || (GNUNET_OK != + pq_exec + (plugin, "CREATE INDEX idx_comb4 ON gn080 (prio,hash,anonLevel)", + __LINE__)) + || (GNUNET_OK != + pq_exec (plugin, "CREATE INDEX idx_comb7 ON gn080 (expire,hash)", + __LINE__))) + { + PQclear (ret); + PQfinish (plugin->dbh); + plugin->dbh = NULL; + return GNUNET_SYSERR; + } + } + PQclear (ret); +#if 1 + ret = PQexec (plugin->dbh, + "ALTER TABLE gn080 ALTER value SET STORAGE EXTERNAL"); + if ( (ret == NULL) || + ((PQresultStatus (ret) != PGRES_COMMAND_OK) ) ) + { + check_result (plugin, + ret, PGRES_COMMAND_OK, "ALTER TABLE", "gn080", __LINE__); + PQfinish (plugin->dbh); + plugin->dbh = NULL; + return GNUNET_SYSERR; + } + ret = PQexec (plugin->dbh, + "ALTER TABLE gn080 ALTER hash SET STORAGE PLAIN"); + if ( (ret == NULL) || + ((PQresultStatus (ret) != PGRES_COMMAND_OK) ) ) + { + check_result (plugin, + ret, PGRES_COMMAND_OK, "ALTER TABLE", "gn080", __LINE__); + PQfinish (plugin->dbh); + plugin->dbh = NULL; + return GNUNET_SYSERR; + } + ret = PQexec (plugin->dbh, + "ALTER TABLE gn080 ALTER vhash SET STORAGE PLAIN"); + if ( (ret == NULL) || + ((PQresultStatus (ret) != PGRES_COMMAND_OK) ) ) + { + check_result (plugin, + ret, PGRES_COMMAND_OK, "ALTER TABLE", "gn080", __LINE__); + PQfinish (plugin->dbh); + plugin->dbh = NULL; + return GNUNET_SYSERR; + } +#endif + if ((GNUNET_OK != + pq_prepare (plugin, + "getvt", + "SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 " + "WHERE hash=$1 AND vhash=$2 AND type=$3 " + "AND oid >= $4 ORDER BY oid ASC LIMIT 1 OFFSET $5", + 5, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "gett", + "SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 " + "WHERE hash=$1 AND type=$2" + "AND oid >= $3 ORDER BY oid ASC LIMIT 1 OFFSET $4", + 4, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "getv", + "SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 " + "WHERE hash=$1 AND vhash=$2" + "AND oid >= $3 ORDER BY oid ASC LIMIT 1 OFFSET $4", + 4, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "get", + "SELECT size, type, prio, anonLevel, expire, hash, value, oid FROM gn080 " + "WHERE hash=$1" + "AND oid >= $2 ORDER BY oid ASC LIMIT 1 OFFSET $3", + 3, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "put", + "INSERT INTO gn080 (size, type, prio, anonLevel, expire, hash, vhash, value) " + "VALUES ($1, $2, $3, $4, $5, $6, $7, $8)", + 8, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "update", + "UPDATE gn080 SET prio = prio + $1, expire = CASE WHEN expire < $2 THEN $2 ELSE expire END " + "WHERE oid = $3", + 3, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "select_low_priority", + SELECT_IT_LOW_PRIORITY, + 2, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "select_non_anonymous", + SELECT_IT_NON_ANONYMOUS, + 2, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "select_expiration_time", + SELECT_IT_EXPIRATION_TIME, + 2, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "select_migration_order", + SELECT_IT_MIGRATION_ORDER, + 3, + __LINE__)) || + (GNUNET_OK != + pq_prepare (plugin, + "delrow", + "DELETE FROM gn080 " "WHERE oid=$1", 1, __LINE__))) + { + PQfinish (plugin->dbh); + plugin->dbh = NULL; + return GNUNET_SYSERR; + } + return GNUNET_OK; +} + + +/** + * Delete the row identified by the given rowid (qid + * in postgres). + * + * @return GNUNET_OK on success + */ +static int +delete_by_rowid (struct Plugin *plugin, + unsigned int rowid) +{ + const char *paramValues[] = { (const char *) &rowid }; + int paramLengths[] = { sizeof (rowid) }; + const int paramFormats[] = { 1 }; + PGresult *ret; + + ret = PQexecPrepared (plugin->dbh, + "delrow", + 1, paramValues, paramLengths, paramFormats, 1); + if (GNUNET_OK != + check_result (plugin, + ret, PGRES_COMMAND_OK, "PQexecPrepared", "delrow", + __LINE__)) + { + return GNUNET_SYSERR; + } + PQclear (ret); + return GNUNET_OK; +} + + +/** + * Get an estimate of how much space the database is + * currently using. + * + * @param cls our "struct Plugin*" + * @return number of bytes used on disk + */ +static unsigned long long +postgres_plugin_get_size (void *cls) +{ + struct Plugin *plugin = cls; + double ret; + + ret = plugin->payload; + return (unsigned long long) (ret * 1.00); + /* benchmarking shows XX% overhead */ +} + + +/** + * Store an item in the datastore. + * + * @param cls closure + * @param key key for the item + * @param size number of bytes in data + * @param data content stored + * @param type type of the content + * @param priority priority of the content + * @param anonymity anonymity-level for the content + * @param expiration expiration time for the content + * @param msg set to error message + * @return GNUNET_OK on success + */ +static int +postgres_plugin_put (void *cls, + const GNUNET_HashCode * key, + uint32_t size, + const void *data, + enum GNUNET_BLOCK_Type type, + uint32_t priority, + uint32_t anonymity, + struct GNUNET_TIME_Absolute expiration, + char **msg) +{ + struct Plugin *plugin = cls; + GNUNET_HashCode vhash; + PGresult *ret; + const char *paramValues[] = { + (const char *) &size, + (const char *) &type, + (const char *) &priority, + (const char *) &anonymity, + (const char *) &expiration.value, + (const char *) key, + (const char *) &vhash, + (const char *) data + }; + int paramLengths[] = { + sizeof (size), + sizeof (type), + sizeof (priority), + sizeof (anonymity), + sizeof (expiration.value), + sizeof (GNUNET_HashCode), + sizeof (GNUNET_HashCode), + size + }; + const int paramFormats[] = { 1, 1, 1, 1, 1, 1, 1, 1 }; + + GNUNET_CRYPTO_hash (data, size, &vhash); + ret = PQexecPrepared (plugin->dbh, + "put", 8, paramValues, paramLengths, paramFormats, 1); + if (GNUNET_OK != check_result (plugin, ret, + PGRES_COMMAND_OK, + "PQexecPrepared", "put", __LINE__)) + return GNUNET_SYSERR; + PQclear (ret); + plugin->payload += size; + return GNUNET_OK; +} + +/** + * Function invoked on behalf of a "PluginIterator" + * asking the database plugin to call the iterator + * with the next item. + * + * @param cls the 'struct NextRequestClosure' + * @param tc scheduler context + */ +static void +postgres_next_request_cont (void *next_cls, + const struct GNUNET_SCHEDULER_TaskContext *tc) +{ + struct NextRequestClosure *nrc = next_cls; + struct Plugin *plugin = nrc->plugin; + const int paramFormats[] = { 1, 1, 1, 1, 1 }; + int iret; + PGresult *res; + enum GNUNET_BLOCK_Type type; + unsigned int anonymity; + unsigned int priority; + unsigned int size; + unsigned int rowid; + struct GNUNET_TIME_Absolute expiration_time; + GNUNET_HashCode key; + + plugin->next_task = GNUNET_SCHEDULER_NO_TASK; + plugin->next_task_nc = NULL; + if (GNUNET_YES == nrc->end_it) + { + nrc->iter (nrc->iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + GNUNET_free (nrc); + return; + } + + + if (nrc->count == 0) + nrc->limit_off = nrc->off; + else + nrc->limit_off = 0; + + res = PQexecPrepared (plugin->dbh, + nrc->pname, + nrc->nparams, + nrc->paramValues, + nrc->paramLengths, + paramFormats, 1); + if (GNUNET_OK != check_result (plugin, + res, + PGRES_TUPLES_OK, + "PQexecPrepared", + nrc->pname, + __LINE__)) + { + nrc->iter (nrc->iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + GNUNET_free (nrc); + return; + } + + if (0 == PQntuples (res)) + { + /* no result */ + nrc->iter (nrc->iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + GNUNET_free (nrc); + return; + } + if ((1 != PQntuples (res)) || + (8 != PQnfields (res)) || + (sizeof (unsigned int) != PQfsize (res, 0)) || + (sizeof (unsigned int) != PQfsize (res, 7))) + { + GNUNET_break (0); + nrc->iter (nrc->iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + GNUNET_free (nrc); + return; + } + rowid = *(unsigned int *) PQgetvalue (res, 0, 7); + size = *(unsigned int *) PQgetvalue (res, 0, 0); + if ((sizeof (unsigned int) != PQfsize (res, 1)) || + (sizeof (unsigned int) != PQfsize (res, 2)) || + (sizeof (unsigned int) != PQfsize (res, 3)) || + (sizeof (unsigned long long) != PQfsize (res, 4)) || + (sizeof (GNUNET_HashCode) != PQgetlength (res, 0, 5)) || + (size != PQgetlength (res, 0, 6))) + { + GNUNET_break (0); + delete_by_rowid (plugin, rowid); + nrc->iter (nrc->iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + GNUNET_free (nrc); + return; + } + + type = *(unsigned int *) PQgetvalue (res, 0, 1); + priority = *(unsigned int *) PQgetvalue (res, 0, 2); + anonymity = *(unsigned int *) PQgetvalue (res, 0, 3); + expiration_time.value = *(unsigned long long *) PQgetvalue (res, 0, 4); + size = PQgetlength (res, 0, 6); + memcpy (&key, PQgetvalue (res, 0, 5), sizeof (GNUNET_HashCode)); + + nrc->last_prio = priority; + nrc->last_expire = expiration_time.value; + nrc->last_rowid = rowid + 1; + nrc->count++; + iret = nrc->iter (nrc->iter_cls, + nrc, + &key, + size, + PQgetvalue (res, 0, 6), + (enum GNUNET_BLOCK_Type) type, + priority, + anonymity, + expiration_time, + rowid); + if (iret == GNUNET_SYSERR) + return; + if (iret == GNUNET_NO) + { + plugin->payload -= size; + delete_by_rowid (plugin, rowid); + } +} + + + +/** + * Function invoked on behalf of a "PluginIterator" + * asking the database plugin to call the iterator + * with the next item. + * + * @param next_cls whatever argument was given + * to the PluginIterator as "next_cls". + * @param end_it set to GNUNET_YES if we + * should terminate the iteration early + * (iterator should be still called once more + * to signal the end of the iteration). + */ +static void +postgres_plugin_next_request (void *next_cls, + int end_it) +{ + struct NextRequestClosure *nrc = next_cls; + + if (GNUNET_YES == end_it) + nrc->end_it = GNUNET_YES; + nrc->plugin->next_task_nc = nrc; + nrc->plugin->next_task = GNUNET_SCHEDULER_add_now (nrc->plugin->env->sched, + &postgres_next_request_cont, + nrc); +} + + +/** + * Update the priority for a particular key in the datastore. If + * the expiration time in value is different than the time found in + * the datastore, the higher value should be kept. For the + * anonymity level, the lower value is to be used. The specified + * priority should be added to the existing priority, ignoring the + * priority in value. + * + * Note that it is possible for multiple values to match this put. + * In that case, all of the respective values are updated. + * + * @param cls our "struct Plugin*" + * @param uid unique identifier of the datum + * @param delta by how much should the priority + * change? If priority + delta < 0 the + * priority should be set to 0 (never go + * negative). + * @param expire new expiration time should be the + * MAX of any existing expiration time and + * this value + * @param msg set to error message + * @return GNUNET_OK on success + */ +static int +postgres_plugin_update (void *cls, + uint64_t uid, + int delta, struct GNUNET_TIME_Absolute expire, + char **msg) +{ + struct Plugin *plugin = cls; + unsigned int oid = (unsigned int) uid; /* only 32 bit for postgres */ + PGresult *ret; + + const char *paramValues[] = { + (const char *) &delta, + (const char *) &expire.value, + (const char *) &oid, + }; + int paramLengths[] = { + sizeof (delta), + sizeof (expire.value), + sizeof (oid), + }; + const int paramFormats[] = { 1, 1, 1 }; + + ret = PQexecPrepared (plugin->dbh, + "update", + 3, paramValues, paramLengths, paramFormats, 1); + if (GNUNET_OK != check_result (plugin, + ret, + PGRES_COMMAND_OK, + "PQexecPrepared", "update", __LINE__)) + return GNUNET_SYSERR; + PQclear (ret); + return GNUNET_OK; +} + + +/** + * Call a method for each key in the database and + * call the callback method on it. + * + * @param type entries of which type should be considered? + * @param iter maybe NULL (to just count); iter + * should return GNUNET_SYSERR to abort the + * iteration, GNUNET_NO to delete the entry and + * continue and GNUNET_OK to continue iterating + */ +static void +postgres_iterate (struct Plugin *plugin, + unsigned int type, + int is_asc, + unsigned int iter_select, + PluginIterator iter, void *iter_cls) +{ + struct NextRequestClosure *nrc; + + nrc = GNUNET_malloc (sizeof (struct NextRequestClosure)); + nrc->plugin = plugin; + nrc->iter = iter; + nrc->iter_cls = iter_cls; + if (is_asc) + { + nrc->last_prio = 0; + nrc->last_rowid = 0; + nrc->last_expire = 0; + } + else + { + nrc->last_prio = 0x7FFFFFFFL; + nrc->last_rowid = 0xFFFFFFFF; + nrc->last_expire = 0x7FFFFFFFFFFFFFFFLL; + } + switch (iter_select) + { + case 0: + nrc->pname = "select_low_priority"; + nrc->nparams = 2; + nrc->paramValues[0] = (const char *) &nrc->last_prio; + nrc->paramValues[1] = (const char *) &nrc->last_rowid; + nrc->paramLengths[0] = sizeof (nrc->last_prio); + nrc->paramLengths[1] = sizeof (nrc->last_rowid); + break; + case 1: + nrc->pname = "select_non_anonymous"; + nrc->nparams = 2; + nrc->paramValues[0] = (const char *) &nrc->last_prio; + nrc->paramValues[1] = (const char *) &nrc->last_rowid; + nrc->paramLengths[0] = sizeof (nrc->last_prio); + nrc->paramLengths[1] = sizeof (nrc->last_rowid); + break; + case 2: + nrc->pname = "select_expiration_time"; + nrc->nparams = 2; + nrc->paramValues[0] = (const char *) &nrc->last_expire; + nrc->paramValues[1] = (const char *) &nrc->last_rowid; + nrc->paramLengths[0] = sizeof (nrc->last_expire); + nrc->paramLengths[1] = sizeof (nrc->last_rowid); + break; + case 3: + nrc->pname = "select_migration_order"; + nrc->nparams = 3; + nrc->paramValues[0] = (const char *) &nrc->last_expire; + nrc->paramValues[1] = (const char *) &nrc->last_rowid; + nrc->paramValues[2] = (const char *) &nrc->now; + nrc->paramLengths[0] = sizeof (nrc->last_expire); + nrc->paramLengths[1] = sizeof (nrc->last_rowid); + nrc->paramLengths[2] = sizeof (nrc->now); + break; + default: + GNUNET_break (0); + iter (iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + return; + } + nrc->now = GNUNET_TIME_absolute_get (); + postgres_plugin_next_request (nrc, + GNUNET_NO); +} + + +/** + * Select a subset of the items in the datastore and call + * the given iterator for each of them. + * + * @param cls our "struct Plugin*" + * @param type entries of which type should be considered? + * Use 0 for any type. + * @param iter function to call on each matching value; + * will be called once with a NULL value at the end + * @param iter_cls closure for iter + */ +static void +postgres_plugin_iter_low_priority (void *cls, + enum GNUNET_BLOCK_Type type, + PluginIterator iter, + void *iter_cls) +{ + struct Plugin *plugin = cls; + + postgres_iterate (plugin, + type, + GNUNET_YES, 0, + iter, iter_cls); +} + + + + +/** + * Iterate over the results for a particular key + * in the datastore. + * + * @param cls closure + * @param key maybe NULL (to match all entries) + * @param vhash hash of the value, maybe NULL (to + * match all values that have the right key). + * Note that for DBlocks there is no difference + * betwen key and vhash, but for other blocks + * there may be! + * @param type entries of which type are relevant? + * Use 0 for any type. + * @param iter function to call on each matching value; + * will be called once with a NULL value at the end + * @param iter_cls closure for iter + */ +static void +postgres_plugin_get (void *cls, + const GNUNET_HashCode * key, + const GNUNET_HashCode * vhash, + enum GNUNET_BLOCK_Type type, + PluginIterator iter, void *iter_cls) +{ + struct Plugin *plugin = cls; + struct NextRequestClosure *nrc; + const int paramFormats[] = { 1, 1, 1, 1, 1 }; + PGresult *ret; + + if (key == NULL) + { + postgres_plugin_iter_low_priority (plugin, type, + iter, iter_cls); + return; + } + nrc = GNUNET_malloc (sizeof (struct NextRequestClosure)); + nrc->plugin = plugin; + nrc->iter = iter; + nrc->iter_cls = iter_cls; + nrc->key = *key; + if (vhash != NULL) + nrc->vhash = *vhash; + nrc->paramValues[0] = (const char*) &nrc->key; + nrc->paramLengths[0] = sizeof (GNUNET_HashCode); + nrc->type = type; + if (type != 0) + { + if (vhash != NULL) + { + nrc->paramValues[1] = (const char *) &nrc->vhash; + nrc->paramLengths[1] = sizeof (nrc->vhash); + nrc->paramValues[2] = (const char *) &nrc->type; + nrc->paramLengths[2] = sizeof (nrc->type); + nrc->paramValues[3] = (const char *) &nrc->last_rowid; + nrc->paramLengths[3] = sizeof (nrc->last_rowid); + nrc->paramValues[4] = (const char *) &nrc->limit_off; + nrc->paramLengths[4] = sizeof (nrc->limit_off); + nrc->nparams = 5; + nrc->pname = "getvt"; + ret = PQexecParams (plugin->dbh, + "SELECT count(*) FROM gn080 WHERE hash=$1 AND vhash=$2 AND type=$3", + 3, + NULL, + nrc->paramValues, + nrc->paramLengths, + paramFormats, 1); + } + else + { + nrc->paramValues[1] = (const char *) &nrc->type; + nrc->paramLengths[1] = sizeof (nrc->type); + nrc->paramValues[2] = (const char *) &nrc->last_rowid; + nrc->paramLengths[2] = sizeof (nrc->last_rowid); + nrc->paramValues[3] = (const char *) &nrc->limit_off; + nrc->paramLengths[3] = sizeof (nrc->limit_off); + nrc->nparams = 4; + nrc->pname = "gett"; + ret = PQexecParams (plugin->dbh, + "SELECT count(*) FROM gn080 WHERE hash=$1 AND type=$2", + 2, + NULL, + nrc->paramValues, + nrc->paramLengths, + paramFormats, 1); + } + } + else + { + if (vhash != NULL) + { + nrc->paramValues[1] = (const char *) &nrc->vhash; + nrc->paramLengths[1] = sizeof (nrc->vhash); + nrc->paramValues[2] = (const char *) &nrc->last_rowid; + nrc->paramLengths[2] = sizeof (nrc->last_rowid); + nrc->paramValues[3] = (const char *) &nrc->limit_off; + nrc->paramLengths[3] = sizeof (nrc->limit_off); + nrc->nparams = 4; + nrc->pname = "getv"; + ret = PQexecParams (plugin->dbh, + "SELECT count(*) FROM gn080 WHERE hash=$1 AND vhash=$2", + 2, + NULL, + nrc->paramValues, + nrc->paramLengths, + paramFormats, 1); + } + else + { + nrc->paramValues[1] = (const char *) &nrc->last_rowid; + nrc->paramLengths[1] = sizeof (nrc->last_rowid); + nrc->paramValues[2] = (const char *) &nrc->limit_off; + nrc->paramLengths[2] = sizeof (nrc->limit_off); + nrc->nparams = 3; + nrc->pname = "get"; + ret = PQexecParams (plugin->dbh, + "SELECT count(*) FROM gn080 WHERE hash=$1", + 1, + NULL, + nrc->paramValues, + nrc->paramLengths, + paramFormats, 1); + } + } + if (GNUNET_OK != check_result (plugin, + ret, + PGRES_TUPLES_OK, + "PQexecParams", + nrc->pname, + __LINE__)) + { + iter (iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + return; + } + if ((PQntuples (ret) != 1) || + (PQnfields (ret) != 1) || + (PQgetlength (ret, 0, 0) != sizeof (unsigned long long))) + { + GNUNET_break (0); + PQclear (ret); + iter (iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + return; + } + nrc->total = GNUNET_ntohll (*(const unsigned long long *) PQgetvalue (ret, 0, 0)); + PQclear (ret); + if (nrc->total == 0) + { + iter (iter_cls, + NULL, NULL, 0, NULL, 0, 0, 0, + GNUNET_TIME_UNIT_ZERO_ABS, 0); + return; + } + nrc->off = GNUNET_CRYPTO_random_u64 (GNUNET_CRYPTO_QUALITY_WEAK, + nrc->total); + postgres_plugin_next_request (nrc, + GNUNET_NO); +} + + +/** + * Select a subset of the items in the datastore and call + * the given iterator for each of them. + * + * @param cls our "struct Plugin*" + * @param type entries of which type should be considered? + * Use 0 for any type. + * @param iter function to call on each matching value; + * will be called once with a NULL value at the end + * @param iter_cls closure for iter + */ +static void +postgres_plugin_iter_zero_anonymity (void *cls, + enum GNUNET_BLOCK_Type type, + PluginIterator iter, + void *iter_cls) +{ + struct Plugin *plugin = cls; + + postgres_iterate (plugin, + type, GNUNET_NO, 1, + iter, iter_cls); +} + + +/** + * Select a subset of the items in the datastore and call + * the given iterator for each of them. + * + * @param cls our "struct Plugin*" + * @param type entries of which type should be considered? + * Use 0 for any type. + * @param iter function to call on each matching value; + * will be called once with a NULL value at the end + * @param iter_cls closure for iter + */ +static void +postgres_plugin_iter_ascending_expiration (void *cls, + enum GNUNET_BLOCK_Type type, + PluginIterator iter, + void *iter_cls) +{ + struct Plugin *plugin = cls; + + postgres_iterate (plugin, type, GNUNET_YES, 2, + iter, iter_cls); +} + + + +/**< |