From 172567f6ee168846b41d11ed7b8bd9c1ccbb659b Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Sun, 22 Aug 2021 13:58:49 +0100
Subject: Wait on specific events being processed

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

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

diff --git a/lib/output/pq/updateDatabase.cpp b/lib/output/pq/updateDatabase.cpp
index 485c914..330a0af 100644
--- a/lib/output/pq/updateDatabase.cpp
+++ b/lib/output/pq/updateDatabase.cpp
@@ -183,6 +183,12 @@ namespace MyGrate::Output::Pq {
 		}
 	}
 
+	const EventCounter &
+	UpdateDatabase::getProcessedCounts() const
+	{
+		return processed;
+	}
+
 	void
 	UpdateDatabase::tableMap(MariaDB_Event_Ptr e)
 	{
@@ -207,6 +213,7 @@ namespace MyGrate::Output::Pq {
 	void
 	UpdateDatabase::afterEvent(const MariaDB_Event_Ptr & e)
 	{
+		processed.tick(e->event_type);
 		if (!intx) {
 			output::pq::sql::updateSourcePosition::execute(this, e->next_event_pos, source);
 			commitTx();
-- 
cgit v1.2.3