From 8cca2127e0e11486cc45ae1a8198bd778301f935 Mon Sep 17 00:00:00 2001
From: Trygve Laugstøl <trygvis@inamo.no>
Date: Mon, 19 Aug 2013 19:45:28 +0200
Subject: o Dropping the Sequences class. Moving more stuff into SqlUnit to
 make it easier available in a generic form. o Fixing the complicated cases
 where the generation got confused with both setter and constructor injection.

---
 .../java/io/trygvis/container/myapp/AddressBookDirect.java  | 13 +++++--------
 1 file changed, 5 insertions(+), 8 deletions(-)

(limited to 'myapp/src/main/java/io/trygvis/container/myapp')

diff --git a/myapp/src/main/java/io/trygvis/container/myapp/AddressBookDirect.java b/myapp/src/main/java/io/trygvis/container/myapp/AddressBookDirect.java
index 7cb6b54..cd76c84 100644
--- a/myapp/src/main/java/io/trygvis/container/myapp/AddressBookDirect.java
+++ b/myapp/src/main/java/io/trygvis/container/myapp/AddressBookDirect.java
@@ -23,6 +23,7 @@ public class AddressBookDirect {
             sessionFactory = new MyAppSqlSessionFactory(new DriverManagerDataSource(jdbcUrl));
             new AddressBookDirect().work();
         } catch (EOFException ignore) {
+            // Happens when the user closes stdin.
         }
     }
 
@@ -89,19 +90,15 @@ public class AddressBookDirect {
     }
 
     public void create() throws SQLException {
-        session.executeUpdate(CompanyDao.createTableSql);
-        session.executeUpdate(ContactDao.createTableSql);
-        for (String sql : Sequences.createSequences) {
-            session.executeUpdate(sql);
+        for (String s : sessionFactory.getUnit().create()) {
+            session.executeUpdate(s);
         }
     }
 
     public void drop() throws SQLException {
-        for (String sql : Sequences.dropSequences) {
-            session.executeUpdate(sql);
+        for (String s : sessionFactory.getUnit().drop()) {
+            session.executeUpdate(s);
         }
-        session.executeUpdate(ContactDao.dropTableSql);
-        session.executeUpdate(CompanyDao.dropTableSql);
     }
 
     // -----------------------------------------------------------------------
-- 
cgit v1.2.3