From c5974cfe2726d8088b08dc52edd4a825dc86e147 Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Sat, 5 Jun 2021 17:31:57 +0100
Subject: Add conversion operators to get common types from DbValues

---
 lib/dbTypes.h                | 74 ++++++++++++++++++++++++++++++++++++++++++++
 lib/input/mysqlConn.cpp      |  1 -
 lib/input/mysqlConn.h        |  4 ++-
 lib/input/mysqlStmt.h        |  4 ++-
 lib/output/dumpToConsole.cpp |  1 -
 lib/output/pq/pqConn.cpp     |  1 -
 lib/output/pq/pqConn.h       |  4 ++-
 lib/output/pq/pqStmt.h       |  4 ++-
 8 files changed, 86 insertions(+), 7 deletions(-)

(limited to 'lib')

diff --git a/lib/dbTypes.h b/lib/dbTypes.h
index 068d642..332b896 100644
--- a/lib/dbTypes.h
+++ b/lib/dbTypes.h
@@ -2,6 +2,7 @@
 #define MYGRATE_DBTYPES_H
 
 #include "bitset.h"
+#include <boost/numeric/conversion/cast.hpp>
 #include <cstdint>
 #include <span>
 #include <string_view>
@@ -36,6 +37,60 @@ namespace MyGrate {
 	using DbValueV = std::variant<std::nullptr_t, double, float, int8_t, uint8_t, int16_t, uint16_t, int32_t, uint32_t,
 			int64_t, uint64_t, timespec, Date, Time, DateTime, std::string_view, BitSet, Blob>;
 
+	namespace detail {
+		template<typename I>
+		concept HasToString = requires
+		{
+			std::to_string(I {});
+		};
+
+		template<typename T> struct is_false {
+			static constexpr bool value {false};
+		};
+
+		template<typename R, template<typename> typename ConceptT> struct SafeExtract {
+			R
+			operator()(const R & i) const
+			{
+				return i;
+			}
+
+			template<typename I>
+			R
+			operator()(const I & i) const
+			{
+				if constexpr (ConceptT<I>::value) {
+					return boost::numeric_cast<R>(i);
+				}
+				else {
+					throw std::logic_error("Unreasonable conversion requested");
+				}
+			}
+		};
+
+		struct ToString {
+			std::string
+			operator()(const std::string_view & i) const
+			{
+				return std::string {i};
+			}
+
+			template<HasToString I>
+			std::string
+			operator()(const I & i) const
+			{
+				return std::to_string(i);
+			}
+
+			template<typename I>
+			std::string
+			operator()(const I &) const
+			{
+				throw std::logic_error("Unreasonable to_string requested");
+			}
+		};
+	}
+
 	class DbValue : public DbValueV {
 	public:
 		using DbValueV::DbValueV;
@@ -54,6 +109,25 @@ namespace MyGrate {
 		{
 			return std::get<T>(static_cast<const DbValueV &>(*this));
 		}
+
+		template<typename R> operator R() const
+		{
+			if constexpr (std::is_integral_v<R>) {
+				return visit(detail::SafeExtract<R, std::is_integral> {});
+			}
+			else if constexpr (std::is_floating_point_v<R>) {
+				return visit(detail::SafeExtract<R, std::is_floating_point> {});
+			}
+			else if constexpr (std::is_same_v<std::string_view, R>) {
+				return get<std::string_view>();
+			}
+			else if constexpr (std::is_same_v<std::string, R>) {
+				return visit(detail::ToString {});
+			}
+			else {
+				static_assert(detail::is_false<R>::value, "Cannot extract one of these");
+			}
+		}
 	};
 }
 
diff --git a/lib/input/mysqlConn.cpp b/lib/input/mysqlConn.cpp
index 46ed7c6..4899deb 100644
--- a/lib/input/mysqlConn.cpp
+++ b/lib/input/mysqlConn.cpp
@@ -4,7 +4,6 @@
 #include <cstddef>
 #include <cstring>
 #include <dbConn.h>
-#include <dbTypes.h>
 #include <helpers.h>
 #include <memory>
 #include <mysql.h>
diff --git a/lib/input/mysqlConn.h b/lib/input/mysqlConn.h
index 29a34db..db30ff0 100644
--- a/lib/input/mysqlConn.h
+++ b/lib/input/mysqlConn.h
@@ -3,10 +3,12 @@
 
 #include <cstddef>
 #include <dbConn.h>
-#include <dbTypes.h>
 #include <initializer_list>
 #include <mysql.h>
 
+namespace MyGrate {
+	class DbValue;
+}
 namespace MyGrate::Input {
 	class MySQLConn : public MYSQL, public DbConn {
 	public:
diff --git a/lib/input/mysqlStmt.h b/lib/input/mysqlStmt.h
index a42e0db..69e62d4 100644
--- a/lib/input/mysqlStmt.h
+++ b/lib/input/mysqlStmt.h
@@ -3,12 +3,14 @@
 
 #include "dbConn.h"
 #include "dbRecordSet.h"
-#include "dbTypes.h"
 #include <cstddef>
 #include <initializer_list>
 #include <memory>
 #include <mysql.h>
 
+namespace MyGrate {
+	class DbValue;
+}
 namespace MyGrate::Input {
 	using StmtPtr = std::unique_ptr<MYSQL_STMT, decltype(&mysql_stmt_close)>;
 
diff --git a/lib/output/dumpToConsole.cpp b/lib/output/dumpToConsole.cpp
index 96850f3..949c16f 100644
--- a/lib/output/dumpToConsole.cpp
+++ b/lib/output/dumpToConsole.cpp
@@ -10,7 +10,6 @@
 #include <row.h>
 #include <streamSupport.h>
 #include <utility>
-#include <variant>
 
 namespace MyGrate::Output {
 	void
diff --git a/lib/output/pq/pqConn.cpp b/lib/output/pq/pqConn.cpp
index 81d9610..b3a6ca6 100644
--- a/lib/output/pq/pqConn.cpp
+++ b/lib/output/pq/pqConn.cpp
@@ -2,7 +2,6 @@
 #include "pqBindings.h"
 #include "pqStmt.h"
 #include <dbConn.h>
-#include <dbTypes.h>
 #include <helpers.h>
 #include <libpq-fe.h>
 #include <memory>
diff --git a/lib/output/pq/pqConn.h b/lib/output/pq/pqConn.h
index ae5fc9a..5c9effb 100644
--- a/lib/output/pq/pqConn.h
+++ b/lib/output/pq/pqConn.h
@@ -3,7 +3,6 @@
 
 #include <cstddef>
 #include <dbConn.h>
-#include <dbTypes.h>
 #include <functional>
 #include <initializer_list>
 #include <libpq-fe.h>
@@ -11,6 +10,9 @@
 #include <memory>
 #include <string>
 
+namespace MyGrate {
+	class DbValue;
+}
 namespace MyGrate::Output::Pq {
 	class PqConn : public DbConn {
 	public:
diff --git a/lib/output/pq/pqStmt.h b/lib/output/pq/pqStmt.h
index b806617..6531180 100644
--- a/lib/output/pq/pqStmt.h
+++ b/lib/output/pq/pqStmt.h
@@ -3,13 +3,15 @@
 
 #include "dbConn.h"
 #include "dbRecordSet.h"
-#include "dbTypes.h"
 #include <cstddef>
 #include <initializer_list>
 #include <libpq-fe.h>
 #include <memory>
 #include <string>
 
+namespace MyGrate {
+	class DbValue;
+}
 namespace MyGrate::Output::Pq {
 	class PqConn;
 
-- 
cgit v1.2.3