postgresql_engine.hpp 1.18 KB
Newer Older
louiz’'s avatar
louiz’ committed
1 2
#pragma once

3 4 5 6
#include <biboumi.h>
#include <string>
#include <stdexcept>
#include <memory>
louiz’'s avatar
louiz’ committed
7 8

#include <database/statement.hpp>
9
#include <database/engine.hpp>
louiz’'s avatar
louiz’ committed
10 11 12 13

#include <tuple>
#include <set>

14 15 16 17
#ifdef PQ_FOUND

#include <libpq-fe.h>

louiz’'s avatar
louiz’ committed
18 19 20 21 22 23 24 25 26
class PostgresqlEngine: public DatabaseEngine
{
 public:
  PostgresqlEngine(PGconn*const conn);

  ~PostgresqlEngine();

  static std::unique_ptr<DatabaseEngine> open(const std::string& string);

27
  std::set<std::string> get_all_columns_from_table(const std::string& table_name) override final;
louiz’'s avatar
louiz’ committed
28 29 30 31
  std::tuple<bool, std::string> raw_exec(const std::string& query) override final;
  std::unique_ptr<Statement> prepare(const std::string& query) override;
  void extract_last_insert_rowid(Statement& statement) override;
  std::string get_returning_id_sql_string(const std::string& col_name) override;
32
  std::string id_column_type() override;
louiz’'s avatar
louiz’ committed
33 34 35
private:
  PGconn* const conn;
};
36 37 38

#else

39 40
using namespace std::string_literals;

41 42 43 44 45 46
class PostgresqlEngine
{
public:
  static std::unique_ptr<DatabaseEngine> open(const std::string& string)
  {
    throw std::runtime_error("Cannot open postgresql database "s + string + ": biboumi is not compiled with libpq.");
47
    return {};
48 49 50 51
  }
};

#endif