diff options
| -rw-r--r-- | odb/pgsql/object-statements.cxx | 17 | ||||
| -rw-r--r-- | odb/pgsql/object-statements.hxx | 413 | ||||
| -rw-r--r-- | odb/pgsql/object-statements.ixx | 54 | ||||
| -rw-r--r-- | odb/pgsql/object-statements.txx | 89 |
4 files changed, 573 insertions, 0 deletions
diff --git a/odb/pgsql/object-statements.cxx b/odb/pgsql/object-statements.cxx new file mode 100644 index 0000000..921883d --- /dev/null +++ b/odb/pgsql/object-statements.cxx @@ -0,0 +1,17 @@ +// file : odb/pgsql/object-statements.cxx +// author : Constantin Michael <constantin@codesynthesis.com> +// copyright : Copyright (c) 2005-2011 Code Synthesis Tools CC +// license : GNU GPL v2; see accompanying LICENSE file + +#include <odb/pgsql/object-statements.hxx> + +namespace odb +{ + namespace pgsql + { + object_statements_base:: + ~object_statements_base () + { + } + } +} diff --git a/odb/pgsql/object-statements.hxx b/odb/pgsql/object-statements.hxx new file mode 100644 index 0000000..3059a01 --- /dev/null +++ b/odb/pgsql/object-statements.hxx @@ -0,0 +1,413 @@ +// file : odb/sqlite/object-statements.hxx +// author : Constantin Michael <constantin@codesynthesis.com> +// copyright : Copyright (c) 2005-2011 Code Synthesis Tools CC +// license : GNU GPL v2; see accompanying LICENSE file + +#ifndef ODB_PGSQL_OBJECT_STATEMENTS_HXX +#define ODB_PGSQL_OBJECT_STATEMENTS_HXX + +#include <odb/pre.hxx> + +#include <vector> +#include <cassert> +#include <cstddef> // std::size_t + +#include <odb/forward.hxx> +#include <odb/traits.hxx> +#include <odb/cache-traits.hxx> + +#include <odb/details/shared-ptr.hxx> + +#include <odb/pgsql/version.hxx> +#include <odb/pgsql/binding.hxx> +#include <odb/pgsql/statement.hxx> + +#include <odb/pgsql/details/export.hxx> + +namespace odb +{ + namespace pgsql + { + class connection; + + class LIBODB_PGSQL_EXPORT object_statements_base: + public details::shared_base + { + public: + typedef pgsql::connection connection_type; + + connection_type& + connection () + { + return conn_; + } + + // Locking. + // + void + lock () + { + assert (!locked_); + locked_ = true; + } + + void + unlock () + { + assert (locked_); + locked_ = false; + } + + bool + locked () const + { + return locked_; + } + + public: + virtual + ~object_statements_base (); + + protected: + object_statements_base (connection_type& conn) + : conn_ (conn), locked_ (false) + { + } + + protected: + connection_type& conn_; + bool locked_; + }; + + template <typename T> + class object_statements: public object_statements_base + { + public: + typedef T object_type; + typedef odb::object_traits<object_type> object_traits; + typedef typename object_traits::id_type id_type; + typedef typename object_traits::pointer_type pointer_type; + typedef typename object_traits::image_type image_type; + typedef typename object_traits::id_image_type id_image_type; + + typedef pointer_cache_traits<pointer_type> object_cache_traits; + + typedef + typename object_traits::container_statement_cache_type + container_statement_cache_type; + + typedef pgsql::insert_statement persist_statement_type; + typedef pgsql::select_statement find_statement_type; + typedef pgsql::update_statement update_statement_type; + typedef pgsql::delete_statement erase_statement_type; + + // Automatic lock. + // + struct auto_lock + { + // Lock the statements unless they are already locked in which + // case subsequent calls to locked() will return false. + // + auto_lock (object_statements&); + + // Unlock the statemens if we are holding the lock and clear + // the delayed loads. This should only happen in case an + // exception is thrown. In normal circumstances, the user + // should call unlock() explicitly. + // + ~auto_lock (); + + // Return true if this auto_lock instance holds the lock. + // + bool + locked () const; + + // Unlock the statemens. + // + void + unlock (); + + private: + auto_lock (const auto_lock&); + auto_lock& operator= (const auto_lock&); + + private: + object_statements& s_; + bool locked_; + }; + + // + // + object_statements (connection_type&); + + // Delayed loading. + // + void + delay_load (const id_type& id, + object_type& obj, + const typename object_cache_traits::position_type& p) + { + delayed_.push_back (delayed_load (id, obj, p)); + } + + void + load_delayed () + { + assert (locked ()); + + if (!delayed_.empty ()) + load_delayed_ (); + } + + void + clear_delayed () + { + if (!delayed_.empty ()) + clear_delayed_ (); + } + + // Object image. + // + image_type& + image () + { + return image_; + } + + std::size_t + in_image_version () const + { + return in_image_version_; + } + + std::size_t + out_image_version () const + { + return out_image_version_; + } + + void + in_image_version (std::size_t v) + { + in_image_version_ = v; + } + + void + out_image_version (std::size_t v) + { + out_image_version_ = v; + } + + binding& + in_image_binding () + { + return in_image_binding_; + } + + binding& + out_image_binding () + { + return out_image_binding_; + } + + bool* + out_image_truncated () + { + return out_image_truncated_; + } + + // Object id image. + // + id_image_type& + id_image () + { + return id_image_; + } + + std::size_t + id_image_version () const + { + return id_image_version_; + } + + void + id_image_version (std::size_t v) + { + id_image_version_ = v; + } + + binding& + id_image_binding () + { + return id_image_binding_; + } + + // Statements. + // + persist_statement_type& + persist_statement () + { + if (persist_ == 0) + { + persist_.reset ( + new (details::shared) persist_statement_type ( + conn_, + // @@ name, + object_traits::persist_statement, + object_traits::persist_statement_types, + object_traits::persist_statement_count, + in_image_binding_, + in_image_native_binding)); + + persist_->cached (true); + } return *persist_; + } + + find_statement_type& + find_statement () + { + if (find_ == 0) + { + find_.reset ( + new (details::shared) find_statement_type ( + conn_, + // @@ name, + object_traits::find_statement, + object_traits::find_statement_types, + object_traits::find_statement_count, + id_image_binding_, + id_image_native_binding_, + out_image_binding_)); + + find_->cached (true); + } + + return *find_; + } + + update_statement_type& + update_statement () + { + if (update_ == 0) + { + update_.reset ( + new (details::shared) update_statement_type ( + conn_, + // @@ name, + object_traits::update_statement, + object_traits::update_statement_types, + object_traits::update_statement_count, + id_image_binding_, + id_image_native_binding, + in_image_binding_ + in_image_native_binding)); + + update_->cached (true); + } + + return *update_; + } + + erase_statement_type& + erase_statement () + { + if (erase_ == 0) + { + erase_.reset ( + new (details::shared) erase_statement_type ( + conn_, + // @@ name, + object_traits::erase_statement, + object_traits::erase_statement_types, + object_traits::erase_statment_count, + id_image_binding_ + id_image_native_binding_)); + + erase_->cached (true); + } + + return *erase_; + } + + // Container statement cache. + // + container_statement_cache_type& + container_statment_cache () + { + return container_statement_cache_; + } + + private: + object_statements (const object_statements&); + object_statements& operator= (const object_statements&); + + private: + void + load_delayed_ (); + + void + clear_delayed_ (); + + private: + container_statement_cache_type container_statement_cache_; + + image_type image_; + + // In (send) binding. The last element is the id parameter. + // + std::size_t in_image_version_; + binding in_image_binding_; + bind in_image_bind_[object_traits::in_column_count + 1]; + native_binding in_image_native_binding_; + char* in_image_values_[object_traits::in_column_count + 1]; + int in_image_lengths_[object_traits::in_column_count + 1]; + int in_image_formats_[object_traits::in_column_count + 1]; + + // Out (receive) binding. + // + std::size_t out_image_version_; + binding out_image_binding_; + bind out_image_bind_[object_traits::out_column_count]; + bool out_image_truncated_[object_traits::out_column_count]; + + // Id image binding (only in). + // + id_image_type id_image_; + std::size_t id_image_version_; + binding id_image_binding_; + native_binding id_image_native_binding_; + + details::shared_ptr<persist_statement_type> persist_; + details::shared_ptr<find_statement_type> find_; + details::shared_ptr<update_statement_type> update_; + details::shared_ptr<erase_statement_type> erase_; + + // Delayed loading. + // + struct delayed_load + { + typedef typename object_cache_traits::position_type position_type; + + delayed_load () {} + delayed_load (const id_type& i, object_type& o, const position_type& p) + : id (i), obj (&o), pos (p) + { + } + + id_type id; + object_type* obj; + position_type pos; + }; + + typedef std::vector<delayed_load> delayed_loads; + delayed_loads delayed_; + }; + } +} + +#include <odb/pgsql/object-statements.ixx> +#include <odb/pgsql/object-statements.txx> + +#include <odb/post.hxx> + +#endif // ODB_PGSQL_OBJECT_STATEMENTS_HXX diff --git a/odb/pgsql/object-statements.ixx b/odb/pgsql/object-statements.ixx new file mode 100644 index 0000000..ea19baa --- /dev/null +++ b/odb/pgsql/object-statements.ixx @@ -0,0 +1,54 @@ +// file : odb/pgsql/object-statements.ixx +// author : Constantin Michael <constantin@codesynthesis.com> +// copyright : Copyright (c) 2005-2011 Code Synthesis Tools CC +// license : GNU GPL v2; see accompanying LICENSE file + +namespace odb +{ + namespace pgsql + { + // + // auto_lock + // + template <typename T> + inline object_statements<T>::auto_lock:: + auto_lock (object_statements& s) + : s_ (s) + { + if (!s_.locked ()) + { + s_.lock (); + locked_ = true; + } + else + locked_ = false; + } + + template <typename T> + inline object_statements<T>::auto_lock:: + ~auto_lock () + { + if (locked_) + { + s_.unlock (); + s_.clear_delayed (); + } + } + + template <typename T> + inline bool object_statements<T>::auto_lock:: + locked () const + { + return locked_; + } + + template <typename T> + inline void object_statements<T>::auto_lock:: + unlock () + { + assert (locked_); + s_.unlock (); + locked_ = false; + } + } +} diff --git a/odb/pgsql/object-statements.txx b/odb/pgsql/object-statements.txx new file mode 100644 index 0000000..6209d59 --- /dev/null +++ b/odb/pgsql/object-statements.txx @@ -0,0 +1,89 @@ +// file : odb/pgsql/object-statements.txx +// author : Constantin Michael <constantin@codesynthesis.com> +// copyright : Copyright (c) 2005-2011 Code Synthesis Tools CC +// license : GNU GPL v2; see accompanying LICENSE file + +#include <cstddef> // std::size_t +#include <cstring> // std::memset + +#include <odb/session.hxx> +#include <odb/exceptions.hxx> + +#include <odb/pgsql/connection.hxx> + +namespace odb +{ + namespace pgsql + { + template <typename T> + object_statements<T>:: + object_statements (connection_type& conn) + : object_statements_base (conn), + container_statement_cache_ (conn), + in_image_binding_ (in_image_bind_, object_traits::in_column_count), + in_image_native_binding_ (in_image_values_, in_image_lengths_, in_image_formats_, object_traits::in_column_count) + out_image_binding_ (out_image_bind_, object_traits::out_column_count), + id_image_binding_ (in_image_bind_ + object_traits::in_column_count, 1), + id_image_native_binding_ (in_image_values_ + object_traits::in_column_count, in_image_lengths_ + object_traits::in_column_count, in_image_formats_ + object_traits::in_column_count, 1) + { + image_.version = 0; + in_image_version_ = 0; + out_image_version_ = 0; + + id_image_.version = 0; + id_image_version_ = 0; + + std::memset (in_image_bind_, 0, sizeof (in_image_bind_)); + std::memset (out_image_bind_, 0, sizeof (out_image_bind_)); + std::memset (out_image_truncated_, 0, sizeof (out_image_truncated_)); + + // @@ Is it necessary to zero native array contents? + // + + for (std::size_t i (0); i < object_traits::out_column_count; ++i) + out_image_bind_[i].truncated = out_image_truncated_ + i; + } + + template <typename T> + void object_statements<T>:: + load_delayed_ () + { + // We should be careful here: the delayed vector can change + // from under us as a result of a recursive load. + // + database& db (connection ().database ()); + + while (!delayed_.empty ()) + { + delayed_load l (delayed_.back ()); + typename object_cache_traits::insert_guard g (l.pos); + delayed_.pop_back (); + + if (!object_traits::find_ (*this, l.id)) + throw object_not_persistent (); + + object_traits::init (*l.obj, image (), db); + object_traits::load_ (*this, *l.obj); // Load containers, etc. + g.release (); + } + } + + template <typename T> + void object_statements<T>:: + clear_delayed_ () + { + // Remove the objects from the session cache. + // + if (session::has_current ()) + { + for (typename delayed_loads::iterator i (delayed_.begin ()), + e (delayed_.end ()); i != e; ++i) + { + object_cache_traits::erase (i->pos); + } + } + + delayed_.clear (); + } + } +} |