Subversion Repositories XServices

Rev

Rev 198 | Show entire file | Ignore whitespace | Details | Blame | Last modification | View Log | RSS feed

Rev 198 Rev 199
Line 28... Line 28...
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;
Line -... Line 32...
- 
 
 32 
 
32 
 
 33 
 
33 
import org.apache.logging.log4j.Level;
 - 
 
34 
import org.apache.logging.log4j.LogManager;
 - 
 
35 
import org.apache.logging.log4j.Logger;
 34 
import lombok.extern.slf4j.Slf4j;
Line 36... Line 35...
36 
import org.quartz.utils.ConnectionProvider;
 35 
import org.quartz.utils.ConnectionProvider;
37 
 
 36 
 
38 
/**
 37 
/**
39 
 * @author Brian Rosenberger
 38 
 * @author Brian Rosenberger
- 
 
 39 
 *
40 
 *
40 
 */
Line 41... Line 41...
41 
 */
 41 
@Slf4j
42 
public class BrutexQuartzConnectionProvider implements ConnectionProvider {
 - 
 
43
42 
public class BrutexQuartzConnectionProvider implements ConnectionProvider {
Line 44... Line 43...
44 
	private Connection conn = null;
 43
45 
	private final Logger logger = LogManager.getLogger();
 44 
	private Connection conn = null;
46
45 
 
47 
 
 46 
 
48 
	public Connection getConnection() throws SQLException {
 47 
	public Connection getConnection() throws SQLException {
49 
		if( conn!= null) { // Todo: && conn.conn.isValid(5)) {) {
 48 
		if( conn!= null) { // Todo: && conn.conn.isValid(5)) {) {
50 
			logger.debug("Checking tables on pre-exisiting database connection.");
 49 
			log.debug("Checking tables on pre-exisiting database connection.");
51 
			checkTables();
 50 
			checkTables();
52 
			return conn;
 51 
			return conn;
53 
		}
 52 
		}
54 
		try {
 53 
		try {
55 
			// Class.forName("org.hsqldb.jdbc.JDBCDriver" );
 54 
			// Class.forName("org.hsqldb.jdbc.JDBCDriver" );
56 
			Class.forName("org.apache.derby.jdbc.EmbeddedDriver");
 55 
			Class.forName("org.apache.derby.jdbc.EmbeddedDriver");
57 
		} catch (Exception e) {
 56 
		} catch (Exception e) {
Line 58... Line 57...
58 
			logger.fatal("Failed to load Derby JDBC driver.");
 57 
			log.error("Failed to load Derby JDBC driver.");
Line 104... Line 103...
104 
		}
 103 
		}
105 
		dbDir.delete();
 104 
		dbDir.delete();
106 
	}
 105 
	}
Line 107... Line 106...
107
106
108 
	private synchronized void checkTables() throws SQLException {
 107 
	private synchronized void checkTables() throws SQLException {
109 
		logger.debug("Checking QUARTZ database schema.");
 108 
		log.debug("Checking QUARTZ database schema.");
110 
		if(!isConnected(false)) {
 109 
		if(!isConnected(false)) {
111 
			logger.error("Failed to validate QUARTZ database schema.");
 110 
			log.error("Failed to validate QUARTZ database schema.");
112 
			return;
 111 
			return;
113 
		}
 112 
		}
114 
		List<String> ddl_list = new ArrayList<String>(11);
 113 
		List<String> ddl_list = new ArrayList<String>(11);
115 
		ddl_list.add("QRTZ_JOB_DETAILS");
 114 
		ddl_list.add("QRTZ_JOB_DETAILS");
Line 129... Line 128...
129 
 
 128 
 
130 
		DatabaseMetaData dmd = conn.getMetaData();
 129 
		DatabaseMetaData dmd = conn.getMetaData();
131 
		for (String tbl : ddl_list) {
 130 
		for (String tbl : ddl_list) {
132 
			ResultSet rs = dmd.getTables(null, "APP", tbl, null);
 131 
			ResultSet rs = dmd.getTables(null, "APP", tbl, null);
133 
			if (!rs.next()) {
 132 
			if (!rs.next()) {
134 
				logger.log(Level.INFO, "Adding DDL for table "+ tbl);
 133 
				log.debug("Adding DDL for table {}.", tbl);
135 
				Statement st = conn.createStatement();
 134 
				Statement st = conn.createStatement();
136 
				File ddlFile = new File(ddl + tbl + ".ddl");
 135 
				File ddlFile = new File(ddl + tbl + ".ddl");
137 
				String create = "";
 136 
				String create = "";
138 
				try {
 137 
				try {
139 
						BufferedReader r = new BufferedReader(new FileReader(ddlFile));
 138 
						BufferedReader r = new BufferedReader(new FileReader(ddlFile));
140 
						while (r.ready()) {
 139 
						while (r.ready()) {
141 
							create += r.readLine() + "\n";
 140 
							create += r.readLine() + "\n";
142 
						}
 141 
						}
143 
						create.trim();
 142 
						create.trim();
144 
						if( st.execute(create)) {
 143 
						if( st.execute(create)) {
145 
							logger.log(Level.INFO, "Table " + tbl + " created.");
 144 
							log.debug("Table {} created.", tbl);
146 
						}
- 
 
147 
					} catch (FileNotFoundException ex) {
 - 
 
148 
						ex.printStackTrace();
 145 
						}
149 
					} catch (IOException ex) {
 146 
					} catch (IOException ex) {
150 
						ex.printStackTrace();
 147 
						ex.printStackTrace();
151 
					} catch (SQLException ex) {
 148 
					} catch (SQLException ex) {
152 
						logger.log(Level.ERROR, "Error executing statement "+ create );
 149 
						log.error("Error executing statement {}", create, ex );
153 
						System.out.println(ex.getMessage());
 150 
						System.out.println(ex.getMessage());
154 
					}
 151 
					}
155 
				} else {
 152 
				} else {
156 
					logger.trace("Table "+tbl+" exists.");
 153 
					log.trace("Table {} exists.", tbl);
157 
				}
 154 
				}
158 
			}
 155 
			}
Line 159... Line 156...
159 
		}
 156 
		}
160
157
161 
	private synchronized boolean isConnected(boolean fail) throws SQLException {
 158 
	private synchronized boolean isConnected(boolean fail) throws SQLException {
162 
		if(conn!=null ) { // Todo: && conn.conn.isValid(5)) {) {
 159 
		if(conn!=null ) { // Todo: && conn.conn.isValid(5)) {) {
163 
			return true;
 160 
			return true;
164 
		} else {
 161 
		} else {
165 
			String t = this.getClass().getClassLoader().getResource("/").toString().substring(6); // WEB-INF/classes
 162 
			String t = this.getClass().getClassLoader().getResource("/").toString().substring(6); // WEB-INF/classes
166 
			t += "../data/db";
 163 
			t += "../data/db";
167 
			logger.debug("Database directory is set to '" + t + "'");
 164 
			log.debug("Database directory is set to '{}'", t);
168 
			try {
 165 
			try {
169 
				this.conn = DriverManager.getConnection("jdbc:derby:" + t + ";create=true;");
 166 
				this.conn = DriverManager.getConnection("jdbc:derby:" + t + ";create=true;");
170 
			} catch (SQLException ex) {
 167 
			} catch (SQLException ex) {
171 
				logger.error(ex.getMessage(), ex);
 168 
				log.error(ex.getMessage(), ex);
172 
				if(!fail) {
 169 
				if(!fail) {
173 
					logger.warn("Deleting database directory.");
 170 
					log.warn("Deleting database directory.");
174 
					recursiveDelete(new File(t));
 171 
					recursiveDelete(new File(t));
175 
					logger.warn("Retrying to connect to database.");
 172 
					log.warn("Retrying to connect to database.");
176 
					return isConnected(true);
 173 
					return isConnected(true);
177 
				} else {
 174 
				} else {
178 
					return false;
 175 
					return false;
179 
				}
 176 
				}
180 
			}
 177 
			}
181 
		}
 178 
		}
182 
		return false;
 - 
 
183 
	}
 179 
		return false;