From 6d012073a48ca7044eea6835bed02220b28fc48c Mon Sep 17 00:00:00 2001 From: Boris Kolpackov Date: Tue, 10 Jul 2012 15:17:14 +0200 Subject: Add support for custom database type mapping New pragma qualifier, map, and specifiers: as, to, from. New tests: /custom. --- odb/pgsql/details/conversion.hxx | 59 ++++++++++++++++++++++++++++++++++++++ odb/pgsql/query.cxx | 19 ++++++++++++- odb/pgsql/query.hxx | 61 +++++++++++++++++++++++++--------------- odb/pgsql/query.ixx | 10 ++++--- odb/pgsql/query.txx | 32 ++++++++++----------- 5 files changed, 137 insertions(+), 44 deletions(-) create mode 100644 odb/pgsql/details/conversion.hxx (limited to 'odb/pgsql') diff --git a/odb/pgsql/details/conversion.hxx b/odb/pgsql/details/conversion.hxx new file mode 100644 index 0000000..230be2a --- /dev/null +++ b/odb/pgsql/details/conversion.hxx @@ -0,0 +1,59 @@ +// file : odb/pgsql/details/conversion.hxx +// copyright : Copyright (c) 2009-2012 Code Synthesis Tools CC +// license : GNU GPL v2; see accompanying LICENSE file + +#ifndef ODB_PGSQL_DETAILS_CONVERSION_HXX +#define ODB_PGSQL_DETAILS_CONVERSION_HXX + +#include + +#include + +namespace odb +{ + // @@ Revise this. + // + namespace details {} + + namespace pgsql + { + namespace details + { + using namespace odb::details; + + // Detect whether conversion is specified in type_traits. + // + template + meta::yes + conversion_p_test (typename type_traits::conversion*); + + template + meta::no + conversion_p_test (...); + + template + struct conversion_p + { + static const bool value = + sizeof (conversion_p_test (0)) == sizeof (meta::yes); + }; + + template ::value> + struct conversion; + + template + struct conversion + { + static const char* to () {return type_traits::conversion::to ();} + }; + + template + struct conversion + { + static const char* to () {return 0;} + }; + } + } +} + +#endif // ODB_PGSQL_DETAILS_CONVERSION_HXX diff --git a/odb/pgsql/query.cxx b/odb/pgsql/query.cxx index e3e4d24..9d50c3b 100644 --- a/odb/pgsql/query.cxx +++ b/odb/pgsql/query.cxx @@ -187,10 +187,13 @@ namespace odb } void query:: - add (details::shared_ptr p) + add (details::shared_ptr p, const char* conv) { clause_.push_back (clause_part (clause_part::param)); + if (conv != 0) + clause_.back ().part = conv; + parameters_.push_back (p); bind_.push_back (bind ()); binding_.bind = &bind_[0]; @@ -354,8 +357,22 @@ namespace odb ostringstream os; os << param++; + + // Add the conversion expression, if any. + // + string::size_type p; + if (!i->part.empty ()) + { + p = i->part.find ("(?)"); + r.append (i->part, 0, p); + } + r += '$'; r += os.str (); + + if (!i->part.empty ()) + r.append (i->part, p + 3, string::npos); + break; } case clause_part::native: diff --git a/odb/pgsql/query.hxx b/odb/pgsql/query.hxx index f00d232..0b5ff55 100644 --- a/odb/pgsql/query.hxx +++ b/odb/pgsql/query.hxx @@ -23,6 +23,7 @@ #include #include +#include namespace odb { @@ -103,7 +104,7 @@ namespace odb clause_part (bool p): kind (boolean), bool_part (p) {} kind_type kind; - std::string part; + std::string part; // If kind is param, then part is conversion expr. bool bool_part; }; @@ -146,7 +147,8 @@ namespace odb query (val_bind v) : binding_ (0, 0), native_binding_ (0, 0, 0, 0) { - append::db_type_id> (v); + append::db_type_id> ( + v, details::conversion::to ()); } template @@ -154,7 +156,8 @@ namespace odb query (ref_bind r) : binding_ (0, 0), native_binding_ (0, 0, 0, 0) { - append::db_type_id> (r); + append::db_type_id> ( + r, details::conversion::to ()); } template @@ -237,7 +240,8 @@ namespace odb query& operator+= (val_bind v) { - append::db_type_id> (v); + append::db_type_id> ( + v, details::conversion::to ()); return *this; } @@ -245,18 +249,19 @@ namespace odb query& operator+= (ref_bind r) { - append::db_type_id> (r); + append::db_type_id> ( + r, details::conversion::to ()); return *this; } public: template void - append (val_bind); + append (val_bind, const char* conv); template void - append (ref_bind); + append (ref_bind, const char* conv); void append (const std::string& native); @@ -266,7 +271,7 @@ namespace odb private: void - add (details::shared_ptr); + add (details::shared_ptr, const char* conv); private: typedef std::vector clause_type; @@ -414,10 +419,11 @@ namespace odb template struct query_column { - // Note that we keep shalow copies of the table and column names. + // Note that we keep shallow copies of the table, column, and conversion + // expression. The latter can be NULL. // - query_column (const char* table, const char* column) - : table_ (table), column_ (column) + query_column (const char* table, const char* column, const char* conv) + : table_ (table), column_ (column), conversion_ (conv) { } @@ -433,6 +439,14 @@ namespace odb return column_; } + // Can be NULL. + // + const char* + conversion () const + { + return conversion_; + } + // is_null, is_not_null // public: @@ -485,7 +499,7 @@ namespace odb { query q (table_, column_); q += "="; - q.append (v); + q.append (v, conversion_); return q; } @@ -502,7 +516,7 @@ namespace odb { query q (table_, column_); q += "="; - q.append (r); + q.append (r, conversion_); return q; } @@ -570,7 +584,7 @@ namespace odb { query q (table_, column_); q += "!="; - q.append (v); + q.append (v, conversion_); return q; } @@ -587,7 +601,7 @@ namespace odb { query q (table_, column_); q += "!="; - q.append (r); + q.append (r, conversion_); return q; } @@ -655,7 +669,7 @@ namespace odb { query q (table_, column_); q += "<"; - q.append (v); + q.append (v, conversion_); return q; } @@ -672,7 +686,7 @@ namespace odb { query q (table_, column_); q += "<"; - q.append (r); + q.append (r, conversion_); return q; } @@ -740,7 +754,7 @@ namespace odb { query q (table_, column_); q += ">"; - q.append (v); + q.append (v, conversion_); return q; } @@ -757,7 +771,7 @@ namespace odb { query q (table_, column_); q += ">"; - q.append (r); + q.append (r, conversion_); return q; } @@ -825,7 +839,7 @@ namespace odb { query q (table_, column_); q += "<="; - q.append (v); + q.append (v, conversion_); return q; } @@ -842,7 +856,7 @@ namespace odb { query q (table_, column_); q += "<="; - q.append (r); + q.append (r, conversion_); return q; } @@ -910,7 +924,7 @@ namespace odb { query q (table_, column_); q += ">="; - q.append (v); + q.append (v, conversion_); return q; } @@ -927,7 +941,7 @@ namespace odb { query q (table_, column_); q += ">="; - q.append (r); + q.append (r, conversion_); return q; } @@ -1071,6 +1085,7 @@ namespace odb private: const char* table_; const char* column_; + const char* conversion_; }; // Provide operator+() for using columns to construct native diff --git a/odb/pgsql/query.ixx b/odb/pgsql/query.ixx index 9d28032..d850c1a 100644 --- a/odb/pgsql/query.ixx +++ b/odb/pgsql/query.ixx @@ -8,20 +8,22 @@ namespace odb { template inline void query:: - append (val_bind v) + append (val_bind v, const char* conv) { add ( details::shared_ptr ( - new (details::shared) query_param_impl (v))); + new (details::shared) query_param_impl (v)), + conv); } template inline void query:: - append (ref_bind r) + append (ref_bind r, const char* conv) { add ( details::shared_ptr ( - new (details::shared) query_param_impl (r))); + new (details::shared) query_param_impl (r)), + conv); } } } diff --git a/odb/pgsql/query.txx b/odb/pgsql/query.txx index d4fab74..274e703 100644 --- a/odb/pgsql/query.txx +++ b/odb/pgsql/query.txx @@ -19,7 +19,7 @@ namespace odb // append (c.table (), c.column ()); append ("="); - append (val_bind (true)); + append (val_bind (true), c.conversion ()); } // query_column @@ -30,9 +30,9 @@ namespace odb { query q (table_, column_); q += "IN ("; - q.append (val_bind (v1)); + q.append (val_bind (v1), conversion_); q += ","; - q.append (val_bind (v2)); + q.append (val_bind (v2), conversion_); q += ")"; return q; } @@ -43,11 +43,11 @@ namespace odb { query q (table_, column_); q += "IN ("; - q.append (val_bind (v1)); + q.append (val_bind (v1), conversion_); q += ","; - q.append (val_bind (v2)); + q.append (val_bind (v2), conversion_); q += ","; - q.append (val_bind (v3)); + q.append (val_bind (v3), conversion_); q += ")"; return q; } @@ -58,13 +58,13 @@ namespace odb { query q (table_, column_); q += "IN ("; - q.append (val_bind (v1)); + q.append (val_bind (v1), conversion_); q += ","; - q.append (val_bind (v2)); + q.append (val_bind (v2), conversion_); q += ","; - q.append (val_bind (v3)); + q.append (val_bind (v3), conversion_); q += ","; - q.append (val_bind (v4)); + q.append (val_bind (v4), conversion_); q += ")"; return q; } @@ -75,15 +75,15 @@ namespace odb { query q (table_, column_); q += "IN ("; - q.append (val_bind (v1)); + q.append (val_bind (v1), conversion_); q += ","; - q.append (val_bind (v2)); + q.append (val_bind (v2), conversion_); q += ","; - q.append (val_bind (v3)); + q.append (val_bind (v3), conversion_); q += ","; - q.append (val_bind (v4)); + q.append (val_bind (v4), conversion_); q += ","; - q.append (val_bind (v5)); + q.append (val_bind (v5), conversion_); q += ")"; return q; } @@ -101,7 +101,7 @@ namespace odb if (i != begin) q += ","; - q.append (val_bind (*i)); + q.append (val_bind (*i), conversion_); } q += ")"; return q; -- cgit v1.1