Line data    Source code

       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            0 : balances_cb (void *cls,
      57              :              PGresult *result,
      58              :              unsigned int num_results)
      59              : {
      60            0 :   struct BalancesContext *dcc = cls;
      61            0 :   struct TALER_AUDITORDB_PostgresContext *pg = dcc->pg;
      62              : 
      63            0 :   for (unsigned int i = 0; i < num_results; i++)
      64              :   {
      65              :     struct TALER_AUDITORDB_Balances dc;
      66            0 :     struct GNUNET_PQ_ResultSpec rs[] = {
      67            0 :       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            0 :       GNUNET_PQ_result_spec_allow_null (
      73            0 :         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            0 :     if (GNUNET_OK !=
      81            0 :         GNUNET_PQ_extract_result (result,
      82              :                                   rs,
      83              :                                   i))
      84              :     {
      85            0 :       GNUNET_break (0);
      86            0 :       dcc->qs = GNUNET_DB_STATUS_HARD_ERROR;
      87            0 :       return;
      88              :     }
      89            0 :     dcc->qs = i + 1;
      90            0 :     rval = dcc->cb (dcc->cb_cls,
      91              :                     &dc);
      92            0 :     GNUNET_PQ_cleanup_result (rs);
      93            0 :     if (GNUNET_OK != rval)
      94            0 :       break;
      95              :   }
      96              : }
      97              : 
      98              : 
      99              : enum GNUNET_DB_QueryStatus
     100            0 : 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            0 :   struct GNUNET_PQ_QueryParam params[] = {
     107              :     NULL == balance_key
     108            0 :     ? GNUNET_PQ_query_param_null ()
     109            0 :     : GNUNET_PQ_query_param_string (balance_key),
     110              :     GNUNET_PQ_query_param_end
     111              :   };
     112            0 :   struct BalancesContext dcc = {
     113              :     .cb = cb,
     114              :     .cb_cls = cb_cls,
     115              :     .pg = pg
     116              :   };
     117              :   enum GNUNET_DB_QueryStatus qs;
     118              : 
     119            0 :   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            0 :   qs = GNUNET_PQ_eval_prepared_multi_select (
     128              :     pg->conn,
     129              :     "iterate_balances",
     130              :     params,
     131              :     &balances_cb,
     132              :     &dcc);
     133            0 :   if (qs > 0)
     134            0 :     return dcc.qs;
     135            0 :   GNUNET_break (GNUNET_DB_STATUS_HARD_ERROR != qs);
     136            0 :   return qs;
     137              : }