From 4b0a9726955bcca76bba07bc6f30ece1ecf83720 Mon Sep 17 00:00:00 2001
From: Eike Stepper
Date: Sun, 11 May 2008 11:11:56 +0000
Subject: Added store test framework

---
 .../src/org/eclipse/net4j/db/ddl/IDBSchema.java    |  7 +++
 .../eclipse/net4j/internal/db/ddl/DBSchema.java    | 58 ++++++++++++++++++++++
 2 files changed, 65 insertions(+)

(limited to 'plugins/org.eclipse.net4j.db')

diff --git a/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/db/ddl/IDBSchema.java b/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/db/ddl/IDBSchema.java
index c4b4c65ba6..fc82187e34 100644
--- a/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/db/ddl/IDBSchema.java
+++ b/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/db/ddl/IDBSchema.java
@@ -16,6 +16,7 @@ import org.eclipse.net4j.db.IDBConnectionProvider;
 
 import javax.sql.DataSource;
 
+import java.io.PrintStream;
 import java.sql.Connection;
 import java.util.Set;
 
@@ -41,4 +42,10 @@ public interface IDBSchema extends IDBSchemaElement
   public void drop(IDBAdapter dbAdapter, DataSource dataSource) throws DBException;
 
   public void drop(IDBAdapter dbAdapter, IDBConnectionProvider connectionProvider) throws DBException;
+
+  public void export(Connection connection, PrintStream out) throws DBException;
+
+  public void export(DataSource dataSource, PrintStream out) throws DBException;
+
+  public void export(IDBConnectionProvider connectionProvider, PrintStream out) throws DBException;
 }
diff --git a/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/internal/db/ddl/DBSchema.java b/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/internal/db/ddl/DBSchema.java
index d3e281ee8b..93a0a70898 100644
--- a/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/internal/db/ddl/DBSchema.java
+++ b/plugins/org.eclipse.net4j.db/src/org/eclipse/net4j/internal/db/ddl/DBSchema.java
@@ -14,12 +14,15 @@ import org.eclipse.net4j.db.DBException;
 import org.eclipse.net4j.db.DBUtil;
 import org.eclipse.net4j.db.IDBAdapter;
 import org.eclipse.net4j.db.IDBConnectionProvider;
+import org.eclipse.net4j.db.IDBRowHandler;
 import org.eclipse.net4j.db.ddl.IDBSchema;
 import org.eclipse.net4j.db.ddl.IDBTable;
 
 import javax.sql.DataSource;
 
+import java.io.PrintStream;
 import java.sql.Connection;
+import java.util.Arrays;
 import java.util.HashMap;
 import java.util.Map;
 import java.util.Set;
@@ -138,6 +141,61 @@ public class DBSchema extends DBSchemaElement implements IDBSchema
     }
   }
 
+  public void export(Connection connection, PrintStream out) throws DBException
+  {
+    for (DBTable table : getTables())
+    {
+      export(table, connection, out);
+    }
+  }
+
+  private void export(final DBTable table, Connection connection, final PrintStream out)
+  {
+    if (DBUtil.select(connection, new IDBRowHandler()
+    {
+      public boolean handle(int row, Object... values)
+      {
+        if (row == 0)
+        {
+          String tableName = table.getName();
+          out.println(tableName);
+          for (int i = 0; i < tableName.length(); i++)
+          {
+            out.print("=");
+          }
+
+          out.println();
+        }
+
+        out.println(Arrays.asList(values));
+        return true;
+      }
+    }, table.getFields()) > 0)
+    {
+      out.println();
+    }
+  }
+
+  public void export(DataSource dataSource, PrintStream out) throws DBException
+  {
+    export(DBUtil.createConnectionProvider(dataSource), out);
+  }
+
+  public void export(IDBConnectionProvider connectionProvider, PrintStream out) throws DBException
+  {
+    Connection connection = null;
+
+    try
+    {
+      connection = connectionProvider.getConnection();
+      export(connection, out);
+    }
+    finally
+    {
+      DBUtil.close(connection);
+    }
+  }
+
   void assertUnlocked() throws DBException
   {
     if (locked)
-- 
cgit v1.2.3