PostgreSQL files prefixed with pq-

1 files changed, 156 insertions, 0 deletions

diff --git a/libpqpp/pq-command.cpp b/libpqpp/pq-command.cpp
new file mode 100644
index 0000000..aeb4914
--- /dev/null
+++ b/libpqpp/pq-command.cpp
@@ -0,0 +1,156 @@
+#include "pq-command.h"
+#include "pq-connection.h"
+#include <stdlib.h>
+#include <string.h>
+#include <boost/date_time/posix_time/posix_time.hpp>
+
+static std::string addrStr(void * p, unsigned int no) {
+	std::string r(50, ' ');
+	r.resize(snprintf(const_cast<char *>(r.c_str()), r.length(), "pStatement_%u_%p", no, p));
+	return r;
+}
+
+PQ::Command::Command(const Connection * conn, const std::string & sql, unsigned int no) :
+	DB::Command(sql),
+	stmntName(addrStr(this, no)),
+	c(conn)
+{
+}
+
+PQ::Command::~Command()
+{
+	for (std::vector<char *>::const_iterator i = values.begin(); i != values.end(); ++i) {
+		free(*i);
+	}
+}
+
+void
+PQ::Command::prepareSql(std::string & psql, const std::string & sql)
+{
+	char buf[4];
+	int p = 1;
+	bool inquote = false;
+	for(std::string::const_iterator i = sql.begin(); i != sql.end(); ++i) {
+		if (*i == '?' && !inquote) {
+			snprintf(buf, 4, "$%d", p++);
+			psql += buf;
+		}
+		else if (*i == '\'') {
+			inquote = !inquote;
+			psql += *i;
+		}
+		else {
+			psql += *i;
+		}
+	}
+}
+
+void
+PQ::Command::paramsAtLeast(unsigned int n)
+{
+	if (values.size() <= n) {
+		values.resize(n + 1, NULL);
+		lengths.resize(n + 1, 0);
+		formats.resize(n + 1, 0);
+	}
+	else {
+		free(values[n]);
+		values[n] = NULL;
+	}
+}
+
+void
+PQ::Command::bindParamI(unsigned int n, int v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%d", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamI(unsigned int n, long int v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%ld", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamI(unsigned int n, long long int v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%lld", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamI(unsigned int n, unsigned int v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%u", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamI(unsigned int n, long unsigned int v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%lu", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamI(unsigned int n, long long unsigned int v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%llu", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamB(unsigned int n, bool v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%s", v ? "true" : "false");
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamF(unsigned int n, double v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%g", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamF(unsigned int n, float v)
+{
+	paramsAtLeast(n);
+	lengths[n] = asprintf(&values[n], "%g", v);
+	formats[n] = 0;
+}
+void
+PQ::Command::bindParamS(unsigned int n, const Glib::ustring & s)
+{
+	paramsAtLeast(n);
+	values[n] = strndup(s.c_str(), s.bytes());
+	formats[n] = 0;
+	lengths[n] = s.bytes();
+}
+void
+PQ::Command::bindParamT(unsigned int n, const boost::posix_time::time_duration & v)
+{
+	paramsAtLeast(n);
+	auto buf = boost::posix_time::to_simple_string(v);
+	values[n] = strdup(buf.c_str());
+	formats[n] = 0;
+	lengths[n] = buf.length();
+}
+void
+PQ::Command::bindParamT(unsigned int n, const boost::posix_time::ptime & v)
+{
+	paramsAtLeast(n);
+	auto buf = boost::posix_time::to_iso_extended_string(v);
+	values[n] = strdup(buf.c_str());
+	formats[n] = 0;
+	lengths[n] = buf.length();
+}
+void
+PQ::Command::bindNull(unsigned int n)
+{
+	paramsAtLeast(n);
+}
+