diff options
| -rw-r--r-- | odb/wrapper-traits.hxx | 101 |
1 files changed, 100 insertions, 1 deletions
diff --git a/odb/wrapper-traits.hxx b/odb/wrapper-traits.hxx index 35dec0e..8816f5a 100644 --- a/odb/wrapper-traits.hxx +++ b/odb/wrapper-traits.hxx @@ -7,10 +7,11 @@ #include <odb/pre.hxx> -#include <memory> // std::auto_ptr +#include <memory> // std::auto_ptr, std::unique_ptr, std::shared_ptr/weak_ptr #include <odb/nullable.hxx> +#include <odb/details/config.hxx> // ODB_CXX11 #include <odb/details/meta/remove-const.hxx> namespace odb @@ -122,6 +123,104 @@ namespace odb } }; +#ifdef ODB_CXX11 + + // Specialization for C++11 std::unique_ptr. + // + template <typename T, typename D> + class wrapper_traits<std::unique_ptr<T, D>> + { + public: + // T can be const. + // + typedef T wrapped_type; + typedef std::unique_ptr<T, D> wrapper_type; + + // T can be const. + // + typedef + typename odb::details::meta::remove_const<T>::result + unrestricted_wrapped_type; + + static const bool null_handler = true; + static const bool null_default = false; + + static bool + get_null (const wrapper_type& p) + { + return !p; + } + + static void + set_null (wrapper_type& p) + { + p.reset (); + } + + static const wrapped_type& + get_ref (const wrapper_type& p) + { + return *p; + } + + static unrestricted_wrapped_type& + set_ref (wrapper_type& p) + { + if (!p) + p.reset (new unrestricted_wrapped_type ()); + + return const_cast<unrestricted_wrapped_type&> (*p); + } + }; + + // Specialization for C++11 std::shared_ptr. + // + template <typename T> + class wrapper_traits<std::shared_ptr<T>> + { + public: + typedef T wrapped_type; + typedef std::shared_ptr<T> wrapper_type; + + // T can be const. + // + typedef + typename odb::details::meta::remove_const<T>::result + unrestricted_wrapped_type; + + static const bool null_handler = true; + static const bool null_default = false; + + static bool + get_null (const wrapper_type& p) + { + return !p; + } + + static void + set_null (wrapper_type& p) + { + p.reset (); + } + + static const wrapped_type& + get_ref (const wrapper_type& p) + { + return *p; + } + + static unrestricted_wrapped_type& + set_ref (wrapper_type& p) + { + if (!p) + p.reset (new unrestricted_wrapped_type); + + return const_cast<unrestricted_wrapped_type&> (*p); + } + }; + +#endif // ODB_CXX11 + // Specialization for odb::nullable. // template <typename T> |