Create indexes and keys after copying table data

2 files changed, 24 insertions, 1 deletions

diff --git a/lib/output/pq/updateDatabase.cpp b/lib/output/pq/updateDatabase.cpp
index ea8797b..46641fc 100644
--- a/lib/output/pq/updateDatabase.cpp
+++ b/lib/output/pq/updateDatabase.cpp
@@ -10,6 +10,7 @@
 #include <input/mysqlRecordSet.h>
 #include <input/replStream.h>
 #include <input/sql/selectColumns.h>
+#include <input/sql/selectIndexes.h>
 #include <input/sql/showMasterStatus.h>
 #include <memory>
 #include <mysql_types.h>
@@ -114,7 +115,6 @@ namespace MyGrate::Output::Pq {
 						ct << " not null";
 					}
 					if (col[3]) {
-						ct << " primary key";
 						tableDef->keys += 1;
 					}
 					tableDef->columns.push_back(
@@ -123,6 +123,7 @@ namespace MyGrate::Output::Pq {
 				ct << ")";
 				this->query(ct.str().c_str());
 				this->copyTableContent(conn, tableName, tableDef);
+				this->copyIndexes(conn, tableName);
 			});
 			tables.emplace(tableName, std::move(tableDef));
 		});
@@ -253,6 +254,27 @@ namespace MyGrate::Output::Pq {
 	}
 
 	void
+	UpdateDatabase::copyIndexes(Input::MySQLConn * conn, const char * tableName)
+	{
+		auto idxs = input::sql::selectIndexes::execute(conn, tableName);
+		for (const auto idx : *idxs) {
+			const auto [name, columns, nonunique]
+					= *idx.create<std::tuple<std::string_view, std::string_view, bool>, 3, 1>();
+			if (nonunique) {
+				query(scprintf<"CREATE INDEX %? ON %?.%?(%?)">(name, schema, tableName, columns).c_str());
+			}
+			else if (name == "PRIMARY") {
+				query(scprintf<"ALTER TABLE %?.%? ADD CONSTRAINT pk_%? PRIMARY KEY(%?)">(
+						schema, tableName, name, columns)
+								.c_str());
+			}
+			else {
+				query(scprintf<"CREATE UNIQUE INDEX %? ON %?.%?(%?)">(name, schema, tableName, columns).c_str());
+			}
+		}
+	}
+
+	void
 	UpdateDatabase::tableMap(MariaDB_Event_Ptr e)
 	{
 		if (*e->event.table_map.database == database) {
diff --git a/lib/output/pq/updateDatabase.h b/lib/output/pq/updateDatabase.h
index 716d7f6..63a96b9 100644
--- a/lib/output/pq/updateDatabase.h
+++ b/lib/output/pq/updateDatabase.h
@@ -78,6 +78,7 @@ namespace MyGrate::Output::Pq {
 		static void copyKeys(const Row & r, const TableDefPtr &, std::back_insert_iterator<Row> &&);
 
 		void copyTableContent(Input::MySQLConn *, const char * tableName, const TableDefPtr &);
+		void copyIndexes(Input::MySQLConn *, const char * tableName);
 
 		using Tables = std::map<std::string, TableDefPtr, std::less<>>;
 		Tables tables;