Add hooks for generating extra code in object traits

1 files changed, 15 insertions, 0 deletions

diff --git a/odb/relational/header.hxx b/odb/relational/header.hxx
index 3ecc8d8..16c7855 100644
--- a/odb/relational/header.hxx
+++ b/odb/relational/header.hxx
@@ -728,6 +728,16 @@ namespace relational
       }
 
       virtual void
+      object_public_extra_pre (type&)
+      {
+      }
+
+      virtual void
+      object_public_extra_post (type&)
+      {
+      }
+
+      virtual void
       traverse_object (type& c)
       {
         bool abst (abstract (c));
@@ -745,6 +755,8 @@ namespace relational
            << "{"
            << "public:" << endl;
 
+        object_public_extra_pre (c);
+
         // object_type & pointer_type
         //
         os << "typedef " << type << " object_type;"
@@ -869,6 +881,7 @@ namespace relational
         //
         if (abst)
         {
+          object_public_extra_post (c);
           os << "};";
           return;
         }
@@ -974,6 +987,8 @@ namespace relational
              << endl;
         }
 
+        object_public_extra_post (c);
+
         // Implementation details.
         //
         os << "public:" << endl;