iterate_purses.c (4751B)
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/taler_pq_lib.h" 17 #include "pg_helper.h" 18 #include "auditor-database/iterate_purses.h" 19 20 21 /** 22 * Hard upper bound on the number of records returned by a single 23 * call, regardless of the limit requested by the client. 24 */ 25 #define MAX_RECORDS 50000 26 27 28 struct PursesContext 29 { 30 31 /** 32 * Function to call for each bad sig loss. 33 */ 34 TALER_AUDITORDB_PursesCallback cb; 35 36 /** 37 * Closure for @e cb 38 */ 39 void *cb_cls; 40 41 /** 42 * Plugin context. 43 */ 44 struct TALER_AUDITORDB_PostgresContext *pg; 45 46 /** 47 * Query status to return. 48 */ 49 enum GNUNET_DB_QueryStatus qs; 50 }; 51 52 53 /** 54 * Helper function for #TALER_AUDITORDB_iterate_purses(). 55 * To be called with the results of a SELECT statement 56 * that has returned @a num_results results. 57 * 58 * @param cls closure of type `struct PursesContext *` 59 * @param result the postgres result 60 * @param num_results the number of results in @a result 61 */ 62 static void 63 purses_cb (void *cls, 64 PGresult *result, 65 unsigned int num_results) 66 { 67 struct PursesContext *dcc = cls; 68 struct TALER_AUDITORDB_PostgresContext *pg = dcc->pg; 69 70 for (unsigned int i = 0; i < num_results; i++) 71 { 72 struct TALER_AUDITORDB_Purses dc; 73 struct GNUNET_PQ_ResultSpec rs[] = { 74 GNUNET_PQ_result_spec_uint64 ("auditor_purses_rowid", 75 &dc.auditor_purses_rowid), 76 GNUNET_PQ_result_spec_auto_from_type ("purse_pub", 77 &dc.purse_pub), 78 TALER_PQ_RESULT_SPEC_AMOUNT ("balance", 79 &dc.balance), 80 TALER_PQ_RESULT_SPEC_AMOUNT ("target", 81 &dc.target), 82 GNUNET_PQ_result_spec_absolute_time ("expiration_date", 83 &dc.expiration_date), 84 GNUNET_PQ_result_spec_end 85 }; 86 enum GNUNET_GenericReturnValue rval; 87 88 if (GNUNET_OK != 89 GNUNET_PQ_extract_result (result, 90 rs, 91 i)) 92 { 93 GNUNET_break (0); 94 dcc->qs = GNUNET_DB_STATUS_HARD_ERROR; 95 return; 96 } 97 dcc->qs = i + 1; 98 rval = dcc->cb (dcc->cb_cls, 99 dc.auditor_purses_rowid, 100 &dc); 101 GNUNET_PQ_cleanup_result (rs); 102 if (GNUNET_OK != rval) 103 break; 104 } 105 } 106 107 108 enum GNUNET_DB_QueryStatus 109 TALER_AUDITORDB_iterate_purses (struct TALER_AUDITORDB_PostgresContext *pg, 110 int64_t limit, 111 uint64_t offset, 112 TALER_AUDITORDB_PursesCallback cb, 113 void *cb_cls) 114 { 115 uint64_t plimit = GNUNET_MIN ((uint64_t) MAX_RECORDS, 116 TALER_AUDITORDB_abs_limit (limit)); 117 struct GNUNET_PQ_QueryParam params[] = { 118 GNUNET_PQ_query_param_uint64 (&offset), 119 GNUNET_PQ_query_param_uint64 (&plimit), 120 GNUNET_PQ_query_param_end 121 }; 122 struct PursesContext dcc = { 123 .cb = cb, 124 .cb_cls = cb_cls, 125 .pg = pg 126 }; 127 enum GNUNET_DB_QueryStatus qs; 128 129 PREPARE (pg, 130 "iterate_purses_desc", 131 "SELECT" 132 " auditor_purses_rowid," 133 " purse_pub," 134 " balance," 135 " target," 136 " expiration_date" 137 " FROM auditor_purses" 138 " WHERE (auditor_purses_rowid < $1)" 139 " ORDER BY auditor_purses_rowid DESC" 140 " LIMIT $2" 141 ); 142 PREPARE (pg, 143 "iterate_purses_asc", 144 "SELECT" 145 " auditor_purses_rowid," 146 " purse_pub," 147 " balance," 148 " target," 149 " expiration_date" 150 " FROM auditor_purses" 151 " WHERE (auditor_purses_rowid > $1)" 152 " ORDER BY auditor_purses_rowid ASC" 153 " LIMIT $2" 154 ); 155 qs = GNUNET_PQ_eval_prepared_multi_select ( 156 pg->conn, 157 (limit > 0) 158 ? "iterate_purses_asc" 159 : "iterate_purses_desc", 160 params, 161 &purses_cb, 162 &dcc); 163 164 if (qs > 0) 165 return dcc.qs; 166 GNUNET_break (GNUNET_DB_STATUS_HARD_ERROR != qs); 167 return qs; 168 }