exchange

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

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 }