Add PostgreSQL

1 files changed, 107 insertions, 0 deletions

diff --git a/mapping/traits-pgsql.hxx b/mapping/traits-pgsql.hxx
new file mode 100644
index 0000000..6ee396d
--- /dev/null
+++ b/mapping/traits-pgsql.hxx
@@ -0,0 +1,107 @@
+// file      : mapping/traits-pgsql.hxx
+// author    : Constantin Michael <constantin@codesynthesis.com>
+// copyright : not copyrighted - public domain
+
+#ifndef TRAITS_PGSQL_HXX
+#define TRAITS_PGSQL_HXX
+
+//
+// PostgreSQL implementation.
+//
+
+#include <cstring>
+#include <ctime>
+
+#include <odb/pgsql/traits.hxx>
+#include <odb/pgsql/details/endian-traits.hxx>
+
+#include "person.hxx" // date
+
+namespace odb
+{
+  namespace pgsql
+  {
+    template <>
+    class value_traits<bool, id_string>
+    {
+    public:
+      typedef bool value_type;
+      typedef bool query_type;
+      typedef details::buffer image_type;
+
+      static void
+      set_value (bool& v,
+                 const details::buffer& b,
+                 std::size_t n,
+                 bool is_null)
+      {
+        v = (!is_null && n == 4 && std::strncmp ("true", b.data (), n) == 0);
+      }
+
+      static void
+      set_image (details::buffer& b,
+                 std::size_t& n,
+                 bool& is_null,
+                 bool v)
+      {
+        is_null = false;
+        n = v ? 4 : 5;
+
+        if (n > b.capacity ())
+          b.capacity (n);
+
+        std::memcpy (b.data (), (v ? "true" : "false"), n);
+      }
+    };
+
+    template <>
+    class value_traits<date, id_date>
+    {
+    public:
+      typedef date value_type;
+      typedef date query_type;
+      typedef int image_type;
+
+      static const time_t pg_epoch_tt;
+      static const time_t seconds_per_day = 86400;
+
+      static void
+      set_value (date& v, const int& i, bool is_null)
+      {
+        if (is_null)
+        {
+          v = date (0, 0, 0);
+          return;
+        }
+
+        time_t v_tt (pg_epoch_tt +
+                     static_cast<time_t> (details::endian_traits::ntoh (i)) *
+                     seconds_per_day);
+
+        tm v_tm (*localtime (&v_tt));
+
+        v = date (v_tm.tm_year + 1900, v_tm.tm_mon, v_tm.tm_mday);
+      }
+
+      static void
+      set_image (int& i, bool& is_null, const date& v)
+      {
+        is_null = false;
+
+        tm v_tm;
+        memset (&v_tm, 0, sizeof (v_tm));
+
+        v_tm.tm_mday = v.day ();
+        v_tm.tm_mon = v.month ();
+        v_tm.tm_year = v.year () - 1900;
+
+        time_t v_tt (mktime (&v_tm));
+
+        i = details::endian_traits::hton (
+          static_cast<int> ((v_tt - pg_epoch_tt) / seconds_per_day));
+      }
+    };
+  }
+}
+
+#endif // TRAITS_PGSQL_HXX