Coverage Report

Coverage Report - org.kuali.core.db.torque.KualiTorqueSQLTask

Classes in this File

Line Coverage

Branch Coverage

Complexity

KualiTorqueSQLTask

0/42

0/8

1.556

 package org.kuali.core.db.torque;
 
 import java.io.File;
 import java.io.FileInputStream;
 import java.io.FileOutputStream;
 import java.util.Iterator;
 import java.util.Properties;
 
 import org.apache.torque.task.TorqueDataModelTask;
 import org.apache.velocity.context.Context;
 
 /**
  * Generate schema.sql and schema-constraints.sql from schema.xml files
  */
 public class KualiTorqueSQLTask extends TorqueDataModelTask {
         Utils utils = new Utils();
         PrettyPrint prettyPrint;
 
         private String database;
 
         private String suffix = "";
 
         /**
          * Sets the name of the database to generate sql for.
          * 
          * @param database
          *            the name of the database to generate sql for.
          */
         public void setDatabase(String database) {
                 this.database = database;
         }
 
         /**
          * Returns the name of the database to generate sql for.
          * 
          * @return the name of the database to generate sql for.
          */
         public String getDatabase() {
                 return database;
         }
 
         /**
          * Sets the suffix of the generated sql files.
          * 
          * @param suffix
          *            the suffix of the generated sql files.
          */
         public void setSuffix(String suffix) {
                 this.suffix = suffix;
         }
 
         /**
          * Returns the suffix of the generated sql files.
          * 
          * @return the suffix of the generated sql files.
          */
         public String getSuffix() {
                 return suffix;
         }
 
         protected DatabaseParser getDatabaseParser() {
                 return new KualiXmlToAppData(getTargetDatabase(), getTargetPackage());
         }
 
         /**
          * create the sql -> database map.
          * 
          * @throws Exception
          */
         protected void createSqlDbMap() throws Exception {
                 if (getSqlDbMap() == null) {
                         return;
                 }
 
                 // Produce the sql -> database map
                 Properties sqldbmap = new Properties();
                 Properties sqldbmap_c = new Properties();
 
                 // Check to see if the sqldbmap has already been created.
                 File file = new File(getSqlDbMap());
 
                 if (file.exists()) {
                         FileInputStream fis = new FileInputStream(file);
                         sqldbmap.load(fis);
                         fis.close();
                 }
 
                 Iterator<String> i = getDataModelDbMap().keySet().iterator();
 
                 while (i.hasNext()) {
                         String dataModelName = (String) i.next();
 
                         String databaseName;
 
                         if (getDatabase() == null) {
                                 databaseName = (String) getDataModelDbMap().get(dataModelName);
                         } else {
                                 databaseName = getDatabase();
                         }
 
                         String sqlFile = dataModelName + getSuffix() + ".sql";
                         sqldbmap.setProperty(sqlFile, databaseName);
                         sqlFile = dataModelName + getSuffix() + "-constraints.sql";
                         sqldbmap_c.setProperty(sqlFile, databaseName);
                 }
 
                 sqldbmap.store(new FileOutputStream(getSqlDbMap()), "Sqlfile -> Database map");
                 sqldbmap_c.store(new FileOutputStream(getSqlDbMap() + "-constraints"), "Sqlfile -> Database map");
         }
 
         public void onBeforeGenerate() {
                 prettyPrint = new PrettyPrint("[INFO] Generating schema SQL ");
                 utils.left(prettyPrint);
         }
 
         public void onAfterGenerate() {
                 utils.right(prettyPrint);
                 prettyPrint = null;
         }
 
         /**
          * Set up the initial context for generating the SQL from the XML schema.
          * 
          * @return the context
          * @throws Exception
          */
         public Context initControlContext() throws Exception {
                 super.initControlContext();
 
                 createSqlDbMap();
 
                 return context;
         }
 }

1		package org.kuali.core.db.torque;
2
3		import java.io.File;
4		import java.io.FileInputStream;
5		import java.io.FileOutputStream;
6		import java.util.Iterator;
7		import java.util.Properties;
8
9		import org.apache.torque.task.TorqueDataModelTask;
10		import org.apache.velocity.context.Context;
11
12		/**
13		* Generate schema.sql and schema-constraints.sql from schema.xml files
14		*/
15	0	public class KualiTorqueSQLTask extends TorqueDataModelTask {
16	0	Utils utils = new Utils();
17		PrettyPrint prettyPrint;
18
19		private String database;
20
21	0	private String suffix = "";
22
23		/**
24		* Sets the name of the database to generate sql for.
25		*
26		* @param database
27		* the name of the database to generate sql for.
28		*/
29		public void setDatabase(String database) {
30	0	this.database = database;
31	0	}
32
33		/**
34		* Returns the name of the database to generate sql for.
35		*
36		* @return the name of the database to generate sql for.
37		*/
38		public String getDatabase() {
39	0	return database;
40		}
41
42		/**
43		* Sets the suffix of the generated sql files.
44		*
45		* @param suffix
46		* the suffix of the generated sql files.
47		*/
48		public void setSuffix(String suffix) {
49	0	this.suffix = suffix;
50	0	}
51
52		/**
53		* Returns the suffix of the generated sql files.
54		*
55		* @return the suffix of the generated sql files.
56		*/
57		public String getSuffix() {
58	0	return suffix;
59		}
60
61		protected DatabaseParser getDatabaseParser() {
62	0	return new KualiXmlToAppData(getTargetDatabase(), getTargetPackage());
63		}
64
65		/**
66		* create the sql -> database map.
67		*
68		* @throws Exception
69		*/
70		protected void createSqlDbMap() throws Exception {
71	0	if (getSqlDbMap() == null) {
72	0	return;
73		}
74
75		// Produce the sql -> database map
76	0	Properties sqldbmap = new Properties();
77	0	Properties sqldbmap_c = new Properties();
78
79		// Check to see if the sqldbmap has already been created.
80	0	File file = new File(getSqlDbMap());
81
82	0	if (file.exists()) {
83	0	FileInputStream fis = new FileInputStream(file);
84	0	sqldbmap.load(fis);
85	0	fis.close();
86		}
87
88	0	Iterator<String> i = getDataModelDbMap().keySet().iterator();
89
90	0	while (i.hasNext()) {
91	0	String dataModelName = (String) i.next();
92
93		String databaseName;
94
95	0	if (getDatabase() == null) {
96	0	databaseName = (String) getDataModelDbMap().get(dataModelName);
97		} else {
98	0	databaseName = getDatabase();
99		}
100
101	0	String sqlFile = dataModelName + getSuffix() + ".sql";
102	0	sqldbmap.setProperty(sqlFile, databaseName);
103	0	sqlFile = dataModelName + getSuffix() + "-constraints.sql";
104	0	sqldbmap_c.setProperty(sqlFile, databaseName);
105	0	}
106
107	0	sqldbmap.store(new FileOutputStream(getSqlDbMap()), "Sqlfile -> Database map");
108	0	sqldbmap_c.store(new FileOutputStream(getSqlDbMap() + "-constraints"), "Sqlfile -> Database map");
109	0	}
110
111		public void onBeforeGenerate() {
112	0	prettyPrint = new PrettyPrint("[INFO] Generating schema SQL ");
113	0	utils.left(prettyPrint);
114	0	}
115
116		public void onAfterGenerate() {
117	0	utils.right(prettyPrint);
118	0	prettyPrint = null;
119	0	}
120
121		/**
122		* Set up the initial context for generating the SQL from the XML schema.
123		*
124		* @return the context
125		* @throws Exception
126		*/
127		public Context initControlContext() throws Exception {
128	0	super.initControlContext();
129
130	0	createSqlDbMap();
131
132	0	return context;
133		}
134		}