Make sure size() can still be called once we reached the end of result set

1 files changed, 11 insertions, 4 deletions

diff --git a/odb/mysql/no-id-object-result.txx b/odb/mysql/no-id-object-result.txx
index 864b49d..6418987 100644
--- a/odb/mysql/no-id-object-result.txx
+++ b/odb/mysql/no-id-object-result.txx
@@ -133,7 +133,7 @@ namespace odb
     void no_id_object_result_impl<T>::
     cache ()
     {
-      if (!statement_->cached ())
+      if (!this->end_ && !statement_->cached ())
       {
         statement_->cache ();
 
@@ -149,10 +149,17 @@ namespace odb
     std::size_t no_id_object_result_impl<T>::
     size ()
     {
-      if (!statement_->cached ())
-        throw result_not_cached ();
+      if (!this->end_)
+      {
+        if (!statement_->cached ())
+          throw result_not_cached ();
 
-      return statement_->result_size ();
+        return statement_->result_size ();
+      }
+      else
+        // If count is not zero, then it is one past the result size.
+        //
+        return count_ == 0 ? 0 : count_ - 1;
     }
   }
 }