PowerDNS / pdns / 19957761886

/*

 * This file is part of PowerDNS or dnsdist.

 * Copyright -- PowerDNS.COM B.V. and its contributors

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of version 2 of the GNU General Public License as

 * published by the Free Software Foundation.

 *

 * In addition, for the avoidance of any doubt, permission is granted to

 * link this program with OpenSSL and to (re)distribute the binaries

 * produced as the result of such linking.

 *

 * This program 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 this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

 */

#pragma once

#include "pdns/namespaces.hh"

#include "pdns/backends/gsql/ssql.hh"

#include <libpq-fe.h>

class SPgSQL : public SSql

{

public:

  SPgSQL(const string& database, const string& host = "", const string& port = "",

         const string& user = "", const string& password = "",

         const string& extra_connection_parameters = "", const bool use_prepared = true);

  ~SPgSQL() override;

  SSqlException sPerrorException(const string& reason) override;

  void setLog(bool state) override;

  unique_ptr<SSqlStatement> prepare(const string& query, int nparams) override;

  void execute(const string& query) override;

  void startTransaction() override;

  void rollback() override;

  void commit() override;

  bool isConnectionUsable() override;

  void reconnect() override;

private:

  PGconn* d_db;

  string d_connectstr;

  string d_connectlogstr;

  static bool s_dolog;

  bool d_in_trx;

  bool d_use_prepared;

  unsigned int d_nstatements;

};