diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/dbTypes.h | 74 | ||||
-rw-r--r-- | lib/input/mysqlConn.cpp | 1 | ||||
-rw-r--r-- | lib/input/mysqlConn.h | 4 | ||||
-rw-r--r-- | lib/input/mysqlStmt.h | 4 | ||||
-rw-r--r-- | lib/output/dumpToConsole.cpp | 1 | ||||
-rw-r--r-- | lib/output/pq/pqConn.cpp | 1 | ||||
-rw-r--r-- | lib/output/pq/pqConn.h | 4 | ||||
-rw-r--r-- | lib/output/pq/pqStmt.h | 4 |
8 files changed, 86 insertions, 7 deletions
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; |