1 files changed, 119 insertions, 0 deletions
diff --git a/das-java/trunk/tools/src/main/java/org/apache/tuscany/das/rdb/util/DBToSchemaFile.java b/das-java/trunk/tools/src/main/java/org/apache/tuscany/das/rdb/util/DBToSchemaFile.java
new file mode 100644
index 0000000000..1dfa7919fd
--- /dev/null
+++ b/das-java/trunk/tools/src/main/java/org/apache/tuscany/das/rdb/util/DBToSchemaFile.java
@@ -0,0 +1,119 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ * 
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.    
+ */
+package org.apache.tuscany.das.rdb.util;
+
+import java.io.File;
+import java.io.InputStreamReader;
+
+import javax.xml.stream.XMLInputFactory;
+import javax.xml.stream.XMLStreamReader;
+
+import org.apache.log4j.Logger;
+import org.apache.tools.ant.Project;
+import org.apache.torque.task.TorqueJDBCTransformTask;
+
+public class DBToSchemaFile {
+	private static final Logger logger = Logger.getLogger(DBToSchemaFile.class);
+	private static ModelXSDGenOption mo = null;//schemaFileName should not be null. id modelFileName null STDOUT
+	
+	protected static void schemaFileFromDB() throws Exception {
+		Project p = new Project();
+	    p.setBaseDir(new File("."));
+		TorqueJDBCTransformTask tsk = new TorqueJDBCTransformTask();
+		tsk.setProject(p);
+		tsk.setDbDriver(mo.getDriverClass());
+		tsk.setDbUrl(mo.getDatabaseURL());
+		tsk.setSameJavaName(true);
+		tsk.setDbSchema(mo.getSchemaName());
+		tsk.setTaskName("jdbc");
+		tsk.setDbUser(mo.getUserName());
+		tsk.setDbPassword(mo.getPassword());
+		if(!mo.getSchemaFile().trim().equals("")) {
+			File schemaFile = new File( mo.getSchemaFile());
+			schemaFile.createNewFile();
+		}
+		tsk.setOutputFile(mo.getSchemaFile());
+		tsk.execute();
+	}
+	
+	protected static void read(String dbInfoFileName) throws Exception {
+		XMLInputFactory xmlFactory = XMLInputFactory.newInstance();
+		
+		XMLStreamReader reader = xmlFactory.createXMLStreamReader(new InputStreamReader(DBToXSDGenerator.getStream(dbInfoFileName)));
+		mo = new ModelXSDGenOption();
+        while (true) {
+            int event = reader.next();
+            if(javax.xml.stream.XMLStreamConstants.END_DOCUMENT == event) {
+            	break;
+            }
+            
+            switch (event) {            
+	            case javax.xml.stream.XMLStreamConstants.START_ELEMENT: {
+	               if (reader.getName().getLocalPart().equals("ConnectionProperties")) {
+	            	   mo.setDriverClass(reader.getAttributeValue(null, "driverClass"));
+	                   mo.setDatabaseURL(reader.getAttributeValue(null, "databaseURL"));
+	                   mo.setSchemaName(reader.getAttributeValue(null, "schemaName"));
+	                   mo.setUserName(reader.getAttributeValue(null, "userName"));
+	                	if(mo.getUserName() == null)
+	                		mo.setUserName("");
+	                	mo.setPassword(reader.getAttributeValue(null, "password"));
+	                	if(mo.getPassword() == null)
+	                		mo.setPassword("");
+	                } else if (reader.getName().getLocalPart().equals("ConnectionInfo")) {	                
+	                	//ignore
+	                } else if (reader.getName().getLocalPart().equals("Config")) {
+	                	//ignore
+	                } else if (reader.getName().getLocalPart().equals("OutFiles")) {	                
+	                	mo.setSchemaFile(reader.getAttributeValue(null, "schemaFile"));
+	                	mo.setModelFile(reader.getAttributeValue(null, "modelFile"));
+	                } else {
+	                	throw new RuntimeException("not got dbInfo  - tableNames List or connectionInfo:"+reader.getName()+":");
+	                }
+	                break;
+	            }
+            }            
+        }
+        
+        if (logger.isDebugEnabled()) {
+			logger.debug("driverClass:"+mo.getDriverClass());
+			logger.debug("url:"+mo.getDatabaseURL());
+			logger.debug("schemaName:"+mo.getSchemaName());
+			logger.debug("schemaFileName:"+mo.getSchemaFile());
+			logger.debug("modelFileName:"+mo.getModelFile());
+			logger.debug("userName:"+mo.getUserName());
+			logger.debug("password:"+mo.getPassword());
+		}            
+
+        if(mo.getDriverClass() == null || mo.getDatabaseURL() == null || mo.getSchemaName() == null || mo.getSchemaFile() == null 
+        		|| mo.getModelFile() == null) {
+     	   throw new RuntimeException("Required inputs missing - check driverClass, url, schemaName, schemaFile, modelFile!");
+        }
+
+        return;
+    }
+
+	public static ModelXSDGenOption getMo() {
+		return mo;
+	}
+
+	public static void setMo(ModelXSDGenOption mo) {
+		DBToSchemaFile.mo = mo;
+	}
+	
+}