diff options
author | Nic Eigel <nic@eigel.ch> | 2024-06-24 00:08:47 +0200 |
---|---|---|
committer | Nic Eigel <nic@eigel.ch> | 2024-06-24 00:08:47 +0200 |
commit | 74b23809f3e100ad7d63cd746fd5ff3a93b4fdd5 (patch) | |
tree | 84c79a84255730725c110c61054964f641b8c16d /src/auditordb/pg_get_balances.c | |
parent | ed2d7f089fd071d9983d829a18774616f2a6a147 (diff) | |
download | exchange-74b23809f3e100ad7d63cd746fd5ff3a93b4fdd5.tar.xz |
real-time-auditor
Diffstat (limited to 'src/auditordb/pg_get_balances.c')
-rw-r--r-- | src/auditordb/pg_get_balances.c | 188 |
1 files changed, 188 insertions, 0 deletions
diff --git a/src/auditordb/pg_get_balances.c b/src/auditordb/pg_get_balances.c new file mode 100644 index 000000000..8f8f2d682 --- /dev/null +++ b/src/auditordb/pg_get_balances.c @@ -0,0 +1,188 @@ +/* + This file is part of TALER + Copyright (C) 2024 Taler Systems SA + + TALER 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. + + TALER 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 + TALER; see the file COPYING. If not, see <http://www.gnu.org/licenses/> + */ + + +#include "platform.h" +#include "taler_error_codes.h" +#include "taler_dbevents.h" +#include "taler_pq_lib.h" +#include "pg_helper.h" + +#include "pg_get_balances.h" + + +struct BalancesContext +{ + + /** + * Function to call for each bad sig loss. + */ + TALER_AUDITORDB_BalancesCallback cb; + + /** + * Closure for @e cb + */ + void *cb_cls; + + /** + * Plugin context. + */ + struct PostgresClosure *pg; + + /** + * Query status to return. + */ + enum GNUNET_DB_QueryStatus qs; +}; + + +/** + * Helper function for #TAH_PG_get_balances(). + * To be called with the results of a SELECT statement + * that has returned @a num_results results. + * + * @param cls closure of type `struct BalancesContext *` + * @param result the postgres result + * @param num_results the number of results in @a result + */ +static void +balances_cb (void *cls, + PGresult *result, + unsigned int num_results) +{ + struct BalancesContext *dcc = cls; + struct PostgresClosure *pg = dcc->pg; + + for (unsigned int i = 0; i < num_results; i++) + { + uint64_t serial_id; + + struct TALER_AUDITORDB_Balances dc; + + struct GNUNET_PQ_ResultSpec rs[] = { + + GNUNET_PQ_result_spec_uint64 ("row_id", &serial_id), + + GNUNET_PQ_result_spec_string ("balance_key", &dc.balance_key), + TALER_PQ_RESULT_SPEC_AMOUNT ("balance_value", &dc.balance_value), + + + GNUNET_PQ_result_spec_end + }; + enum GNUNET_GenericReturnValue rval; + + if (GNUNET_OK != + GNUNET_PQ_extract_result (result, + rs, + i)) + { + GNUNET_break (0); + dcc->qs = GNUNET_DB_STATUS_HARD_ERROR; + return; + } + + dcc->qs = i + 1; + + rval = dcc->cb (dcc->cb_cls, + serial_id, + &dc); + GNUNET_PQ_cleanup_result (rs); + if (GNUNET_OK != rval) + break; + } +} + + +enum GNUNET_DB_QueryStatus +TAH_PG_get_balances ( + void *cls, + int64_t limit, + uint64_t offset, + bool return_suppressed, + const char *balance_key, + TALER_AUDITORDB_BalancesCallback cb, + void *cb_cls) +{ + + + /*if true, does not filter for a specific balance*/ + bool any_balance = true; + const char *bk; + + if (balance_key != NULL) + { + any_balance = false; + bk = balance_key; + } + else + { + bk = ""; + } + + uint64_t plimit = (uint64_t) ((limit < 0) ? -limit : limit); + + struct PostgresClosure *pg = cls; + struct GNUNET_PQ_QueryParam params[] = { + GNUNET_PQ_query_param_uint64 (&offset), + GNUNET_PQ_query_param_uint64 (&plimit), + GNUNET_PQ_query_param_bool (any_balance), + GNUNET_PQ_query_param_string (bk), + GNUNET_PQ_query_param_end + }; + struct BalancesContext dcc = { + .cb = cb, + .cb_cls = cb_cls, + .pg = pg + }; + enum GNUNET_DB_QueryStatus qs; + + PREPARE (pg, + "auditor_balances_get_desc", + "SELECT" + " row_id," + " balance_key," + " balance_value" + " FROM auditor_balances" + " WHERE (row_id < $1)" + " AND ($3 OR balance_key = $4)" + " ORDER BY row_id DESC" + " LIMIT $2" + ); + PREPARE (pg, + "auditor_balances_get_asc", + "SELECT" + " row_id," + " balance_key," + " balance_value" + " FROM auditor_balances" + " WHERE (row_id > $1)" + " AND ($3 OR balance_key = $4)" + " ORDER BY row_id ASC" + " LIMIT $2" + ); + qs = GNUNET_PQ_eval_prepared_multi_select (pg->conn, + (limit > 0) + ? "auditor_balances_get_asc" + : "auditor_balances_get_desc", + params, + &balances_cb, + &dcc); + + if (qs > 0) + return dcc.qs; + GNUNET_break (GNUNET_DB_STATUS_HARD_ERROR != qs); + return qs; +} |