/* Copyright (c) 2001-2010, The HSQL Development Group * All rights reserved. * * Redistribution and use in source and binary forms, with or without * modification, are permitted provided that the following conditions are met: * * Redistributions of source code must retain the above copyright notice, this * list of conditions and the following disclaimer. * * Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * * Neither the name of the HSQL Development Group nor the names of its * contributors may be used to endorse or promote products derived from this * software without specific prior written permission. * * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE * ARE DISCLAIMED. IN NO EVENT SHALL HSQL DEVELOPMENT GROUP, HSQLDB.ORG, * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. */ package org.hsqldb.persist; import java.io.EOFException; import org.hsqldb.Database; import org.hsqldb.HsqlNameManager.HsqlName; import org.hsqldb.Session; import org.hsqldb.Statement; import org.hsqldb.StatementDML; import org.hsqldb.StatementTypes; import org.hsqldb.error.Error; import org.hsqldb.error.ErrorCode; import org.hsqldb.lib.IntKeyHashMap; import org.hsqldb.lib.StopWatch; import org.hsqldb.result.Result; import org.hsqldb.result.ResultProperties; import org.hsqldb.scriptio.ScriptReaderBase; import org.hsqldb.scriptio.ScriptReaderDecode; import org.hsqldb.scriptio.ScriptReaderText; /** * Restores the state of a Database instance from an SQL log file. <p> * * If there is an error, processing stops at that line and the message is * logged to the application log. If memory runs out, an exception is thrown. * * @author Fred Toussi (fredt@users dot sourceforge.net) * @version 1.9.0 * @since 1.7.2 */ public class ScriptRunner { /** * This is used to read the *.log file and manage any necessary * transaction rollback. */ public static void runScript(Database database, String logFilename) { IntKeyHashMap sessionMap = new IntKeyHashMap(); Session current = null; int currentId = 0; database.setReferentialIntegrity(false); ScriptReaderBase scr = null; String statement; int statementType; Statement dummy = new StatementDML(StatementTypes.UPDATE_CURSOR, StatementTypes.X_SQL_DATA_CHANGE, null); try { StopWatch sw = new StopWatch(); Crypto crypto = database.logger.getCrypto(); if (crypto == null) { scr = new ScriptReaderText(database, logFilename); } else { scr = new ScriptReaderDecode(database, logFilename, crypto, true); } while (scr.readLoggedStatement(current)) { int sessionId = scr.getSessionNumber(); if (current == null || currentId != sessionId) { currentId = sessionId; current = (Session) sessionMap.get(currentId); if (current == null) { current = database.getSessionManager().newSession(database, database.getUserManager().getSysUser(), false, true, null, 0); sessionMap.put(currentId, current); } } if (current.isClosed()) { sessionMap.remove(currentId); continue; } Result result = null; statementType = scr.getStatementType(); switch (statementType) { case ScriptReaderBase.ANY_STATEMENT : statement = scr.getLoggedStatement(); result = current.executeDirectStatement(statement, ResultProperties.defaultPropsValue); if (result != null && result.isError()) { if (result.getException() != null) { throw result.getException(); } throw Error.error(result); } break; case ScriptReaderBase.COMMIT_STATEMENT : current.commit(false); break; case ScriptReaderBase.INSERT_STATEMENT : { current.sessionContext.currentStatement = dummy; current.beginAction(dummy); Object[] data = scr.getData(); scr.getCurrentTable().insertNoCheckFromLog(current, data); current.endAction(Result.updateOneResult); break; } case ScriptReaderBase.DELETE_STATEMENT : { current.sessionContext.currentStatement = dummy; current.beginAction(dummy); Object[] data = scr.getData(); scr.getCurrentTable().deleteNoCheckFromLog(current, data); current.endAction(Result.updateOneResult); break; } case ScriptReaderBase.SET_SCHEMA_STATEMENT : { HsqlName name = database.schemaManager.findSchemaHsqlName( scr.getCurrentSchema()); current.setCurrentSchemaHsqlName(name); break; } case ScriptReaderBase.SESSION_ID : { break; } } if (current.isClosed()) { sessionMap.remove(currentId); } } } catch (Throwable e) { // catch out-of-memory errors and terminate if (e instanceof EOFException) { // end of file - normal end } else if (e instanceof OutOfMemoryError) { database.logger.logSevereEvent("out of memory processing " + logFilename + " line: " + scr.getLineNumber(), e); throw Error.error(ErrorCode.OUT_OF_MEMORY); } else { // stop processing on bad script line database.logger.logSevereEvent(logFilename + " line: " + scr.getLineNumber(), e); } } finally { if (scr != null) { scr.close(); } database.getSessionManager().closeAllSessions(); database.setReferentialIntegrity(true); } } }