pq_prepare.c

Go to the documentation of this file.

/*
   This file is part of GNUnet
   Copyright (C) 2017, 2019 GNUnet e.V.
 
   GNUnet is free software: you can redistribute it and/or modify it
   under the terms of the GNU Affero General Public License as published
   by the Free Software Foundation, either version 3 of the License,
   or (at your option) any later version.
 
   GNUnet 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
   Affero General Public License for more details.
 
   You should have received a copy of the GNU Affero General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
     SPDX-License-Identifier: AGPL3.0-or-later
 */
#include "platform.h"
#include "pq.h"
 
 
struct GNUNET_PQ_PreparedStatement
GNUNET_PQ_make_prepare (const char *name,
                        const char *sql)
{
  struct GNUNET_PQ_PreparedStatement ps = {
    .name = name,
    .sql = sql
  };
 
  /* PostgreSQL truncates prepared statements at 63 characters,
     assert this one is short enough or at least fail hard if
     it is not. */
  GNUNET_assert (strlen (name) < 64);
  return ps;
}
 
 
enum GNUNET_GenericReturnValue
GNUNET_PQ_prepare_anon (struct GNUNET_PQ_Context *pg,
                        const char *sql)
{
  PGresult *ret;
 
  GNUNET_log_from (GNUNET_ERROR_TYPE_INFO,
                   "pq",
                   "Preparing SQL: %s\n",
                   sql);
  ret = PQprepare (pg->conn,
                   "",
                   sql,
                   0,
                   NULL);
  if (PGRES_COMMAND_OK != PQresultStatus (ret))
  {
    GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
                     "pq",
                     "PQprepare (`%s') failed with error: %s\n",
                     sql,
                     PQerrorMessage (pg->conn));
    PQclear (ret);
    return GNUNET_SYSERR;
  }
  PQclear (ret);
  return GNUNET_OK;
}
 
 
enum GNUNET_GenericReturnValue
GNUNET_PQ_prepare_statements (struct GNUNET_PQ_Context *db,
                              const struct GNUNET_PQ_PreparedStatement *ps)
{
  for (unsigned int i = 0;
       NULL != ps[i].name;
       i++)
  {
    PGresult *ret;
 
    GNUNET_log_from (GNUNET_ERROR_TYPE_INFO,
                     "pq",
                     "Preparing SQL statement `%s' as `%s'\n",
                     ps[i].sql,
                     ps[i].name);
    ret = PQprepare (db->conn,
                     ps[i].name,
                     ps[i].sql,
                     0,
                     NULL);
    if (PGRES_COMMAND_OK != PQresultStatus (ret))
    {
      GNUNET_log_from (GNUNET_ERROR_TYPE_ERROR,
                       "pq",
                       "PQprepare (`%s' as `%s') failed with error: %s\n",
                       ps[i].sql,
                       ps[i].name,
                       PQerrorMessage (db->conn));
      PQclear (ret);
      ret = PQdescribePrepared (db->conn,
                                ps[i].name);
      if (PGRES_COMMAND_OK != PQresultStatus (ret))
      {
        PQclear (ret);
        return GNUNET_SYSERR;
      }
      GNUNET_log_from (GNUNET_ERROR_TYPE_WARNING,
                       "pq",
                       "Statement `%s' already known. Ignoring the issue in the hope that you are using connection pooling...\n",
                       ps[i].name);
    }
    PQclear (ret);
  }
  return GNUNET_OK;
}
 
 
/* end of pq/pq_prepare.c */