summaryrefslogtreecommitdiff
path: root/odb-tests/pgsql/custom/test.hxx
diff options
context:
space:
mode:
Diffstat (limited to 'odb-tests/pgsql/custom/test.hxx')
-rw-r--r--odb-tests/pgsql/custom/test.hxx86
1 files changed, 86 insertions, 0 deletions
diff --git a/odb-tests/pgsql/custom/test.hxx b/odb-tests/pgsql/custom/test.hxx
new file mode 100644
index 0000000..ec3d496
--- /dev/null
+++ b/odb-tests/pgsql/custom/test.hxx
@@ -0,0 +1,86 @@
+// file : pgsql/custom/test.hxx
+// license : GNU GPL v2; see accompanying LICENSE file
+
+#ifndef TEST_HXX
+#define TEST_HXX
+
+#include <string>
+#include <vector>
+
+#include <odb/core.hxx>
+
+// Map POINT PostgreSQL type to the point C++ struct. The other half
+// of this mapping is in traits.hxx (value_traits<point, id_string>).
+//
+#pragma db map type("POINT") as("TEXT") to("(?)::POINT") from("(?)::TEXT")
+
+#pragma db value type("POINT")
+struct point
+{
+ point () {}
+ point (double x_, double y_): x (x_), y (y_) {}
+
+ double x;
+ double y;
+};
+
+inline bool
+operator== (const point& a, const point& b)
+{
+ return a.x == b.x && a.y == b.y;
+}
+
+// Map NUMERIC PostgreSQL type to std::string (or any other type that
+// provides the value_traits<?, id_string> specialization).
+//
+#pragma db map type("NUMERIC *(\\(.+\\))?") \
+ as("TEXT") \
+ to("(?)::NUMERIC$1") \
+ from("(?)::TEXT")
+
+// Map INTEGER[] PostgreSQL type to std::vector<int>. The other half of
+// this mapping is in traits.hxx (value_traits<std::vector<int>, id_string>).
+//
+#pragma db map type("INTEGER *\\[(\\d*)\\]") \
+ as("TEXT") \
+ to("(?)::INTEGER[$1]") \
+ from("(?)::TEXT")
+
+#pragma db object
+struct object
+{
+ object () {}
+ object (unsigned long id_) : id (id_) {}
+
+ #pragma db id
+ unsigned long id;
+
+ point p;
+ std::vector<point> pv;
+
+ #pragma db type("NUMERIC(6, 4)")
+ std::string n1;
+
+ #pragma db type("NUMERIC(6)")
+ std::string n2;
+
+ #pragma db type("NUMERIC")
+ std::string n3;
+
+ #pragma db type("INTEGER [123]")
+ std::vector<int> iv;
+
+ bool
+ operator== (const object& y) const
+ {
+ return id == y.id &&
+ p == y.p &&
+ pv == y.pv &&
+ n1 == y.n1 &&
+ n2 == y.n2 &&
+ n3 == y.n3 &&
+ iv == y.iv;
+ }
+};
+
+#endif // TEST_HXX