Mercurial > malikania

annotate database/postgresql/src/driver.cpp @ 29:99792c6c8b06

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Server: add initial postgresql account management, #475
author David Demelier <markand@malikania.fr>
date Thu, 26 May 2016 07:32:05 +0200
parents
children a1e80d991968
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
29
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
1 /*
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
2 * driver.cpp -- PostgreSQL database driver
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
3 *
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
4 * Copyright (c) 2013-2016 David Demelier <markand@malikania.fr>
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
5 *
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
6 * Permission to use, copy, modify, and/or distribute this software for any
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
7 * purpose with or without fee is hereby granted, provided that the above
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
8 * copyright notice and this permission notice appear in all copies.
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
9 *
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
10 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
11 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
12 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
13 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
14 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
15 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
16 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
17 */
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
18
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
19 #include <sstream>
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
20 #include <string>
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
21 #include <unordered_map>
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
22
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
23 #include <malikania/dynlib.h>
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
24
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
25 #include "driver.h"
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
26
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
27 namespace pgsql {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
28
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
29 PGconn *connection{nullptr};
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
30
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
31 std::shared_ptr<PGresult> exec(const std::string &sql)
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
32 {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
33 std::shared_ptr<PGresult> ptr(PQexec(connection, sql.c_str()), PQclear);
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
34
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
35 if (PQresultStatus(ptr.get()) != PGRES_TUPLES_OK && PQresultStatus(ptr.get()) != PGRES_COMMAND_OK)
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
36 throw std::runtime_error(PQresultErrorMessage(ptr.get()));
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
37
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
38 return ptr;
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
39 }
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
40
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
41 std::string escape(const std::string &input)
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
42 {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
43 auto text = PQescapeLiteral(connection, input.c_str(), input.length());
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
44
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
45 if (!text)
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
46 throw std::runtime_error(PQerrorMessage(connection));
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
47
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
48 std::string result(text);
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
49
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
50 PQfreemem(text);
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
51
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
52 return result;
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
53 }
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
54
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
55 } // !pgsql
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
56
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
57 namespace {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
58
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
59 std::string parameters(const std::unordered_map<std::string, std::string> &params)
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
60 {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
61 std::ostringstream oss;
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
62 std::unordered_map<std::string, std::string>::const_iterator it;
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
63
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
64 // Host.
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
65 if ((it = params.find("host")) == params.end())
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
66 throw std::runtime_error("missing 'host' parameter");
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
67
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
68 oss << "host = " << it->second << " ";
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
69
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
70 // Database.
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
71 if ((it = params.find("database")) == params.end())
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
72 throw std::runtime_error("missing 'database' parameter");
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
73
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
74 oss << "dbname = " << it->second << " ";
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
75
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
76 // User.
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
77 if ((it = params.find("user")) == params.end())
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
78 throw std::runtime_error("missing 'user' parameter");
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
79
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
80 oss << "user = " << it->second << " ";
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
81
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
82 // Port (optional).
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
83 if ((it = params.find("port")) != params.end())
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
84 oss << "port = " << it->second << " ";
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
85
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
86 // Password (optional).
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
87 if ((it = params.find("password")) != params.end())
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
88 oss << "password = " << it->second << " ";
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
89
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
90 return oss.str();
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
91 }
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
92
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
93 } // !namespace
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
94
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
95 extern "C" {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
96
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
97 DYNLIB_EXPORT void malikania_driver_load(const std::unordered_map<std::string, std::string> &params)
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
98 {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
99 pgsql::connection = PQconnectdb(parameters(params).c_str());
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
100
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
101 if (!pgsql::connection)
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
102 throw std::runtime_error(PQerrorMessage(pgsql::connection));
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
103 }
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
104
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
105 DYNLIB_EXPORT void malikania_driver_unload()
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
106 {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
107 if (pgsql::connection) {
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
108 PQfinish(pgsql::connection);
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
109 pgsql::connection = nullptr;
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
110 }
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
111 }
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
112
99792c6c8b06 Server: add initial postgresql account management, #475
David Demelier <markand@malikania.fr>
parents:
diff changeset
113 } // !C