connection.h
Go to the documentation of this file.
1 /*
2  * Copyright (C) 2005 Tommi Maekitalo
3  *
4  * This library is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU Lesser General Public
6  * License as published by the Free Software Foundation; either
7  * version 2.1 of the License, or (at your option) any later version.
8  *
9  * As a special exception, you may use this file as part of a free
10  * software library without restriction. Specifically, if other files
11  * instantiate templates or use macros or inline functions from this
12  * file, or you compile this file and link it with other files to
13  * produce an executable, this file does not by itself cause the
14  * resulting executable to be covered by the GNU General Public
15  * License. This exception does not however invalidate any other
16  * reasons why the executable file might be covered by the GNU Library
17  * General Public License.
18  *
19  * This library is distributed in the hope that it will be useful,
20  * but WITHOUT ANY WARRANTY; without even the implied warranty of
21  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
22  * Lesser General Public License for more details.
23  *
24  * You should have received a copy of the GNU Lesser General Public
25  * License along with this library; if not, write to the Free Software
26  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
27  */
28 
29 #ifndef TNTDB_POSTGRESQL_IMPL_CONNECTION_H
30 #define TNTDB_POSTGRESQL_IMPL_CONNECTION_H
31 
33 #include <tntdb/statement.h>
34 #include <libpq-fe.h>
35 #include <vector>
36 
37 namespace tntdb
38 {
57  namespace postgresql
58  {
61  {
62  PGconn* conn;
63  tntdb::Statement currvalStmt;
64  tntdb::Statement lastvalStmt;
65  unsigned transactionActive;
66  unsigned stmtCounter;
67  std::vector<std::string> stmtsToDeallocate;
68 
69  public:
70  explicit Connection(const char* conninfo);
71  ~Connection();
72 
73  void beginTransaction();
74  void commitTransaction();
75  void rollbackTransaction();
76 
77  size_type execute(const std::string& query);
78  tntdb::Result select(const std::string& query);
79  tntdb::Row selectRow(const std::string& query);
80  tntdb::Value selectValue(const std::string& query);
81  tntdb::Statement prepare(const std::string& query);
82  tntdb::Statement prepareWithLimit(const std::string& query, const std::string& limit, const std::string& offset);
83  bool ping();
84  long lastInsertId(const std::string& name);
85  void lockTable(const std::string& tablename, bool exclusive);
86 
87  PGconn* getPGConn() const { return conn; }
88  unsigned getNextStmtNumber() { return ++stmtCounter; }
89  void deallocateStatement(const std::string& stmtName);
90  void deallocateStatements();
91  };
92 
94  inline bool isError(const PGresult* res)
95  {
96  ExecStatusType status = PQresultStatus(res);
97  return status != PGRES_COMMAND_OK
98  && status != PGRES_TUPLES_OK
99  && status != PGRES_COPY_OUT
100  && status != PGRES_COPY_IN;
101  }
103 
104  }
105 }
106 
107 #endif // TNTDB_POSTGRESQL_IMPL_CONNECTION_H
108