exchange

Base system with REST service to issue digital coins, run by the payment service provider
Log | Files | Refs | Submodules | README | LICENSE

iterate_balances.c (3662B)

      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_balances.h"
     19 
     20 
     21 struct BalancesContext
     22 {
     23 
     24   /**
     25    * Function to call for each bad sig loss.
     26    */
     27   TALER_AUDITORDB_BalancesCallback cb;
     28 
     29   /**
     30    * Closure for @e cb
     31    */
     32   void *cb_cls;
     33 
     34   /**
     35    * Plugin context.
     36    */
     37   struct TALER_AUDITORDB_PostgresContext *pg;
     38 
     39   /**
     40    * Query status to return.
     41    */
     42   enum GNUNET_DB_QueryStatus qs;
     43 };
     44 
     45 
     46 /**
     47  * Helper function for #TALER_AUDITORDB_iterate_balances().
     48  * To be called with the results of a SELECT statement
     49  * that has returned @a num_results results.
     50  *
     51  * @param cls closure of type `struct BalancesContext *`
     52  * @param result the postgres result
     53  * @param num_results the number of results in @a result
     54  */
     55 static void
     56 balances_cb (void *cls,
     57              PGresult *result,
     58              unsigned int num_results)
     59 {
     60   struct BalancesContext *dcc = cls;
     61   struct TALER_AUDITORDB_PostgresContext *pg = dcc->pg;
     62 
     63   for (unsigned int i = 0; i < num_results; i++)
     64   {
     65     struct TALER_AUDITORDB_Balances dc;
     66     struct GNUNET_PQ_ResultSpec rs[] = {
     67       GNUNET_PQ_result_spec_string ("balance_key",
     68                                     &dc.balance_key),
     69       /* 'balance_value' is nullable in the schema; on NULL the amount
     70          is left invalid instead of failing the extraction for the
     71          entire page. */
     72       GNUNET_PQ_result_spec_allow_null (
     73         TALER_PQ_RESULT_SPEC_AMOUNT ("balance_value",
     74                                      &dc.balance_value),
     75         NULL),
     76       GNUNET_PQ_result_spec_end
     77     };
     78     enum GNUNET_GenericReturnValue rval;
     79 
     80     if (GNUNET_OK !=
     81         GNUNET_PQ_extract_result (result,
     82                                   rs,
     83                                   i))
     84     {
     85       GNUNET_break (0);
     86       dcc->qs = GNUNET_DB_STATUS_HARD_ERROR;
     87       return;
     88     }
     89     dcc->qs = i + 1;
     90     rval = dcc->cb (dcc->cb_cls,
     91                     &dc);
     92     GNUNET_PQ_cleanup_result (rs);
     93     if (GNUNET_OK != rval)
     94       break;
     95   }
     96 }
     97 
     98 
     99 enum GNUNET_DB_QueryStatus
    100 TALER_AUDITORDB_iterate_balances (
    101   struct TALER_AUDITORDB_PostgresContext *pg,
    102   const char *balance_key,
    103   TALER_AUDITORDB_BalancesCallback cb,
    104   void *cb_cls)
    105 {
    106   struct GNUNET_PQ_QueryParam params[] = {
    107     NULL == balance_key
    108     ? GNUNET_PQ_query_param_null ()
    109     : GNUNET_PQ_query_param_string (balance_key),
    110     GNUNET_PQ_query_param_end
    111   };
    112   struct BalancesContext dcc = {
    113     .cb = cb,
    114     .cb_cls = cb_cls,
    115     .pg = pg
    116   };
    117   enum GNUNET_DB_QueryStatus qs;
    118 
    119   PREPARE (pg,
    120            "iterate_balances",
    121            "SELECT"
    122            " balance_key"
    123            ",balance_value"
    124            " FROM auditor_balances"
    125            " WHERE ($1::TEXT IS NULL OR balance_key = $1)"
    126            );
    127   qs = GNUNET_PQ_eval_prepared_multi_select (
    128     pg->conn,
    129     "iterate_balances",
    130     params,
    131     &balances_cb,
    132     &dcc);
    133   if (qs > 0)
    134     return dcc.qs;
    135   GNUNET_break (GNUNET_DB_STATUS_HARD_ERROR != qs);
    136   return qs;
    137 }