aboutsummaryrefslogtreecommitdiff
path: root/evolution/add-index/driver.cxx
diff options
context:
space:
mode:
authorBoris Kolpackov <boris@codesynthesis.com>2013-04-04 09:47:33 +0200
committerBoris Kolpackov <boris@codesynthesis.com>2013-04-10 18:46:46 +0200
commit31b400b73be9aa98c0fd448423677db212667c43 (patch)
tree3f8d522d700ab6234da43211e27c136e740ab002 /evolution/add-index/driver.cxx
parent516523183fad016a26c7400d784eb9c0343b7964 (diff)
Generate add/drop index migration statements
Diffstat (limited to 'evolution/add-index/driver.cxx')
-rw-r--r--evolution/add-index/driver.cxx142
1 files changed, 142 insertions, 0 deletions
diff --git a/evolution/add-index/driver.cxx b/evolution/add-index/driver.cxx
new file mode 100644
index 0000000..e4a6e17
--- /dev/null
+++ b/evolution/add-index/driver.cxx
@@ -0,0 +1,142 @@
+// file : evolution/add-index/driver.cxx
+// copyright : Copyright (c) 2009-2013 Code Synthesis Tools CC
+// license : GNU GPL v2; see accompanying LICENSE file
+
+// Test adding a new index.
+//
+
+#include <memory> // std::auto_ptr
+#include <cassert>
+#include <iostream>
+
+#include <odb/database.hxx>
+#include <odb/transaction.hxx>
+
+#include <common/common.hxx>
+
+#include "test1.hxx"
+#include "test2.hxx"
+#include "test1-odb.hxx"
+#include "test2-odb.hxx"
+
+using namespace std;
+using namespace odb::core;
+
+int
+main (int argc, char* argv[])
+{
+ try
+ {
+ auto_ptr<database> db (create_database (argc, argv));
+
+ // 1 - base version
+ // 2 - migration
+ // 3 - current version
+ //
+ unsigned short pass (*argv[argc - 1] - '0');
+
+ switch (pass)
+ {
+ case 1:
+ {
+ using namespace v2;
+
+ object o0 (0);
+ o0.num = 123;
+
+ object o1 (1);
+ o1.num = 234;
+
+ object o2 (2);
+ o2.num = 234;
+
+ // Duplicates are ok.
+ //
+ {
+ transaction t (db->begin ());
+ db->persist (o0);
+ db->persist (o1);
+ db->persist (o2);
+ t.commit ();
+ }
+ break;
+ }
+ case 2:
+ {
+ using namespace v3;
+
+ object o3 (3);
+ o3.num = 234;
+
+ // Duplicates are still ok but we need to remove them before the
+ // post migration step.
+ //
+ {
+ transaction t (db->begin ());
+ db->persist (o3);
+ t.commit ();
+ }
+
+ {
+ typedef odb::query<object> query;
+ typedef odb::result<object> result;
+
+ transaction t (db->begin ());
+ result r (db->query<object> (
+ "ORDER BY" + query::num + "," + query::id));
+
+ unsigned long prev (0);
+ for (result::iterator i (r.begin ()); i != r.end (); ++i)
+ {
+ if (i->num == prev)
+ db->erase (*i);
+
+ prev = i->num;
+ }
+
+ t.commit ();
+ }
+
+ break;
+ }
+ case 3:
+ {
+ using namespace v3;
+
+ {
+ transaction t (db->begin ());
+ auto_ptr<object> p0 (db->load<object> (0));
+ auto_ptr<object> p1 (db->load<object> (1));
+
+ assert (p0->num == 123);
+ assert (p1->num == 234);
+
+ t.commit ();
+ }
+
+ try
+ {
+ object o2 (2);
+ o2.num = 234;
+
+ transaction t (db->begin ());
+ db->persist (o2);
+ assert (false);
+ }
+ catch (const odb::exception& ) {}
+
+ break;
+ }
+ default:
+ {
+ cerr << "unknown pass number '" << argv[argc - 1] << "'" << endl;
+ return 1;
+ }
+ }
+ }
+ catch (const odb::exception& e)
+ {
+ cerr << e.what () << endl;
+ return 1;
+ }
+}