Add query and result implementation

1 files changed, 134 insertions, 0 deletions

diff --git a/odb/pgsql/result.txx b/odb/pgsql/result.txx
new file mode 100644
index 0000000..f96fe81
--- /dev/null
+++ b/odb/pgsql/result.txx
@@ -0,0 +1,134 @@
+// file      : odb/pgsql/result.txx
+// author    : Constantin Michael <constantin@codesynthesis.com>
+// copyright : Copyright (c) 2009-2011 Code Synthesis Tools CC
+// license   : GNU GPL v2; see accompanying LICENSE file
+
+#include <odb/exceptions.hxx>
+
+namespace odb
+{
+  namespace pgsql
+  {
+    template <typename T>
+    result_impl<T>::
+    ~result_impl ()
+    {
+    }
+
+    template <typename T>
+    result_impl<T>::
+    result_impl (const query&,
+                 details::shared_ptr<select_statement> st,
+                 object_statements<object_type>& sts)
+        : odb::result_impl<T> (sts.connection ().database ()),
+          statement_ (st),
+          statements_ (sts),
+          count_ (0)
+    {
+    }
+
+    template <typename T>
+    void result_impl<T>::
+    load (object_type& obj)
+    {
+      load_image ();
+
+      // This is a top-level call so the statements cannot be locked.
+      //
+      assert (!statements_.locked ());
+      typename object_statements<object_type>::auto_lock l (statements_);
+
+      typename object_traits::image_type& i (statements_.image ());
+      object_traits::init (obj, i, this->database ());
+
+      // Initialize the id image and binding and load the rest of the object
+      // (containers, etc).
+      //
+      typename object_traits::id_image_type& idi (statements_.id_image ());
+      object_traits::init (idi, object_traits::id (i));
+
+      binding& idb (statements_.id_image_binding ());
+      if (idi.version != statements_.id_image_version () || idb.version == 0)
+      {
+        object_traits::bind (idb.bind, idi);
+        statements_.id_image_version (idi.version);
+        idb.version++;
+      }
+
+      object_traits::load_ (statements_, obj);
+
+      statements_.load_delayed ();
+      l.unlock ();
+    }
+
+    template <typename T>
+    typename result_impl<T>::id_type result_impl<T>::
+    load_id ()
+    {
+      load_image ();
+      return object_traits::id (statements_.image ());
+    }
+
+    template <typename T>
+    void result_impl<T>::
+    next ()
+    {
+      this->current (pointer_type ());
+
+      // Increment the position and postpone the actual row fetching
+      // until later. This way if the same object is loaded in between
+      // iteration, the image won't be messed up.
+      //
+      if (++count_ > statement_->result_size ())
+        this->end_ = true;
+    }
+
+    template <typename T>
+    void result_impl<T>::
+    load_image ()
+    {
+      // The image can grow between calls to load() as a result of other
+      // statements execution.
+      //
+      typename object_traits::image_type& im (statements_.image ());
+
+      if (im.version != statements_.out_image_version ())
+      {
+        binding& b (statements_.out_image_binding ());
+        object_traits::bind (b.bind, im, true);
+        statements_.out_image_version (im.version);
+        b.version++;
+      }
+
+      select_statement::result r (statement_->fetch ());
+
+      if (r == select_statement::truncated)
+      {
+        if (object_traits::grow (im, statements_.out_image_truncated ()))
+          im.version++;
+
+        if (im.version != statements_.out_image_version ())
+        {
+          binding& b (statements_.out_image_binding ());
+          object_traits::bind (b.bind, im, true);
+          statements_.out_image_version (im.version);
+          b.version++;
+          statement_->refetch ();
+        }
+      }
+    }
+
+    template <typename T>
+    void result_impl<T>::
+    cache ()
+    {
+    }
+
+    template <typename T>
+    std::size_t result_impl<T>::
+    size ()
+    {
+      return statement_->result_size ();
+    }
+  }
+}