diff options
Diffstat (limited to 'libpqpp/pq-bulkselectcommand.cpp')
-rw-r--r-- | libpqpp/pq-bulkselectcommand.cpp | 53 |
1 files changed, 53 insertions, 0 deletions
diff --git a/libpqpp/pq-bulkselectcommand.cpp b/libpqpp/pq-bulkselectcommand.cpp new file mode 100644 index 0000000..27674f5 --- /dev/null +++ b/libpqpp/pq-bulkselectcommand.cpp @@ -0,0 +1,53 @@ +#include "pq-bulkselectcommand.h" +#include "pq-connection.h" +#include "pq-column.h" +#include "pq-error.h" + +PQ::BulkSelectCommand::BulkSelectCommand(Connection * conn, const std::string & sql, unsigned int no) : + DB::Command(sql), + DB::SelectCommand(sql), + PQ::Command(conn, sql, no), + executed(false) +{ + prepareSql(preparedSql, sql); +} + +PQ::BulkSelectCommand::~BulkSelectCommand() +{ + if (execRes) { + PQclear(execRes); + } +} + +void +PQ::BulkSelectCommand::execute() +{ + if (!executed) { + execRes = c->checkResult( + PQexecParams(c->conn, preparedSql.c_str(), values.size(), NULL, &values.front(), &lengths.front(), NULL, 0), + PGRES_TUPLES_OK); + nTuples = PQntuples(execRes); + tuple = -1; + unsigned int nFields = PQnfields(execRes); + for (unsigned int f = 0; f < nFields; f += 1) { + insertColumn(DB::ColumnPtr(new Column(this, f))); + } + executed = true; + } +} + +bool +PQ::BulkSelectCommand::fetch() +{ + execute(); + if (++tuple < nTuples) { + return true; + } + else { + PQclear(execRes); + execRes = NULL; + executed = false; + return false; + } +} + |