1 files changed, 100 insertions, 0 deletions
diff --git a/src/auditordb/pg_update_balance.c b/src/auditordb/pg_update_balance.c
new file mode 100644
index 000000000..eff03056b
--- /dev/null
+++ b/src/auditordb/pg_update_balance.c
@@ -0,0 +1,100 @@
+/*
+   This file is part of TALER
+   Copyright (C) 2023 Taler Systems SA
+
+   TALER is free software; you can redistribute it and/or modify it under the
+   terms of the GNU General Public License as published by the Free Software
+   Foundation; either version 3, or (at your option) any later version.
+
+   TALER is distributed in the hope that it will be useful, but WITHOUT ANY
+   WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR
+   A PARTICULAR PURPOSE.  See the GNU General Public License for more details.
+
+   You should have received a copy of the GNU General Public License along with
+   TALER; see the file COPYING.  If not, see <http://www.gnu.org/licenses/>
+ */
+/**
+ * @file auditordb/pg_update_balance.c
+ * @brief Implementation of the update_balance function for Postgres
+ * @author Christian Grothoff
+ */
+#include "platform.h"
+#include "taler_error_codes.h"
+#include "taler_dbevents.h"
+#include "taler_pq_lib.h"
+#include "pg_update_balance.h"
+#include "pg_helper.h"
+
+
+enum GNUNET_DB_QueryStatus
+TAH_PG_update_balance (
+  void *cls,
+  const char *balance_key,
+  const struct TALER_Amount *balance_amount,
+  ...)
+{
+  struct PostgresClosure *pg = cls;
+  unsigned int cnt = 1;
+  va_list ap;
+
+  va_start (ap,
+            balance_amount);
+  while (NULL != va_arg (ap,
+                         const char *))
+  {
+    cnt++;
+    (void) va_arg (ap,
+                   const struct TALER_Amount *);
+  }
+  va_end (ap);
+  {
+    const char *keys[cnt];
+    struct TALER_Amount amounts[cnt];
+    unsigned int off = 1;
+    struct GNUNET_PQ_QueryParam params[] = {
+      GNUNET_PQ_query_param_array_ptrs_string (cnt,
+                                               keys,
+                                               pg->conn),
+      TALER_PQ_query_param_array_amount (cnt,
+                                         amounts,
+                                         pg->conn),
+      GNUNET_PQ_query_param_end
+    };
+    enum GNUNET_DB_QueryStatus qs;
+
+    keys[0] = balance_key;
+    amounts[0] = *balance_amount;
+
+    va_start (ap,
+              balance_amount);
+    while (off < cnt)
+    {
+      keys[off] = va_arg (ap,
+                          const char *);
+      amounts[off] = *va_arg (ap,
+                              const struct TALER_Amount *);
+      off++;
+    }
+    GNUNET_assert (NULL == va_arg (ap,
+                                   const char *));
+    va_end (ap);
+
+    PREPARE (pg,
+             "auditor_balance_update",
+             "UPDATE auditor_balances"
+             "  SET balance_value.val=data.val"
+             "     ,balance_value.frac=data.frac"
+             "  FROM ("
+             "    SELECT *"
+             "      FROM UNNEST (CAST($1 AS TEXT[]),"
+             "                   CAST($2 AS taler_amount[]))"
+             "      AS t(key,val,frac)"
+             "  ) AS data"
+             " WHERE auditor_balances.balance_key=data.key;");
+    qs = GNUNET_PQ_eval_prepared_non_select (pg->conn,
+                                             "auditor_balance_update",
+                                             params);
+    GNUNET_PQ_cleanup_query_params_closures (params);
+    return qs;
+  }
+}