pg_get_balances.c (3473B)
1 /* 2 This file is part of TALER 3 Copyright (C) 2024 Taler Systems SA 4 5 TALER is free software; you can redistribute it and/or modify it under the 6 terms of the GNU General Public License as published by the Free Software 7 Foundation; either version 3, or (at your option) any later version. 8 9 TALER is distributed in the hope that it will be useful, but WITHOUT ANY 10 WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR 11 A PARTICULAR PURPOSE. See the GNU General Public License for more details. 12 13 You should have received a copy of the GNU General Public License along with 14 TALER; see the file COPYING. If not, see <http://www.gnu.org/licenses/> 15 */ 16 #include "taler/platform.h" 17 #include "taler/taler_error_codes.h" 18 #include "taler/taler_dbevents.h" 19 #include "taler/taler_pq_lib.h" 20 #include "pg_helper.h" 21 #include "pg_get_balances.h" 22 23 24 struct BalancesContext 25 { 26 27 /** 28 * Function to call for each bad sig loss. 29 */ 30 TALER_AUDITORDB_BalancesCallback cb; 31 32 /** 33 * Closure for @e cb 34 */ 35 void *cb_cls; 36 37 /** 38 * Plugin context. 39 */ 40 struct PostgresClosure *pg; 41 42 /** 43 * Query status to return. 44 */ 45 enum GNUNET_DB_QueryStatus qs; 46 }; 47 48 49 /** 50 * Helper function for #TAH_PG_get_balances(). 51 * To be called with the results of a SELECT statement 52 * that has returned @a num_results results. 53 * 54 * @param cls closure of type `struct BalancesContext *` 55 * @param result the postgres result 56 * @param num_results the number of results in @a result 57 */ 58 static void 59 balances_cb (void *cls, 60 PGresult *result, 61 unsigned int num_results) 62 { 63 struct BalancesContext *dcc = cls; 64 struct PostgresClosure *pg = dcc->pg; 65 66 for (unsigned int i = 0; i < num_results; i++) 67 { 68 struct TALER_AUDITORDB_Balances dc; 69 struct GNUNET_PQ_ResultSpec rs[] = { 70 GNUNET_PQ_result_spec_string ("balance_key", 71 &dc.balance_key), 72 TALER_PQ_RESULT_SPEC_AMOUNT ("balance_value", 73 &dc.balance_value), 74 GNUNET_PQ_result_spec_end 75 }; 76 enum GNUNET_GenericReturnValue rval; 77 78 if (GNUNET_OK != 79 GNUNET_PQ_extract_result (result, 80 rs, 81 i)) 82 { 83 GNUNET_break (0); 84 dcc->qs = GNUNET_DB_STATUS_HARD_ERROR; 85 return; 86 } 87 dcc->qs = i + 1; 88 rval = dcc->cb (dcc->cb_cls, 89 &dc); 90 GNUNET_PQ_cleanup_result (rs); 91 if (GNUNET_OK != rval) 92 break; 93 } 94 } 95 96 97 enum GNUNET_DB_QueryStatus 98 TAH_PG_get_balances ( 99 void *cls, 100 const char *balance_key, 101 TALER_AUDITORDB_BalancesCallback cb, 102 void *cb_cls) 103 { 104 struct PostgresClosure *pg = cls; 105 struct GNUNET_PQ_QueryParam params[] = { 106 NULL == balance_key 107 ? GNUNET_PQ_query_param_null () 108 : GNUNET_PQ_query_param_string (balance_key), 109 GNUNET_PQ_query_param_end 110 }; 111 struct BalancesContext dcc = { 112 .cb = cb, 113 .cb_cls = cb_cls, 114 .pg = pg 115 }; 116 enum GNUNET_DB_QueryStatus qs; 117 118 PREPARE (pg, 119 "auditor_balances_get", 120 "SELECT" 121 " balance_key" 122 ",balance_value" 123 " FROM auditor_balances" 124 " WHERE ($1::TEXT IS NULL OR balance_key = $1)" 125 ); 126 qs = GNUNET_PQ_eval_prepared_multi_select ( 127 pg->conn, 128 "auditor_balances_get", 129 params, 130 &balances_cb, 131 &dcc); 132 if (qs > 0) 133 return dcc.qs; 134 GNUNET_break (GNUNET_DB_STATUS_HARD_ERROR != qs); 135 return qs; 136 }