Add traits for PostgreSQL query test

2 files changed, 61 insertions, 0 deletions

diff --git a/common/query/traits-pgsql.hxx b/common/query/traits-pgsql.hxx
new file mode 100644
index 0000000..5b5c5d3
--- /dev/null
+++ b/common/query/traits-pgsql.hxx
@@ -0,0 +1,59 @@
+// file      : common/query/traits-pgsql.hxx
+// author    : Constantin Michael <constantin@codesynthesis.com>
+// copyright : Copyright (c) 2009-2011 Code Synthesis Tools CC
+// license   : GNU GPL v2; see accompanying LICENSE file
+
+#ifndef TRAITS_PGSQL_HXX
+#define TRAITS_PGSQL_HXX
+
+#include <string>
+#include <memory>  // std::auto_ptr
+#include <cstring> // std::memcpy
+
+#include <odb/pgsql/traits.hxx>
+
+namespace odb
+{
+  namespace pgsql
+  {
+    template <>
+    class value_traits<std::auto_ptr<std::string>, id_string>
+    {
+    public:
+      typedef std::auto_ptr<std::string> value_type;
+      typedef std::string query_type;
+      typedef details::buffer image_type;
+
+      static void
+      set_value (std::auto_ptr<std::string>& v,
+                 const details::buffer& b,
+                 std::size_t n,
+                 bool is_null)
+      {
+        v.reset (is_null ? 0 : new std::string (b.data (), n));
+      }
+
+      static void
+      set_image (details::buffer& b,
+                 std::size_t& n,
+                 bool& is_null,
+                 const std::auto_ptr<std::string>& v)
+      {
+        is_null = v.get () == 0;
+
+        if (!is_null)
+        {
+          n = v->size ();
+
+          if (n > b.capacity ())
+            b.capacity (n);
+
+          if (n != 0)
+            std::memcpy (b.data (), v->c_str (), n);
+        }
+      }
+    };
+  }
+}
+
+#endif // TRAITS_PGSQL_HXX
diff --git a/common/query/traits.hxx b/common/query/traits.hxx
index ce8b3e6..ccc279b 100644
--- a/common/query/traits.hxx
+++ b/common/query/traits.hxx
@@ -12,6 +12,8 @@
 #  include "traits-mysql.hxx"
 #elif defined(DATABASE_SQLITE)
 #  include "traits-sqlite.hxx"
+#elif defined(DATABASE_PGSQL)
+#  include "traits-pgsql.hxx"
 #endif
 
 #endif // TRAITS_HXX