From a6f3f335e931acd12b908ca0b4028a08c102627c Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Thu, 29 Apr 2021 19:38:05 +0100
Subject: Extend factory map to include make_shared

---
 lib/persistance.cpp | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

(limited to 'lib/persistance.cpp')

diff --git a/lib/persistance.cpp b/lib/persistance.cpp
index 0dd7cba..204e8f0 100644
--- a/lib/persistance.cpp
+++ b/lib/persistance.cpp
@@ -2,19 +2,28 @@
 #include <map>
 
 namespace Persistanace {
-	using NamedTypeFactories = std::map<std::string_view, std::function<std::unique_ptr<Persistable>()>>;
+	using Factories
+			= std::pair<std::function<std::unique_ptr<Persistable>()>, std::function<std::shared_ptr<Persistable>()>>;
+	using NamedTypeFactories = std::map<std::string_view, Factories>;
 	static NamedTypeFactories namedTypeFactories;
 
 	void
-	Persistable::addFactory(const std::string_view t, std::function<std::unique_ptr<Persistable>()> f)
+	Persistable::addFactory(const std::string_view t, std::function<std::unique_ptr<Persistable>()> fu,
+			std::function<std::shared_ptr<Persistable>()> fs)
 	{
-		namedTypeFactories.emplace(t, std::move(f));
+		namedTypeFactories.emplace(t, std::make_pair(std::move(fu), std::move(fs)));
 	}
 
 	std::unique_ptr<Persistable>
 	Persistable::callFactory(const std::string_view t)
 	{
-		return namedTypeFactories.at(t)();
+		return namedTypeFactories.at(t).first();
+	}
+
+	std::shared_ptr<Persistable>
+	Persistable::callSharedFactory(const std::string_view t)
+	{
+		return namedTypeFactories.at(t).second();
 	}
 
 	void
-- 
cgit v1.2.3