All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.quartz.impl.jdbcjobstore.AttributeRestoringConnectionInvocationHandler Maven / Gradle / Ivy

There is a newer version: 2.5.0-rc1
Show newest version
/* 
 * All content copyright Terracotta, Inc., unless otherwise indicated. All rights reserved.
 * 
 * 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 org.quartz.impl.jdbcjobstore;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.sql.Connection;
import java.sql.SQLException;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * 

* Protects a {@link java.sql.Connection}'s attributes from being permanently modfied. *

* *

* Wraps a provided {@link java.sql.Connection} such that its auto * commit and transaction isolation attributes can be overwritten, but * will automatically restored to their original values when the connection * is actually closed (and potentially returned to a pool for reuse). *

* * @see org.quartz.impl.jdbcjobstore.JobStoreSupport#getConnection() * @see org.quartz.impl.jdbcjobstore.JobStoreCMT#getNonManagedTXConnection() */ public class AttributeRestoringConnectionInvocationHandler implements InvocationHandler { private Connection conn; private boolean overwroteOriginalAutoCommitValue; private boolean overwroteOriginalTxIsolationValue; // Set if overwroteOriginalAutoCommitValue is true private boolean originalAutoCommitValue; // Set if overwroteOriginalTxIsolationValue is true private int originalTxIsolationValue; public AttributeRestoringConnectionInvocationHandler( Connection conn) { this.conn = conn; } protected Logger getLog() { return LoggerFactory.getLogger(getClass()); } public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { if (method.getName().equals("setAutoCommit")) { setAutoCommit(((Boolean)args[0]).booleanValue()); } else if (method.getName().equals("setTransactionIsolation")) { setTransactionIsolation(((Integer)args[0]).intValue()); } else if (method.getName().equals("close")) { close(); } else { try { return method.invoke(conn, args); } catch(InvocationTargetException ite) { throw (ite.getCause() != null ? ite.getCause() : ite); } } return null; } /** * Sets this connection's auto-commit mode to the given state, saving * the original mode. The connection's original auto commit mode is restored * when the connection is closed. */ public void setAutoCommit(boolean autoCommit) throws SQLException { boolean currentAutoCommitValue = conn.getAutoCommit(); if (autoCommit != currentAutoCommitValue) { if (overwroteOriginalAutoCommitValue == false) { overwroteOriginalAutoCommitValue = true; originalAutoCommitValue = currentAutoCommitValue; } conn.setAutoCommit(autoCommit); } } /** * Attempts to change the transaction isolation level to the given level, saving * the original level. The connection's original transaction isolation level is * restored when the connection is closed. */ public void setTransactionIsolation(int level) throws SQLException { int currentLevel = conn.getTransactionIsolation(); if (level != currentLevel) { if (overwroteOriginalTxIsolationValue == false) { overwroteOriginalTxIsolationValue = true; originalTxIsolationValue = currentLevel; } conn.setTransactionIsolation(level); } } /** * Gets the underlying connection to which all operations ultimately * defer. This is provided in case a user ever needs to punch through * the wrapper to access vendor specific methods outside of the * standard java.sql.Connection interface. * * @return The underlying connection to which all operations * ultimately defer. */ public Connection getWrappedConnection() { return conn; } /** * Attempts to restore the auto commit and transaction isolation connection * attributes of the wrapped connection to their original values (if they * were overwritten). */ public void restoreOriginalAtributes() { try { if (overwroteOriginalAutoCommitValue) { conn.setAutoCommit(originalAutoCommitValue); } } catch (Throwable t) { getLog().warn("Failed restore connection's original auto commit setting.", t); } try { if (overwroteOriginalTxIsolationValue) { conn.setTransactionIsolation(originalTxIsolationValue); } } catch (Throwable t) { getLog().warn("Failed restore connection's original transaction isolation setting.", t); } } /** * Attempts to restore the auto commit and transaction isolation connection * attributes of the wrapped connection to their original values (if they * were overwritten), before finally actually closing the wrapped connection. */ public void close() throws SQLException { restoreOriginalAtributes(); conn.close(); } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy