From d6c40826ba1762ece7fb674c330a5950935c39f3 Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Sat, 19 Jun 2021 21:22:33 +0100
Subject: Get the target schema name

---
 lib/output/pq/updateDatabase.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'lib/output/pq/updateDatabase.cpp')

diff --git a/lib/output/pq/updateDatabase.cpp b/lib/output/pq/updateDatabase.cpp
index f344f42..52388be 100644
--- a/lib/output/pq/updateDatabase.cpp
+++ b/lib/output/pq/updateDatabase.cpp
@@ -11,11 +11,13 @@
 #include <output/pq/sql/insertSource.h>
 #include <output/pq/sql/selectColumns.h>
 #include <output/pq/sql/selectSource.h>
+#include <output/pq/sql/selectSourceSchema.h>
 #include <output/pq/sql/selectTables.h>
 #include <stdexcept>
 
 namespace MyGrate::Output::Pq {
-	UpdateDatabase::UpdateDatabase(const char * const str, uint64_t s) : PqConn {str}, source {s}
+	UpdateDatabase::UpdateDatabase(const char * const str, uint64_t s) :
+		PqConn {str}, source {s}, schema(**output::pq::sql::selectSourceSchema::execute(this, s))
 	{
 		auto trecs = output::pq::sql::selectTables::execute(this, source);
 		auto crecs = output::pq::sql::selectColumns::execute(this, source);
-- 
cgit v1.2.3