diffcases.diffroleinparent2.1205753_EmbedPooledConnection_0_t Maven / Gradle / Ivy
/*
Derby - Class org.apache.derby.jdbc.EmbedPooledConnection
Licensed to the Apache Software Foundation (ASF) under one or more
contributor license agreements. See the NOTICE file distributed with
this work for additional information regarding copyright ownership.
The ASF licenses this file to You 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.apache.derby.jdbc;
import org.apache.derby.iapi.services.sanity.SanityManager;
import org.apache.derby.iapi.reference.Property;
import org.apache.derby.iapi.error.ExceptionSeverity;
import org.apache.derby.iapi.sql.conn.LanguageConnectionContext;
/* import impl class */
import org.apache.derby.impl.jdbc.Util;
import org.apache.derby.impl.jdbc.EmbedConnection;
import org.apache.derby.iapi.jdbc.BrokeredConnection;
import org.apache.derby.iapi.jdbc.BrokeredConnectionControl;
import org.apache.derby.iapi.jdbc.EngineConnection;
import org.apache.derby.impl.jdbc.EmbedPreparedStatement;
import org.apache.derby.impl.jdbc.EmbedCallableStatement;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.sql.PreparedStatement;
import java.sql.CallableStatement;
import java.util.ArrayList;
import java.util.Iterator;
/* -- New jdbc 20 extension types --- */
import javax.sql.ConnectionEventListener;
import javax.sql.ConnectionEvent;
/**
A PooledConnection object is a connection object that provides hooks for
connection pool management.
This is Derby's implementation of a PooledConnection for use in
the following environments:
- JDBC 3.0 - Java 2 - JDK 1.4, J2SE 5.0
- JDBC 2.0 - Java 2 - JDK 1.2,1.3
*/
class EmbedPooledConnection implements javax.sql.PooledConnection, BrokeredConnectionControl
{
/** the connection string */
private String connString;
/**
* The list of {@code ConnectionEventListener}s. It is initially {@code
* null} and will be initialized lazily when the first listener is added.
*/
private ArrayList eventListener;
/**
* The number of iterators going through the list of connection event
* listeners at the current time. Only one thread may be iterating over the
* list at any time (because of synchronization), but a single thread may
* have multiple iterators if for instance an event listener performs
* database calls that trigger a new event.
*/
private int eventIterators;
EmbedConnection realConnection;
int defaultIsolationLevel;
private boolean defaultReadOnly;
BrokeredConnection currentConnectionHandle;
final ReferenceableDataSource dataSource;
private final String username;
private final String password;
/**
True if the password was passed in on the connection request, false if it came from the data source property.
*/
private final boolean requestPassword;
protected boolean isActive;
/**
* getter function for isActive
* @return boolean is isActive is true
**/
public boolean isActive() {
return isActive;
}
EmbedPooledConnection(ReferenceableDataSource ds, String u, String p,
boolean requestPassword) throws SQLException
{
dataSource = ds;
username = u;
password = p;
this.requestPassword = requestPassword;
isActive = true;
openRealConnection();
}
String getUsername()
{
if (username == null || username.equals(""))
return Property.DEFAULT_USER_NAME;
else
return username;
}
String getPassword()
{
if (password == null)
return "";
else
return password;
}
/**
Create an object handle for a database connection.
@return a Connection object
@exception SQLException - if a database-access error occurs.
*/
public synchronized Connection getConnection() throws SQLException
{
checkActive();
if (realConnection == null)
{
openRealConnection();
}
else
{
resetRealConnection();
}
closeCurrentConnectionHandle();
Connection c = getNewCurrentConnectionHandle();
return c;
}
final void openRealConnection() throws SQLException {
Connection rc = dataSource.getConnection(username, password, requestPassword);
this.realConnection = (EmbedConnection) rc;
defaultIsolationLevel = rc.getTransactionIsolation();
defaultReadOnly = rc.isReadOnly();
if (currentConnectionHandle != null)
realConnection.setApplicationConnection(currentConnectionHandle);
}
final Connection getNewCurrentConnectionHandle() {
Connection applicationConnection = currentConnectionHandle =
((org.apache.derby.jdbc.Driver20) (realConnection.getLocalDriver())).newBrokeredConnection(this);
realConnection.setApplicationConnection(applicationConnection);
return applicationConnection;
}
/**
In this case the Listeners are *not* notified. JDBC 3.0 spec section 11.4
*/
private void closeCurrentConnectionHandle() throws SQLException {
if (currentConnectionHandle != null)
{
ArrayList tmpEventListener = eventListener;
eventListener = null;
try {
currentConnectionHandle.close();
} finally {
eventListener = tmpEventListener;
}
currentConnectionHandle = null;
}
}
void resetRealConnection() throws SQLException {
realConnection.rollback();
realConnection.clearWarnings();
if (realConnection.getTransactionIsolation() != defaultIsolationLevel) {
realConnection.setTransactionIsolation(defaultIsolationLevel);
}
if (!realConnection.getAutoCommit())
realConnection.setAutoCommit(true);
if (realConnection.isReadOnly() != defaultReadOnly)
realConnection.setReadOnly(defaultReadOnly);
if (realConnection.getHoldability() != ResultSet.HOLD_CURSORS_OVER_COMMIT)
realConnection.setHoldability(ResultSet.HOLD_CURSORS_OVER_COMMIT);
realConnection.resetFromPool();
if (SanityManager.DEBUG)
{
SanityManager.ASSERT(realConnection.transactionIsIdle(),
"real connection should have been idle at this point");
}
}
/**
Close the Pooled connection.
@exception SQLException - if a database-access error occurs.
*/
public synchronized void close() throws SQLException
{
if (!isActive)
return;
closeCurrentConnectionHandle();
try {
if (realConnection != null) {
if (!realConnection.isClosed())
realConnection.close();
}
} finally {
realConnection = null;
isActive = false;
eventListener = null;
}
}
/**
Add an event listener.
*/
public final synchronized void addConnectionEventListener(ConnectionEventListener listener)
{
if (!isActive)
return;
if (listener == null)
return;
if (eventListener == null) {
eventListener = new ArrayList();
} else if (eventIterators > 0) {
eventListener = (ArrayList) eventListener.clone();
}
eventListener.add(listener);
}
/**
Remove an event listener.
*/
public final synchronized void removeConnectionEventListener(ConnectionEventListener listener)
{
if (listener == null || eventListener == null) {
return;
}
if (eventIterators > 0) {
eventListener = (ArrayList) eventListener.clone();
}
eventListener.remove(listener);
}
/*
* class specific method
*/
public synchronized EngineConnection getRealConnection()
throws SQLException
{
checkActive();
return realConnection;
}
/**
* @return The underlying language connection.
*/
public synchronized LanguageConnectionContext getLanguageConnection()
throws SQLException
{
checkActive();
return realConnection.getLanguageConnection();
}
public synchronized void notifyError(SQLException exception)
{
if (exception.getErrorCode() < ExceptionSeverity.SESSION_SEVERITY)
return;
fireConnectionEventListeners(exception);
}
/**
* Fire all the {@code ConnectionEventListener}s registered. Callers must
* synchronize on {@code this} to prevent others from modifying the list of
* listeners.
*
* @param exception the exception that caused the event, or {@code null} if
* it is a close event
*/
private void fireConnectionEventListeners(SQLException exception) {
if (eventListener != null && !eventListener.isEmpty()) {
ConnectionEvent event = new ConnectionEvent(this, exception);
eventIterators++;
try {
for (Iterator it = eventListener.iterator(); it.hasNext();) {
ConnectionEventListener l =
(ConnectionEventListener) it.next();
if (exception == null) {
l.connectionClosed(event);
} else {
l.connectionErrorOccurred(event);
}
}
} finally {
eventIterators--;
}
}
}
final void checkActive() throws SQLException {
if (!isActive)
throw Util.noCurrentConnection();
}
/*
** BrokeredConnectionControl api
*/
/**
Returns true if isolation level has been set using either JDBC api or SQL
*/
public boolean isIsolationLevelSetUsingSQLorJDBC() throws SQLException {
if (realConnection != null)
return realConnection.getLanguageConnection().isIsolationLevelSetUsingSQLorJDBC();
else
return false;
}
/**
Reset the isolation level flag used to keep state in
BrokeredConnection. It will get set to true when isolation level
is set using JDBC/SQL. It will get reset to false at the start
and the end of a global transaction.
*/
public void resetIsolationLevelFlag() throws SQLException {
realConnection.getLanguageConnection().resetIsolationLevelFlagUsedForSQLandJDBC();
}
/** @see BrokeredConnectionControl#isInGlobalTransaction() */
public boolean isInGlobalTransaction() {
return false;
}
/**
Notify the control class that a SQLException was thrown
during a call on one of the brokered connection's methods.
*/
public void notifyException(SQLException sqle) {
this.notifyError(sqle);
}
/**
Allow control over setting auto commit mode.
*/
public void checkAutoCommit(boolean autoCommit) throws SQLException {
}
/**
Are held cursors allowed.
*/
public int checkHoldCursors(int holdability, boolean downgrade)
throws SQLException
{
return holdability;
}
/**
Allow control over creating a Savepoint (JDBC 3.0)
*/
public void checkSavepoint() throws SQLException {
}
/**
Allow control over calling rollback.
*/
public void checkRollback() throws SQLException {
}
/**
Allow control over calling commit.
*/
public void checkCommit() throws SQLException {
}
/** @see BrokeredConnectionControl#checkClose() */
public void checkClose() throws SQLException {
if (realConnection != null) {
realConnection.checkForTransactionInProgress();
}
}
/**
Close called on BrokeredConnection. If this call
returns true then getRealConnection().close() will be called.
Notify listners that connection is closed.
Don't close the underlying real connection as
it is pooled.
*/
public synchronized boolean closingConnection() throws SQLException {
currentConnectionHandle = null;
fireConnectionEventListeners(null);
return false;
}
/**
No need to wrap statements for PooledConnections.
*/
public Statement wrapStatement(Statement s) throws SQLException {
return s;
}
/**
* Call the setBrokeredConnectionControl method inside the
* EmbedPreparedStatement class to set the BrokeredConnectionControl
* variable to this instance of EmbedPooledConnection
* This will then be used to call the onStatementErrorOccurred
* and onStatementClose events when the corresponding events
* occur on the PreparedStatement
*
* @param ps PreparedStatment to be wrapped
* @param sql String
* @param generatedKeys Object
* @return returns the wrapped PreparedStatement
* @throws java.sql.SQLException
*/
public PreparedStatement wrapStatement(PreparedStatement ps, String sql, Object generatedKeys) throws SQLException {
/*
*/
EmbedPreparedStatement ps_ = (EmbedPreparedStatement)ps;
ps_.setBrokeredConnectionControl(this);
return (PreparedStatement)ps_;
}
/**
* Call the setBrokeredConnectionControl method inside the
* EmbedCallableStatement class to set the BrokeredConnectionControl
* variable to this instance of EmbedPooledConnection
* This will then be used to call the onStatementErrorOccurred
* and onStatementClose events when the corresponding events
* occur on the CallableStatement
*
* @param cs CallableStatment to be wrapped
* @param sql String
* @return returns the wrapped CallableStatement
* @throws java.sql.SQLException
*/
public CallableStatement wrapStatement(CallableStatement cs, String sql) throws SQLException {
EmbedCallableStatement cs_ = (EmbedCallableStatement)cs;
cs_.setBrokeredConnectionControl(this);
return (CallableStatement)cs_;
}
/**
* Get the string representation of this pooled connection.
*
* A pooled connection is assigned a separate id from a physical
* connection. When a container calls PooledConnection.toString(),
* it gets the string representation of this id. This is useful for
* developers implementing connection pools when they are trying to
* debug pooled connections.
*
* @return a string representation of the uniquie id for this pooled
* connection.
*
*/
public String toString()
{
if ( connString == null )
{
String physicalConnString = isActive ?
realConnection.toString() : "";
connString =
this.getClass().getName() + "@" + this.hashCode() + " " +
"Physical Connection = " + physicalConnString;
}
return connString;
}
/*-----------------------------------------------------------------*/
/*
* These methods are from the BrokeredConnectionControl interface.
* These methods are needed to provide StatementEvent support for
* derby.
* They are actually implemented in EmbedPooledConnection40 but have
* a dummy implementation here so that the compilation wont fail when they
* are compiled with jdk1.4
*/
/**
* Dummy implementation for the actual methods found in
* org.apache.derby.jdbc.EmbedPooledConnection40
* @param statement PreparedStatement
*/
public void onStatementClose(PreparedStatement statement) {
}
/**
* Dummy implementation for the actual methods found in
* org.apache.derby.jdbc.EmbedPooledConnection40
* @param statement PreparedStatement
* @param sqle SQLException
*/
public void onStatementErrorOccurred(PreparedStatement statement,
SQLException sqle) {
}
}