From 6e726340e23a97480b17065a633c36a9ba53cf14 Mon Sep 17 00:00:00 2001 From: Dan Goodliffe Date: Thu, 3 Jun 2021 00:17:28 +0100 Subject: Determine parameter mode at point of call by DB type property It's still constexpr, so that's all good. --- lib/dbConn.h | 2 ++ lib/dbStmt.h | 17 ++++++++--------- lib/input/mysqlConn.h | 2 ++ lib/output/pq/pqConn.h | 2 ++ test/test-mysql.cpp | 7 ++++--- test/test-postgresql.cpp | 18 +++++++++--------- 6 files changed, 27 insertions(+), 21 deletions(-) diff --git a/lib/dbConn.h b/lib/dbConn.h index e4b056c..3f14dcb 100644 --- a/lib/dbConn.h +++ b/lib/dbConn.h @@ -15,6 +15,8 @@ namespace MyGrate { }; using DbPrepStmtPtr = std::unique_ptr; + enum class ParamMode { None, DollarNum, QMark }; + class DbConn { public: virtual ~DbConn() = default; diff --git a/lib/dbStmt.h b/lib/dbStmt.h index 81621de..80dbd71 100644 --- a/lib/dbStmt.h +++ b/lib/dbStmt.h @@ -9,17 +9,16 @@ #include namespace MyGrate { - class DbConn; - enum class ParamMode { None, DollarNum, QMark }; - - template class DbStmt { + template class DbStmt { public: // This don't account for common table expressions, hopefully won't need those :) static constexpr auto isSelect {S.v().starts_with("SELECT") || S.v().starts_with("SHOW") || S.v().find("RETURNING") != std::string_view::npos}; // These don't account for string literals, which we'd prefer to avoid anyway :) - static constexpr auto paramCount {[]() -> std::size_t { + static constexpr std::size_t + paramCount(ParamMode pm) + { switch (pm) { case ParamMode::None: return 0LU; @@ -42,15 +41,15 @@ namespace MyGrate { return c == '?'; }); } - }()}; + } using Return = std::conditional_t; - template + template static Return - execute(DbConn * c, P &&... p) + execute(ConnType * c, P &&... p) { - static_assert(sizeof...(P) == paramCount); + static_assert(sizeof...(P) == paramCount(ConnType::paramMode)); auto stmt {c->prepare(S, sizeof...(P))}; stmt->execute({std::forward(p)...}); if constexpr (isSelect) { diff --git a/lib/input/mysqlConn.h b/lib/input/mysqlConn.h index 2f71262..29a34db 100644 --- a/lib/input/mysqlConn.h +++ b/lib/input/mysqlConn.h @@ -10,6 +10,8 @@ namespace MyGrate::Input { class MySQLConn : public MYSQL, public DbConn { public: + static constexpr auto paramMode {ParamMode::QMark}; + MySQLConn(const char * const host, const char * const user, const char * const pass, unsigned short port); ~MySQLConn(); diff --git a/lib/output/pq/pqConn.h b/lib/output/pq/pqConn.h index 856683d..ae5fc9a 100644 --- a/lib/output/pq/pqConn.h +++ b/lib/output/pq/pqConn.h @@ -14,6 +14,8 @@ namespace MyGrate::Output::Pq { class PqConn : public DbConn { public: + static constexpr auto paramMode {ParamMode::DollarNum}; + explicit PqConn(const char * const str); virtual ~PqConn() = default; diff --git a/test/test-mysql.cpp b/test/test-mysql.cpp index f5281a9..493ec02 100644 --- a/test/test-mysql.cpp +++ b/test/test-mysql.cpp @@ -4,6 +4,7 @@ #include #include #include +#include #include #include #include @@ -42,13 +43,13 @@ BOOST_AUTO_TEST_CASE(simple) using SomeSelect = MyGrate::DbStmt<"SELECT * FROM foo">; using SomeShow = MyGrate::DbStmt<"SHOW MASTER STATUS">; -using SomeUpdate = MyGrate::DbStmt<"UPDATE foo SET blah = ? WHERE bar = ?", MyGrate::ParamMode::QMark>; +using SomeUpdate = MyGrate::DbStmt<"UPDATE foo SET blah = ? WHERE bar = ?">; static_assert(std::is_same_v); static_assert(std::is_same_v); static_assert(std::is_same_v); -static_assert(SomeShow::paramCount == 0); -static_assert(SomeUpdate::paramCount == 2); +static_assert(SomeShow::paramCount(MyGrate::ParamMode::QMark) == 0); +static_assert(SomeUpdate::paramCount(MyGrate::ParamMode::QMark) == 2); BOOST_AUTO_TEST_CASE(stmt) { diff --git a/test/test-postgresql.cpp b/test/test-postgresql.cpp index a24acb8..3a12f43 100644 --- a/test/test-postgresql.cpp +++ b/test/test-postgresql.cpp @@ -2,6 +2,7 @@ #include #include +#include #include #include #include @@ -38,18 +39,17 @@ BOOST_AUTO_TEST_CASE(simple) c.query("DROP TABLE test"); } -using SomeUpdate = MyGrate::DbStmt<"UPDATE foo SET blah = $2 WHERE bar = $1", MyGrate::ParamMode::DollarNum>; -using SomeUpdateRtn - = MyGrate::DbStmt<"UPDATE foo SET blah = $2 WHERE bar = $1 RETURNING x", MyGrate::ParamMode::DollarNum>; -using SomeShow = MyGrate::DbStmt<"SHOW all", MyGrate::ParamMode::DollarNum>; -using SomeIns = MyGrate::DbStmt<"INSERT INTO foo VALUES(..., $87)", MyGrate::ParamMode::DollarNum>; +using SomeUpdate = MyGrate::DbStmt<"UPDATE foo SET blah = $2 WHERE bar = $1">; +using SomeUpdateRtn = MyGrate::DbStmt<"UPDATE foo SET blah = $2 WHERE bar = $1 RETURNING x">; +using SomeShow = MyGrate::DbStmt<"SHOW all">; +using SomeIns = MyGrate::DbStmt<"INSERT INTO foo VALUES(..., $87)">; -static_assert(SomeUpdate::paramCount == 2); +static_assert(SomeUpdate::paramCount(MyGrate::ParamMode::DollarNum) == 2); static_assert(std::is_same_v); -static_assert(SomeUpdateRtn::paramCount == 2); +static_assert(SomeUpdateRtn::paramCount(MyGrate::ParamMode::DollarNum) == 2); static_assert(std::is_same_v); -static_assert(SomeShow::paramCount == 0); -static_assert(SomeIns::paramCount == 87); +static_assert(SomeShow::paramCount(MyGrate::ParamMode::DollarNum) == 0); +static_assert(SomeIns::paramCount(MyGrate::ParamMode::DollarNum) == 87); static_assert(std::is_same_v); BOOST_AUTO_TEST_CASE(stmt) -- cgit v1.2.3