1 files changed, 8 insertions, 4 deletions
diff --git a/lib/output/pq/updateDatabase.h b/lib/output/pq/updateDatabase.h
index bc54282..4e01d9c 100644
--- a/lib/output/pq/updateDatabase.h
+++ b/lib/output/pq/updateDatabase.h
@@ -2,15 +2,19 @@
 #define MYGRATE_OUTPUT_PQ_UPDATEDATABASE_H
 
 #include "pqConn.h"
-#include <string>
+#include <cstdint>
+#include <eventHandlerBase.h>
+#include <eventSourceBase.h>
 
 namespace MyGrate::Output::Pq {
-	class UpdateDatabase : PqConn {
+	class UpdateDatabase : public PqConn, public EventHandlerBase {
 	public:
-		UpdateDatabase(const char * const str, std::string prefix);
+		UpdateDatabase(const char * const str, uint64_t source);
+
+		EventSourceBasePtr getSource();
 
 	private:
-		std::string prefix;
+		uint64_t source;
 	};
 }