WebSVN – XServices – Diff – /xservices/trunk/src/main/java/net/brutex/xservices/util/BrutexHSQLQuartzConnectionProvider.java


/*
 *   Copyright 2011 Brian Rosenberger (Brutex Network)
 *
 *   Licensed 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 net.brutex.xservices.util;
 
import java.io.BufferedReader;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.IOException;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;
 
import org.apache.logging.log4j.Level;
import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.quartz.utils.ConnectionProvider;
 
/**
 * @author Brian Rosenberger
 * 
 */
public class BrutexHSQLQuartzConnectionProvider implements ConnectionProvider {
	
	private Connection conn = null;
	private static final Logger logger = LogManager.getLogger();
	
 
	public Connection getConnection() throws SQLException {
		if( conn!= null ) { // Todo: && conn.conn.isValid(5)) {
			logger.debug("Checking tables on pre-exisiting database connection.");
			checkTables();
			return conn;
		}
		try {
			// Class.forName("org.hsqldb.jdbc.JDBCDriver" );
			Class.forName("org.apache.derby.jdbc.EmbeddedDriver");
		} catch (Exception e) {
			logger.fatal("Failed to load Derby JDBC driver.");
			e.printStackTrace();
			return null;
		}
 
		if(isConnected(false)) {
			checkTables();
		} else {
			return null;
		}
		
		return conn;
	}
 
	public void shutdown() throws SQLException {
		try {
			// Class.forName("org.hsqldb.jdbc.JDBCDriver" );
			Class.forName("org.apache.derby.jdbc.EmbeddedDriver");
		} catch (Exception e) {
			System.err.println("ERROR: failed to load Derby JDBC driver.");
			e.printStackTrace();
			return;
		}
		String t = this.getClass().getClassLoader().getResource("/").toString()
				.substring(6);
		t += "../data/db";
		System.out.println("Shut down embedded database now.");
		Connection c = DriverManager.getConnection("jdbc:derby:" + t
				+ ";shutdown=true;");
 
	}
 
	@Override
	public void initialize() throws SQLException {
 
	}
 
	private synchronized void recursiveDelete(File dbDir) {
		File[] files = dbDir.listFiles();
		for (int i = 0; i < files.length; i++) {
			if (files[i].isFile()) {
				files[i].delete();
			} else {
				recursiveDelete(files[i]);
				files[i].delete();
			}
		}
		dbDir.delete();
	}
	
	private synchronized void checkTables() throws SQLException {
		logger.debug("Checking QUARTZ database schema.");
		if(!isConnected(false)) {
			logger.error("Failed to validate QUARTZ database schema.");
			return;
		}
		List<String> ddl_list = new ArrayList<String>(11);
		ddl_list.add("QRTZ_JOB_DETAILS");
		ddl_list.add("QRTZ_TRIGGERS");
		ddl_list.add("QRTZ_SIMPLE_TRIGGERS");
		ddl_list.add("QRTZ_CRON_TRIGGERS");
		ddl_list.add("QRTZ_SIMPROP_TRIGGERS");
		ddl_list.add("QRTZ_BLOB_TRIGGERS");
		ddl_list.add("QRTZ_CALENDARS");
		ddl_list.add("QRTZ_PAUSED_TRIGGER_GRPS");
		ddl_list.add("QRTZ_FIRED_TRIGGERS");
		ddl_list.add("QRTZ_SCHEDULER_STATE");
		ddl_list.add("QRTZ_LOCKS");
		
		String ddl = this.getClass().getClassLoader().getResource("/").toString()
		.substring(6)+ "../data/";
 
		DatabaseMetaData dmd = conn.getMetaData();
		for (String tbl : ddl_list) {
			ResultSet rs = dmd.getTables(null, "APP", tbl, null);
			if (!rs.next()) {
				logger.log(Level.INFO, "Adding DDL for table "+ tbl);
				Statement st = conn.createStatement();
				File ddlFile = new File(ddl + tbl + ".ddl");
				String create = "";
				try {
						BufferedReader r = new BufferedReader(new FileReader(ddlFile));
						while (r.ready()) {
							create += r.readLine() + "\n";
						}
						create.trim();
						if( st.execute(create)) {
							logger.log(Level.INFO, "Table " + tbl + " created.");
						} 
					} catch (FileNotFoundException ex) {
						ex.printStackTrace();
					} catch (IOException ex) {
						ex.printStackTrace();
					} catch (SQLException ex) {
						logger.log(Level.ERROR, "Error executing statement "+ create );
						System.out.println(ex.getMessage());
					}
				} else {
					logger.trace("Table "+tbl+" exists.");
				}
			}
		}
	
	private synchronized boolean isConnected(boolean fail) throws SQLException {
		if(conn!=null ) { // Todo: && conn.conn.isValid(5)) {) {
			return true;
		} else {
			String t = this.getClass().getClassLoader().getResource("/").toString().substring(6); // WEB-INF/classes
			t += "../data/db";
			logger.debug("Database directory is set to '" + t + "'");
			try {
				this.conn = DriverManager.getConnection("jdbc:derby:" + t + ";create=true;");
			} catch (SQLException ex) {
				logger.error(ex.getMessage(), ex);
				if(!fail) {
					logger.warn("Deleting database directory.");
					recursiveDelete(new File(t));
					logger.warn("Retrying to connect to database.");
					return isConnected(true);
				} else {
					return false;
				}
			}
		}
		return false;
	}
 
 
}

Rev 197	Rev 198
1	`/*`	1	`/*`
2	`* Copyright 2011 Brian Rosenberger (Brutex Network)`	2	`* Copyright 2011 Brian Rosenberger (Brutex Network)`
3	`*`	3	`*`
4	`* Licensed under the Apache License, Version 2.0 (the "License");`	4	`* Licensed under the Apache License, Version 2.0 (the "License");`
5	`* you may not use this file except in compliance with the License.`	5	`* you may not use this file except in compliance with the License.`
6	`* You may obtain a copy of the License at`	6	`* You may obtain a copy of the License at`
7	`*`	7	`*`
8	`* http://www.apache.org/licenses/LICENSE-2.0`	8	`* http://www.apache.org/licenses/LICENSE-2.0`
9	`*`	9	`*`
10	`* Unless required by applicable law or agreed to in writing, software`	10	`* Unless required by applicable law or agreed to in writing, software`
11	`* distributed under the License is distributed on an "AS IS" BASIS,`	11	`* distributed under the License is distributed on an "AS IS" BASIS,`
12	`* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.`	12	`* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.`
13	`* See the License for the specific language governing permissions and`	13	`* See the License for the specific language governing permissions and`
14	`* limitations under the License.`	14	`* limitations under the License.`
15	`*/`	15	`*/`
16		16
17	`package net.brutex.xservices.util;`	17	`package net.brutex.xservices.util;`
18		18
19	`import java.io.BufferedReader;`	19	`import java.io.BufferedReader;`
20	`import java.io.File;`	20	`import java.io.File;`
21	`import java.io.FileNotFoundException;`	21	`import java.io.FileNotFoundException;`
22	`import java.io.FileReader;`	22	`import java.io.FileReader;`
23	`import java.io.IOException;`	23	`import java.io.IOException;`
24	`import java.sql.Connection;`	24	`import java.sql.Connection;`
25	`import java.sql.DatabaseMetaData;`	25	`import java.sql.DatabaseMetaData;`
26	`import java.sql.DriverManager;`	26	`import java.sql.DriverManager;`
27	`import java.sql.ResultSet;`	27	`import java.sql.ResultSet;`
28	`import java.sql.SQLException;`	28	`import java.sql.SQLException;`
29	`import java.sql.Statement;`	29	`import java.sql.Statement;`
30	`import java.util.ArrayList;`	30	`import java.util.ArrayList;`
31	`import java.util.List;`	31	`import java.util.List;`
32		32
33	`import org.apache.logging.log4j.Level;`	33	`import org.apache.logging.log4j.Level;`
34	`import org.apache.logging.log4j.LogManager;`	34	`import org.apache.logging.log4j.LogManager;`
35	`import org.apache.logging.log4j.Logger;`	35	`import org.apache.logging.log4j.Logger;`
36	`import org.quartz.utils.ConnectionProvider;`	36	`import org.quartz.utils.ConnectionProvider;`
37		37
38	`/**`	38	`/**`
39	`* @author Brian Rosenberger`	39	`* @author Brian Rosenberger`
40	`*`	40	`*`
41	`*/`	41	`*/`
42	`public class BrutexHSQLQuartzConnectionProvider implements ConnectionProvider {`	42	`public class BrutexHSQLQuartzConnectionProvider implements ConnectionProvider {`
43		43
44	`private Connection conn = null;`	44	`private Connection conn = null;`
45	`private static final Logger logger = LogManager.getLogger();`	45	`private static final Logger logger = LogManager.getLogger();`
46		46
47		47
48	`public Connection getConnection() throws SQLException {`	48	`public Connection getConnection() throws SQLException {`
49	`if( conn!= null ) { // Todo: && conn.conn.isValid(5)) {`	49	`if( conn!= null ) { // Todo: && conn.conn.isValid(5)) {`
50	`logger.debug("Checking tables on pre-exisiting database connection.");`	50	`logger.debug("Checking tables on pre-exisiting database connection.");`
51	`checkTables();`	51	`checkTables();`
52	`return conn;`	52	`return conn;`
53	`}`	53	`}`
54	`try {`	54	`try {`
55	`// Class.forName("org.hsqldb.jdbc.JDBCDriver" );`	55	`// Class.forName("org.hsqldb.jdbc.JDBCDriver" );`
56	`Class.forName("org.apache.derby.jdbc.EmbeddedDriver");`	56	`Class.forName("org.apache.derby.jdbc.EmbeddedDriver");`
57	`} catch (Exception e) {`	57	`} catch (Exception e) {`
58	`logger.fatal("Failed to load Derby JDBC driver.");`	58	`logger.fatal("Failed to load Derby JDBC driver.");`
59	`e.printStackTrace();`	59	`e.printStackTrace();`
60	`return null;`	60	`return null;`
61	`}`	61	`}`
62		62
63	`if(isConnected(false)) {`	63	`if(isConnected(false)) {`
64	`checkTables();`	64	`checkTables();`
65	`} else {`	65	`} else {`
66	`return null;`	66	`return null;`
67	`}`	67	`}`
68		68
69	`return conn;`	69	`return conn;`
70	`}`	70	`}`
71		71
72	`public void shutdown() throws SQLException {`	72	`public void shutdown() throws SQLException {`
73	`try {`	73	`try {`
74	`// Class.forName("org.hsqldb.jdbc.JDBCDriver" );`	74	`// Class.forName("org.hsqldb.jdbc.JDBCDriver" );`
75	`Class.forName("org.apache.derby.jdbc.EmbeddedDriver");`	75	`Class.forName("org.apache.derby.jdbc.EmbeddedDriver");`
76	`} catch (Exception e) {`	76	`} catch (Exception e) {`
77	`System.err.println("ERROR: failed to load Derby JDBC driver.");`	77	`System.err.println("ERROR: failed to load Derby JDBC driver.");`
78	`e.printStackTrace();`	78	`e.printStackTrace();`
79	`return;`	79	`return;`
80	`}`	80	`}`
81	`String t = this.getClass().getClassLoader().getResource("/").toString()`	81	`String t = this.getClass().getClassLoader().getResource("/").toString()`
82	`.substring(6);`	82	`.substring(6);`
83	`t += "../data/db";`	83	`t += "../data/db";`
84	`System.out.println("Shut down embedded database now.");`	84	`System.out.println("Shut down embedded database now.");`
85	`Connection c = DriverManager.getConnection("jdbc:derby:" + t`	85	`Connection c = DriverManager.getConnection("jdbc:derby:" + t`
86	`+ ";shutdown=true;");`	86	`+ ";shutdown=true;");`
87		87
88	`}`	88	`}`
-		89
-		90	`@Override`
-		91	`public void initialize() throws SQLException {`
-		92
-		93	`}`
89		94
90	`private synchronized void recursiveDelete(File dbDir) {`	95	`private synchronized void recursiveDelete(File dbDir) {`
91	`File[] files = dbDir.listFiles();`	96	`File[] files = dbDir.listFiles();`
92	`for (int i = 0; i < files.length; i++) {`	97	`for (int i = 0; i < files.length; i++) {`
93	`if (files[i].isFile()) {`	98	`if (files[i].isFile()) {`
94	`files[i].delete();`	99	`files[i].delete();`
95	`} else {`	100	`} else {`
96	`recursiveDelete(files[i]);`	101	`recursiveDelete(files[i]);`
97	`files[i].delete();`	102	`files[i].delete();`
98	`}`	103	`}`
99	`}`	104	`}`
100	`dbDir.delete();`	105	`dbDir.delete();`
101	`}`	106	`}`
102		107
103	`private synchronized void checkTables() throws SQLException {`	108	`private synchronized void checkTables() throws SQLException {`
104	`logger.debug("Checking QUARTZ database schema.");`	109	`logger.debug("Checking QUARTZ database schema.");`
105	`if(!isConnected(false)) {`	110	`if(!isConnected(false)) {`
106	`logger.error("Failed to validate QUARTZ database schema.");`	111	`logger.error("Failed to validate QUARTZ database schema.");`
107	`return;`	112	`return;`
108	`}`	113	`}`
109	`List<String> ddl_list = new ArrayList<String>(11);`	114	`List<String> ddl_list = new ArrayList<String>(11);`
110	`ddl_list.add("QRTZ_JOB_DETAILS");`	115	`ddl_list.add("QRTZ_JOB_DETAILS");`
111	`ddl_list.add("QRTZ_TRIGGERS");`	116	`ddl_list.add("QRTZ_TRIGGERS");`
112	`ddl_list.add("QRTZ_SIMPLE_TRIGGERS");`	117	`ddl_list.add("QRTZ_SIMPLE_TRIGGERS");`
113	`ddl_list.add("QRTZ_CRON_TRIGGERS");`	118	`ddl_list.add("QRTZ_CRON_TRIGGERS");`
114	`ddl_list.add("QRTZ_SIMPROP_TRIGGERS");`	119	`ddl_list.add("QRTZ_SIMPROP_TRIGGERS");`
115	`ddl_list.add("QRTZ_BLOB_TRIGGERS");`	120	`ddl_list.add("QRTZ_BLOB_TRIGGERS");`
116	`ddl_list.add("QRTZ_CALENDARS");`	121	`ddl_list.add("QRTZ_CALENDARS");`
117	`ddl_list.add("QRTZ_PAUSED_TRIGGER_GRPS");`	122	`ddl_list.add("QRTZ_PAUSED_TRIGGER_GRPS");`
118	`ddl_list.add("QRTZ_FIRED_TRIGGERS");`	123	`ddl_list.add("QRTZ_FIRED_TRIGGERS");`
119	`ddl_list.add("QRTZ_SCHEDULER_STATE");`	124	`ddl_list.add("QRTZ_SCHEDULER_STATE");`
120	`ddl_list.add("QRTZ_LOCKS");`	125	`ddl_list.add("QRTZ_LOCKS");`
121		126
122	`String ddl = this.getClass().getClassLoader().getResource("/").toString()`	127	`String ddl = this.getClass().getClassLoader().getResource("/").toString()`
123	`.substring(6)+ "../data/";`	128	`.substring(6)+ "../data/";`
124		129
125	`DatabaseMetaData dmd = conn.getMetaData();`	130	`DatabaseMetaData dmd = conn.getMetaData();`
126	`for (String tbl : ddl_list) {`	131	`for (String tbl : ddl_list) {`
127	`ResultSet rs = dmd.getTables(null, "APP", tbl, null);`	132	`ResultSet rs = dmd.getTables(null, "APP", tbl, null);`
128	`if (!rs.next()) {`	133	`if (!rs.next()) {`
129	`logger.log(Level.INFO, "Adding DDL for table "+ tbl);`	134	`logger.log(Level.INFO, "Adding DDL for table "+ tbl);`
130	`Statement st = conn.createStatement();`	135	`Statement st = conn.createStatement();`
131	`File ddlFile = new File(ddl + tbl + ".ddl");`	136	`File ddlFile = new File(ddl + tbl + ".ddl");`
132	`String create = "";`	137	`String create = "";`
133	`try {`	138	`try {`
134	`BufferedReader r = new BufferedReader(new FileReader(ddlFile));`	139	`BufferedReader r = new BufferedReader(new FileReader(ddlFile));`
135	`while (r.ready()) {`	140	`while (r.ready()) {`
136	`create += r.readLine() + "\n";`	141	`create += r.readLine() + "\n";`
137	`}`	142	`}`
138	`create.trim();`	143	`create.trim();`
139	`if( st.execute(create)) {`	144	`if( st.execute(create)) {`
140	`logger.log(Level.INFO, "Table " + tbl + " created.");`	145	`logger.log(Level.INFO, "Table " + tbl + " created.");`
141	`}`	146	`}`
142	`} catch (FileNotFoundException ex) {`	147	`} catch (FileNotFoundException ex) {`
143	`ex.printStackTrace();`	148	`ex.printStackTrace();`
144	`} catch (IOException ex) {`	149	`} catch (IOException ex) {`
145	`ex.printStackTrace();`	150	`ex.printStackTrace();`
146	`} catch (SQLException ex) {`	151	`} catch (SQLException ex) {`
147	`logger.log(Level.ERROR, "Error executing statement "+ create );`	152	`logger.log(Level.ERROR, "Error executing statement "+ create );`
148	`System.out.println(ex.getMessage());`	153	`System.out.println(ex.getMessage());`
149	`}`	154	`}`
150	`} else {`	155	`} else {`
151	`logger.trace("Table "+tbl+" exists.");`	156	`logger.trace("Table "+tbl+" exists.");`
152	`}`	157	`}`
153	`}`	158	`}`
154	`}`	159	`}`
155		160
156	`private synchronized boolean isConnected(boolean fail) throws SQLException {`	161	`private synchronized boolean isConnected(boolean fail) throws SQLException {`
157	`if(conn!=null ) { // Todo: && conn.conn.isValid(5)) {) {`	162	`if(conn!=null ) { // Todo: && conn.conn.isValid(5)) {) {`
158	`return true;`	163	`return true;`
159	`} else {`	164	`} else {`
160	`String t = this.getClass().getClassLoader().getResource("/").toString().substring(6); // WEB-INF/classes`	165	`String t = this.getClass().getClassLoader().getResource("/").toString().substring(6); // WEB-INF/classes`
161	`t += "../data/db";`	166	`t += "../data/db";`
162	`logger.debug("Database directory is set to '" + t + "'");`	167	`logger.debug("Database directory is set to '" + t + "'");`
163	`try {`	168	`try {`
164	`this.conn = DriverManager.getConnection("jdbc:derby:" + t + ";create=true;");`	169	`this.conn = DriverManager.getConnection("jdbc:derby:" + t + ";create=true;");`
165	`} catch (SQLException ex) {`	170	`} catch (SQLException ex) {`
166	`logger.error(ex.getMessage(), ex);`	171	`logger.error(ex.getMessage(), ex);`
167	`if(!fail) {`	172	`if(!fail) {`
168	`logger.warn("Deleting database directory.");`	173	`logger.warn("Deleting database directory.");`
169	`recursiveDelete(new File(t));`	174	`recursiveDelete(new File(t));`
170	`logger.warn("Retrying to connect to database.");`	175	`logger.warn("Retrying to connect to database.");`
171	`return isConnected(true);`	176	`return isConnected(true);`
172	`} else {`	177	`} else {`
173	`return false;`	178	`return false;`
174	`}`	179	`}`
175	`}`	180	`}`
176	`}`	181	`}`
177	`return false;`	182	`return false;`
178	`}`	183	`}`
179		184
180		185
181	`}`	186	`}`

Subversion Repositories XServices

(root)/xservices/trunk/src/main/java/net/brutex/xservices/util/BrutexHSQLQuartzConnectionProvider.java – Rev 197 → 198