From 2043035eee4f8c0f3f896082651d3a5799e90ab8 Mon Sep 17 00:00:00 2001
From: Constantin Michael <constantin@codesynthesis.com>
Date: Fri, 21 Oct 2011 11:22:40 +0200
Subject: Add schema_file implementation to support schema file header and
 footer hooks

---
 odb/relational/oracle/schema.cxx | 25 +++++++++++++++++++++++++
 odb/relational/schema.cxx        |  5 +++++
 odb/relational/schema.hxx        | 19 +++++++++++++++++++
 3 files changed, 49 insertions(+)

diff --git a/odb/relational/oracle/schema.cxx b/odb/relational/oracle/schema.cxx
index 8a10d2e..2eb44d6 100644
--- a/odb/relational/oracle/schema.cxx
+++ b/odb/relational/oracle/schema.cxx
@@ -53,6 +53,31 @@ namespace relational
       entry<schema_emitter> schema_emitter_;
 
       //
+      // File.
+      //
+
+      struct schema_file: relational::schema_file, context
+      {
+        schema_file (const base& x): base (x) {}
+
+        virtual void
+        pre ()
+        {
+          os << "SET FEEDBACK OFF;" << endl
+             << "WHENEVER SQLERROR EXIT FAILURE;" << endl
+             << "WHENEVER OSERROR EXIT FAILURE;" << endl
+             << endl;
+        }
+
+        virtual void
+        post ()
+        {
+          os << "EXIT;" << endl;
+        }
+      };
+      entry<schema_file> schema_file_;
+
+      //
       // Drop.
       //
 
diff --git a/odb/relational/schema.cxx b/odb/relational/schema.cxx
index 4c3bfd0..0e64408 100644
--- a/odb/relational/schema.cxx
+++ b/odb/relational/schema.cxx
@@ -130,6 +130,9 @@ namespace relational
 
       instance<schema_emitter> emitter;
 
+      instance<schema_file> file;
+      file->pre ();
+
       // Drop.
       //
       {
@@ -181,6 +184,8 @@ namespace relational
           unit.dispatch (ctx.unit);
         }
       }
+
+      file->post ();
     }
   }
 }
diff --git a/odb/relational/schema.hxx b/odb/relational/schema.hxx
index 6c4cdee..29d4ff9 100644
--- a/odb/relational/schema.hxx
+++ b/odb/relational/schema.hxx
@@ -78,6 +78,25 @@ namespace relational
     };
 
     //
+    // File.
+    //
+
+    struct schema_file: virtual context
+    {
+      typedef schema_file base;
+
+      virtual void
+      pre ()
+      {
+      }
+
+      virtual void
+      post ()
+      {
+      }
+    };
+
+    //
     // Drop.
     //
 
-- 
cgit v1.1