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

com.sun.gjc.spi.ManagedConnectionImpl Maven / Gradle / Ivy

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 1997-2016 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in compliance with the License.  You can
 * obtain a copy of the License at
 * https://glassfish.dev.java.net/public/CDDL+GPL_1_1.html
 * or packager/legal/LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at packager/legal/LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle designates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your decision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */
// Portions Copyright 2016-2022 Payara Foundation and/or its affiliates.

package com.sun.gjc.spi;

import com.sun.appserv.connectors.internal.api.ConnectorsUtil;
import com.sun.appserv.connectors.internal.spi.BadConnectionEventListener;
import com.sun.enterprise.config.serverbeans.Domain;
import com.sun.enterprise.config.serverbeans.ResourcePool;
import com.sun.enterprise.config.serverbeans.Resources;
import com.sun.enterprise.util.i18n.StringManager;
import com.sun.gjc.common.DataSourceObjectBuilder;
import com.sun.gjc.spi.base.*;
import com.sun.gjc.spi.base.datastructure.Cache;
import com.sun.gjc.spi.base.datastructure.CacheFactory;
import com.sun.gjc.util.SQLTraceDelegator;
import com.sun.gjc.util.StatementLeakDetector;
import com.sun.logging.LogDomains;
import fish.payara.jdbc.RequestTracingListener;
import fish.payara.jdbc.SQLTraceStoreAdapter;
import fish.payara.jdbc.SlowSQLLogger;
import fish.payara.nucleus.requesttracing.RequestTracingService;

import static java.lang.Double.parseDouble;

import java.io.PrintWriter;
import java.sql.CallableStatement;
import java.sql.DatabaseMetaData;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.Hashtable;
import java.util.Iterator;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import java.util.logging.Logger;
import jakarta.resource.NotSupportedException;
import jakarta.resource.ResourceException;
import jakarta.resource.spi.ConnectionEvent;
import jakarta.resource.spi.ConnectionEventListener;
import jakarta.resource.spi.security.PasswordCredential;
import javax.security.auth.Subject;
import javax.sql.PooledConnection;
import javax.sql.XAConnection;
import javax.transaction.xa.XAResource;

import org.glassfish.api.StartupRunLevel;
import org.glassfish.api.admin.ProcessEnvironment;
import org.glassfish.hk2.runlevel.RunLevelController;
import org.glassfish.internal.api.Globals;
import org.glassfish.jdbc.config.JdbcConnectionPool;
import org.glassfish.resourcebase.resources.api.PoolInfo;

/**
 * ManagedConnection implementation for Generic JDBC Connector.
 *
 * @author Evani Sai Surya Kiran
 * @version 1.0, 02/07/22
 */
public class ManagedConnectionImpl implements jakarta.resource.spi.ManagedConnection,
        jakarta.resource.spi.LazyEnlistableManagedConnection,
        jakarta.resource.spi.DissociatableManagedConnection {

    public static final int ISNOTAPOOLEDCONNECTION = 0;
    public static final int ISPOOLEDCONNECTION = 1;
    public static final int ISXACONNECTION = 2;

    protected boolean isDestroyed = false;
    protected boolean isUsable = true;
    protected boolean initSqlExecuted = false;
    protected int connectionCount = 0;

    public static final String IS_SLOW_SQL_LOGGING_DISABLED = "-1";

    protected int connectionType = ISNOTAPOOLEDCONNECTION;
    protected PooledConnection pc = null;
    protected java.sql.Connection actualConnection = null;
    protected Hashtable connectionHandles;
    protected PrintWriter logWriter;
    protected PasswordCredential passwdCredential;
    private jakarta.resource.spi.ManagedConnectionFactory mcf = null;
    protected XAResource xar = null;

    protected ConnectionHolder myLogicalConnection = null;

    //GJCINT
    protected int lastTransactionIsolationLevel;
    protected boolean isClean = true;

    protected boolean transactionInProgress = false;

    protected ConnectionEventListener listener = null;

    protected ConnectionEvent ce = null;

    private boolean defaultAutoCommitValue = true;
    private boolean lastAutoCommitValue = defaultAutoCommitValue;

    private boolean markedForRemoval = false;

    //private boolean statemntWrapping;
    private int statementTimeout;

    private Cache statementCache = null;
    private int cacheSize;
    private String cacheType;
    private boolean statementCaching;
    private long stmtLeakTimeout;
    private boolean stmtLeakReclaim;
    private boolean statementLeakTracing;
    protected StatementLeakDetector leakDetector;

    private SQLTraceDelegator sqlTraceDelegator;
    
    protected final static Logger _logger;

    static {
        _logger = LogDomains.getLogger(ManagedConnectionImpl.class, LogDomains.RSR_LOGGER);
    }

    protected static final StringManager localStrings =
            StringManager.getManager(DataSourceObjectBuilder.class);
    
    private boolean aborted = false;

    private DatabaseMetaData cachedDatabaseMetaData = null;
    private Boolean isClientInfoSupported = null;
    
    private JdbcConnectionPool connectionPool;
    private RequestTracingService requestTracing;
    
    /**
     * Constructor for ManagedConnectionImpl. The pooledConn parameter is expected
     * to be null and sqlConn parameter is the actual connection in case where
     * the actual connection is got from a non pooled datasource object. The
     * pooledConn parameter is expected to be non null and sqlConn parameter
     * is expected to be null in the case where the datasource object is a
     * connection pool datasource or an xa datasource.
     *
     * @param pooledConn PooledConnection object in case the
     *                   physical connection is to be obtained from a pooled
     *                   DataSource; null otherwise
     * @param sqlConn    java.sql.Connection object in case the physical
     *                   connection is to be obtained from a non pooled DataSource;
     *                   null otherwise
     * @param passwdCred object conatining the
     *                   user and password for allocating the connection
     * @throws ResourceException if the ManagedConnectionFactory object
     *                           that created this ManagedConnectionImpl object
     *                           is not the same as returned by PasswordCredential
     *                           object passed
     */
    public ManagedConnectionImpl(PooledConnection pooledConn, java.sql.Connection sqlConn,
                             PasswordCredential passwdCred, 
                             jakarta.resource.spi.ManagedConnectionFactory mcf,
                             PoolInfo poolInfo,
                             int statementCacheSize, String statementCacheType,
                             SQLTraceDelegator delegator,
                             long statementLeakTimeout, boolean statementLeakReclaim)
            throws ResourceException {
        if (pooledConn == null && sqlConn == null) {

            String i18nMsg = localStrings.getString(
                    "jdbc.conn_obj_null");
            throw new ResourceException(i18nMsg);
        }

        if (connectionType == ISNOTAPOOLEDCONNECTION) {
            actualConnection = sqlConn;
        }

        pc = pooledConn;
        connectionHandles = new Hashtable();
        passwdCredential = passwdCred;
        sqlTraceDelegator = delegator;

        this.mcf = mcf;
        if (passwdCredential != null &&
                this.mcf.equals(passwdCredential.getManagedConnectionFactory()) == false) {

            String i18nMsg = localStrings.getString(
                    "jdbc.mc_construct_err");
            throw new ResourceException(i18nMsg);
        }
        logWriter = mcf.getLogWriter();
        ce = new ConnectionEvent(this, ConnectionEvent.CONNECTION_CLOSED);
        tuneStatementCaching(poolInfo, statementCacheSize, statementCacheType);
        tuneStatementLeakTracing(poolInfo, statementLeakTimeout, statementLeakReclaim);
        
        connectionPool = getJdbcConnectionPool(mcf);
        try {
            RunLevelController runLevelController = Globals.getDefaultHabitat().getService(RunLevelController.class);
            if (runLevelController != null && runLevelController.getCurrentRunLevel() == StartupRunLevel.VAL) {
                requestTracing = Globals.getDefaultHabitat().getService(RequestTracingService.class);
            }
        } catch (NullPointerException ex) {
            _logger.log(Level.INFO, "Error retrieving Request Tracing service "
                    + "during initialisation of ManagedConnectionImpl - NullPointerException");
        }
    }

    public StatementLeakDetector getLeakDetector() {
        return leakDetector;
    }

    private void executeInitSql(final String initSql) {
        if(_logger.isLoggable(Level.FINE)) {
            _logger.log(Level.FINE, "jdbc.execute_init_sql_start");
        }
        java.sql.PreparedStatement stmt = null;
        
        if (initSql != null && !initSql.equalsIgnoreCase("null") &&
                !initSql.equals("")) {
        try {
                stmt = actualConnection.prepareStatement(initSql);
                if(_logger.isLoggable(Level.FINE)) {
                    _logger.log(Level.FINE, "jdbc.executing_init_sql", initSql);
                }
                stmt.execute();
            } catch (SQLException sqle) {
                _logger.log(Level.WARNING, "jdbc.exc_init_sql_error", initSql);
                initSqlExecuted = false;
            } finally {
                try {
                    if (stmt != null) {
                        stmt.close();
                    }
                } catch (Exception e) {
                    if(_logger.isLoggable(Level.FINE)) {
                        _logger.log(Level.FINE, "jdbc.exc_init_sql_error_stmt_close", e.getMessage());
                    }
                }
            }
            initSqlExecuted = true;
        }
        if(_logger.isLoggable(Level.FINE)) {
            _logger.log(Level.FINE, "jdbc.execute_init_sql_end");
        }
    }

    private void tuneStatementCaching(PoolInfo poolInfo, int statementCacheSize,
            String statementCacheType) {
        cacheSize = statementCacheSize;
        cacheType = statementCacheType;
        if (cacheSize > 0) {
            try {
                statementCache = CacheFactory.getDataStructure(poolInfo, cacheType, cacheSize);
                statementCaching = true;
            } catch (ResourceException ex) {
                _logger.severe(ex.getMessage());
            }
        }
    }

    private void tuneStatementLeakTracing(PoolInfo poolInfo, long statementLeakTimeout,
            boolean statementLeakReclaim) {
        stmtLeakTimeout = statementLeakTimeout;
        stmtLeakReclaim = statementLeakReclaim;

        if(stmtLeakTimeout > 0) {
            ManagedConnectionFactoryImpl spiMCF = (ManagedConnectionFactoryImpl) mcf;
            statementLeakTracing = true;
            if (leakDetector == null) {
                leakDetector = new StatementLeakDetector(poolInfo, statementLeakTracing,
                        stmtLeakTimeout, stmtLeakReclaim,
                        ((com.sun.gjc.spi.ResourceAdapterImpl) spiMCF.getResourceAdapter()).getTimer());
            }
        }
    }
    /**
     * Adds a connection event listener to the ManagedConnectionImpl instance.
     *
     * @param listener ConnectionEventListener
     * @see #removeConnectionEventListener
     */
    @Override
    public void addConnectionEventListener(ConnectionEventListener listener) {
        this.listener = listener;
    }

    /**
     * Used by the container to change the association of an application-level
     * connection handle with a ManagedConnectionImpl instance.
     *
     * @param connection ConnectionHolder to be associated with
     *                   this ManagedConnectionImpl instance
     * @throws ResourceException if the physical connection is no more
     *                           valid or the connection handle passed is null
     */
    @Override
    public void associateConnection(Object connection) throws ResourceException {
        logFine("In associateConnection");
        checkIfValid();
        if (connection == null) {
            String i18nMsg = localStrings.getString(
                    "jdbc.conn_handle_null");
            throw new ResourceException(i18nMsg);
        }
        ConnectionHolder ch = (ConnectionHolder) connection;

        com.sun.gjc.spi.ManagedConnectionImpl mc = ch.getManagedConnection();
        isClean = false;

        ch.associateConnection(actualConnection, this);
        /**
         * The expectation from the above method is that the connection holder
         * replaces the actual sql connection it holds with the sql connection
         * handle being passed in this method call. Also, it replaces the reference
         * to the ManagedConnectionImpl instance with this ManagedConnectionImpl instance.
         * Any previous statements and result sets also need to be removed.
         */


        ch.setActive(true);
        incrementCount();
        //associate the MC to the supplied logical connection similar to associating the logical connection
        //with this MC and actual-connection.
        myLogicalConnection = ch;
        
        //mc will be null in case we are lazily associating 
        if (mc != null) {
            mc.decrementCount();
        }
    }

    /**
     * Application server calls this method to force any cleanup on the
     * ManagedConnectionImpl instance. This method calls the invalidate
     * method on all ConnectionHandles associated with this ManagedConnectionImpl.
     *
     * @throws ResourceException if the physical connection is no more valid
     */
    @Override
    public void cleanup() throws ResourceException {
        logFine("In cleanup");
        //commenting this as cleanup is called after destroy in case of markConnectionAsBad()
        //checkIfValid();

        /**
         * may need to set the autocommit to true for the non-pooled case.
         */
        //GJCINT
        isClean = true;

        ManagedConnectionFactoryImpl spiMCF = (ManagedConnectionFactoryImpl) mcf;
        resetConnectionProperties(spiMCF);
    }

    /**
     * This method removes all the connection handles from the table
     * of connection handles and invalidates all of them so that any
     * operation on those connection handles throws an exception.
     *
     * @throws ResourceException if there is a problem in retrieving
     *                           the connection handles
     */
    protected void invalidateAllConnectionHandles() throws ResourceException {
        Set handles = connectionHandles.keySet();
        Iterator iter = handles.iterator();
        try {
            while (iter.hasNext()) {
                ConnectionHolder ch = (ConnectionHolder) iter.next();
                ch.invalidate();
            }
        } catch (java.util.NoSuchElementException nsee) {
            throw new ResourceException("Could not find the connection handle: " + nsee.getMessage());
        }
        connectionHandles.clear();
    }

    private void clearStatementCache() {
        if (statementCache != null) {
            if(_logger.isLoggable(Level.FINE)) {
                _logger.fine("Closing statements in statement cache");
            }
            statementCache.flushCache();
            statementCache.clearCache();
        }
    }

    /**
     * Destroys the physical connection to the underlying resource manager.
     *
     * @throws ResourceException if there is an error in closing the physical connection
     */
    @Override
    public void destroy() throws ResourceException {
        logFine("In destroy");
        //GJCINT
        if (isDestroyed) {
            return;
        }
        clearStatementCache();
        //Connection could be closed even before statement is closed. Connection
        //close need not call statement close() method.
        //When application uses a statement, leakDetector could be started.At
        //this point, there is a need to clear the statement leak tasks
        if (leakDetector != null) {
            leakDetector.clearAllStatementLeakTasks();
        }

        try {
            if (connectionType == ISXACONNECTION || connectionType == ISPOOLEDCONNECTION) {
                pc.close();
                pc = null;
                actualConnection = null;
            } else {
                actualConnection.close();
                actualConnection = null;
            }
        } catch (SQLException sqle) {
            isDestroyed = true;
            passwdCredential = null;
            connectionHandles = null;
            String i18nMsg = localStrings.getString(
                    "jdbc.error_in_destroy");
            ResourceException re = new ResourceException(
                    i18nMsg + sqle.getMessage(), sqle);
            throw re;
        }
        isDestroyed = true;
        passwdCredential = null;
        connectionHandles = null;
    }

    /**
     * Creates a new connection handle for the underlying physical
     * connection represented by the ManagedConnectionImpl instance.
     *
     * @param sub       Subject parameter needed for authentication
     * @param cxReqInfo ConnectionRequestInfo carries the user
     *                  and password required for getting this connection.
     * @return Connection    the connection handle Object
     * @throws ResourceException if there is an error in allocating the
     *                           physical connection from the pooled connection
     * @throws jakarta.resource.spi.SecurityException
     *                           if there is a mismatch between the
     *                           password credentials or reauthentication is requested
     */
    @Override
    public Object getConnection(Subject sub, jakarta.resource.spi.ConnectionRequestInfo cxReqInfo)
            throws ResourceException {
        logFine("In getConnection");
        checkIfValid();
        /** Appserver any way doesnt bother about re-authentication today. So commenting this out now.
         com.sun.gjc.spi.ConnectionRequestInfo cxRequestInfo = (com.sun.gjc.spi.ConnectionRequestInfo) cxReqInfo;
         PasswordCredential passwdCred = SecurityUtils.getPasswordCredential(this.mcf, sub, cxRequestInfo);

         if(SecurityUtils.isPasswordCredentialEqual(this.passwdCredential, passwdCred) == false) {
         throw new jakarta.resource.spi.SecurityException("Re-authentication not supported");
         }
         **/

        //GJCINT
        getActualConnection();
        ManagedConnectionFactoryImpl spiMCF = (ManagedConnectionFactoryImpl) mcf;

        String statementTimeoutString = spiMCF.getStatementTimeout();
        if (statementTimeoutString != null) {
            int timeoutValue = Integer.parseInt(statementTimeoutString);
            if (timeoutValue >= 0) {
                statementTimeout = timeoutValue;
            }
        }
        
        /**
         * Register a RequestTracingListener if request tracing is enabled,
         * creating a SQLTraceDelegator if one does not already exist.
         * This check is required here to prevent having to
         * recreate the connection pool to accommodate dynamic request tracing
         * enabling/disabling.
         */
        
        if (sqlTraceDelegator == null) {
            if ((requestTracing != null && requestTracing.isRequestTracingEnabled())
                    || (isSlowQueryLoggingEnabled())) {
                sqlTraceDelegator = new SQLTraceDelegator(spiMCF.getPoolName(),
                        spiMCF.getApplicationName(), spiMCF.getModuleName());
            }
        }

        if (sqlTraceDelegator != null) {
            if (requestTracing != null && requestTracing.isRequestTracingEnabled()) {
                // This method will only register a request tracing listener 
                // if one doesn't already exist
                sqlTraceDelegator.registerSQLTraceListener(new RequestTracingListener());
            } else {
                /**
                 * If request tracing is not enabled, but there is a SQL trace
                 * delegator, deregister the request tracing listener if one is
                 * registered.
                 */
                sqlTraceDelegator.deregisterSQLTraceListener(RequestTracingListener.class);
            }

            if (!isSlowQueryLoggingEnabled()) {
                sqlTraceDelegator.deregisterSQLTraceListener(SlowSQLLogger.class);
                sqlTraceDelegator.deregisterSQLTraceListener(SQLTraceStoreAdapter.class);
            } else {
                sqlTraceDelegator.registerSQLTraceListener(new SlowSQLLogger(getSlowQueryThresholdInMillis(), TimeUnit.MILLISECONDS));
                sqlTraceDelegator.registerSQLTraceListener(new SQLTraceStoreAdapter());
            }
            /**
             * If there are no longer any listeners registered, set the
             * delegator to null to prevent
             * JdbcObjectsFactory().getConnection(...) from using a profiled
             * wrapper unnecessarily.
             */
            if (!sqlTraceDelegator.listenersRegistered()) {
                sqlTraceDelegator = null;
            }
        }

        myLogicalConnection = spiMCF.getJdbcObjectsFactory().getConnection(
                actualConnection, this, cxReqInfo, spiMCF.isStatementWrappingEnabled(),
                sqlTraceDelegator);
        
        //TODO : need to see if this should be executed for every getConnection
        if (!initSqlExecuted) {
            //Check if Initsql is set and execute it
            String initSql = spiMCF.getInitSql();
            executeInitSql(initSql);
        }
        incrementCount();
        isClean = false;

        myLogicalConnection.setActive(true);

        return myLogicalConnection;
    }

    /**
     * Resett connection properties as connections are pooled by application server
* * @param spiMCF * @throws ResourceException */ private void resetConnectionProperties(ManagedConnectionFactoryImpl spiMCF) throws ResourceException { if (isClean) { // If the ManagedConnection is clean, reset the transaction isolation level to what // it was when it was when this ManagedConnection was cleaned up depending on the // ConnectionRequestInfo passed. spiMCF.resetIsolation(this, lastTransactionIsolationLevel); // reset the autocommit value of the connection if application has modified it. resetAutoCommit(); } } /** * To reset AutoCommit of actual connection. * If the last-auto-commit value is different from default-auto-commit value, reset will happen. * If there is a transaction in progress (because of connection sharing), reset will not happen. * * @throws ResourceException */ private void resetAutoCommit() throws ResourceException { if (defaultAutoCommitValue != getLastAutoCommitValue() && !(isTransactionInProgress())) { try { actualConnection.setAutoCommit(defaultAutoCommitValue); } catch (SQLException sqle) { String i18nMsg = localStrings.getString( "jdbc.error_during_setAutoCommit"); throw new ResourceException(i18nMsg + sqle.getMessage(), sqle); } setLastAutoCommitValue(defaultAutoCommitValue); } } /** * Returns an LocalTransactionImpl instance. The LocalTransactionImpl interface * is used by the container to manage local transactions for a RM instance. * * @return LocalTransactionImpl instance * @throws ResourceException if the physical connection is not valid */ @Override public jakarta.resource.spi.LocalTransaction getLocalTransaction() throws ResourceException { logFine("In getLocalTransaction"); checkIfValid(); return new com.sun.gjc.spi.LocalTransactionImpl(this); } /** * Gets the log writer for this ManagedConnectionImpl instance. * * @return PrintWriter instance associated with this * ManagedConnectionImpl instance * @throws ResourceException if the physical connection is not valid * @see setLogWriter */ @Override public PrintWriter getLogWriter() throws ResourceException { logFine("In getLogWriter"); checkIfValid(); return logWriter; } /** * Gets the metadata information for this connection's underlying EIS * resource manager instance. * * @return ManagedConnectionMetaData instance * @throws ResourceException if the physical connection is not valid */ @Override public jakarta.resource.spi.ManagedConnectionMetaData getMetaData() throws ResourceException { logFine("In getMetaData"); checkIfValid(); return new com.sun.gjc.spi.ManagedConnectionMetaDataImpl(this); } /** * Returns an XAResource instance. * * @return XAResource instance * @throws ResourceException if the physical connection is not valid or * there is an error in allocating the * XAResource instance * @throws NotSupportedException if underlying datasource is not an * XADataSource */ @Override public XAResource getXAResource() throws ResourceException { logFine("In getXAResource"); checkIfValid(); if (connectionType == ISXACONNECTION) { try { if (xar == null) { /** * Using the wrapper XAResource. */ xar = new com.sun.gjc.spi.XAResourceImpl(((XAConnection) pc).getXAResource(), this); } return xar; } catch (SQLException sqle) { throw new ResourceException(sqle.getMessage(), sqle); } } else { throw new NotSupportedException("Cannot get an XAResource from a non XA connection"); } } /** * Removes an already registered connection event listener from the * ManagedConnectionImpl instance. * * @param listener ConnectionEventListener to be removed * @see #addConnectionEventListener */ @Override public void removeConnectionEventListener(ConnectionEventListener listener) { } /** * This method is called from XAResource wrapper object * when its XAResource.start() has been called or from * LocalTransactionImpl object when its begin() method is called. */ void transactionStarted() { transactionInProgress = true; } /** * This method is called from XAResource wrapper object * when its XAResource.end() has been called or from * LocalTransactionImpl object when its end() method is called. */ void transactionCompleted() { try { transactionInProgress = false; if (connectionType == ISPOOLEDCONNECTION || connectionType == ISXACONNECTION) { if (connectionCount <= 0) { try { actualConnection.close(); actualConnection = null; } catch (SQLException sqle) { actualConnection = null; } } } } catch (java.lang.NullPointerException e) { if(_logger.isLoggable(Level.FINE)) { _logger.log(Level.FINE, "jdbc.duplicateTxCompleted"); } } if (markedForRemoval) { if (aborted) { com.sun.appserv.connectors.internal.spi.BadConnectionEventListener bcel = (BadConnectionEventListener) listener; bcel.connectionAbortOccurred(ce); _logger.log(Level.INFO, "jdbc.markedForRemoval_conAborted"); markedForRemoval = false; myLogicalConnection.setClosed(true); } else { connectionErrorOccurred(null, null); _logger.log(Level.INFO, "jdbc.markedForRemoval_txCompleted"); markedForRemoval = false; } } isClean = true; } /** * Checks if a this ManagedConnection is involved in a transaction * or not. */ public boolean isTransactionInProgress() { return transactionInProgress; } /** * Sets the log writer for this ManagedConnectionImpl instance. * * @param out PrintWriter to be associated with this * ManagedConnectionImpl instance * @throws ResourceException if the physical connection is not valid * @see getLogWriter */ @Override public void setLogWriter(PrintWriter out) throws ResourceException { checkIfValid(); logWriter = out; } /** * This method determines the type of the connection being held * in this ManagedConnectionImpl. * * @param pooledConn PooledConnection * @return connection type */ protected int getConnectionType(PooledConnection pooledConn) { if (pooledConn == null) { return ISNOTAPOOLEDCONNECTION; } else if (pooledConn instanceof XAConnection) { return ISXACONNECTION; } else { return ISPOOLEDCONNECTION; } } /** * Returns the ManagedConnectionFactory instance that * created this ManagedConnection instance. * * @return ManagedConnectionFactory instance that created this * ManagedConnection instance */ public com.sun.gjc.spi.ManagedConnectionFactoryImpl getManagedConnectionFactory() { return (com.sun.gjc.spi.ManagedConnectionFactoryImpl) mcf; } /** * Returns the actual sql connection for this ManagedConnection. * * @return the physical java.sql.Connection */ //GJCINT java.sql.Connection getActualConnection() throws ResourceException { //GJCINT if (connectionType == ISXACONNECTION || connectionType == ISPOOLEDCONNECTION) { try { if (actualConnection == null) { actualConnection = pc.getConnection(); //re-initialize lastAutoCommitValue such that resetAutoCommit() wont // affect autoCommit of actualConnection setLastAutoCommitValue(defaultAutoCommitValue); } } catch (SQLException sqle) { throw new ResourceException(sqle.getMessage(), sqle); } } return actualConnection; } /** * Returns the PasswordCredential object associated with this ManagedConnection. * * @return PasswordCredential associated with this * ManagedConnection instance */ PasswordCredential getPasswordCredential() { return passwdCredential; } /** * Checks if this ManagedConnection is valid or not and throws an * exception if it is not valid. A ManagedConnection is not valid if * destroy has not been called and no physical connection error has * occurred rendering the physical connection unusable. * * @throws ResourceException if destroy has been called on this * ManagedConnection instance or if a * physical connection error occurred rendering it unusable */ //GJCINT void checkIfValid() throws ResourceException { if (isDestroyed || !isUsable) { String i18nMsg = localStrings.getString( "jdbc.mc_not_usable"); throw new ResourceException(i18nMsg); } } /** * This method is called by the ConnectionHolder when its close method is * called. This ManagedConnection instance invalidates the connection handle * and sends a CONNECTION_CLOSED event to all the registered event listeners. * * @param e Exception that may have occured while closing the connection handle * @param connHolderObject ConnectionHolder that has been closed * @throws SQLException in case closing the sql connection got out of * getConnection on the underlying * PooledConnection throws an exception */ public void connectionClosed(Exception e, ConnectionHolder connHolderObject) throws SQLException { connHolderObject.invalidate(); decrementCount(); ce.setConnectionHandle(connHolderObject); if (markedForRemoval && !transactionInProgress) { com.sun.appserv.connectors.internal.spi.BadConnectionEventListener bcel = (BadConnectionEventListener) listener; bcel.badConnectionClosed(ce); _logger.log(Level.INFO, "jdbc.markedForRemoval_conClosed"); markedForRemoval = false; } else { listener.connectionClosed(ce); } } /** * This method is called by the ConnectionHolder when it detects a connecion * related error. * * @param e Exception that has occurred during an operation on the physical connection * @param connHolderObject ConnectionHolder that detected the physical * connection error */ void connectionErrorOccurred(Exception e, ConnectionHolder connHolderObject) { ConnectionEventListener cel = this.listener; ConnectionEvent ce = null; ce = e == null ? new ConnectionEvent(this, ConnectionEvent.CONNECTION_ERROR_OCCURRED) : new ConnectionEvent(this, ConnectionEvent.CONNECTION_ERROR_OCCURRED, e); if (connHolderObject != null) { ce.setConnectionHandle(connHolderObject); } cel.connectionErrorOccurred(ce); isUsable = false; } /** * This method is called by the XAResource object when its start method * has been invoked. */ void XAStartOccurred() { try { actualConnection.setAutoCommit(false); } catch (Exception e) { _logger.log(Level.WARNING, "XA Start [ setAutoCommit ] failure ", e); connectionErrorOccurred(e, null); } } /** * This method is called by the XAResource object when its end method * has been invoked. */ void XAEndOccurred() { try { actualConnection.setAutoCommit(true); } catch (Exception e) { _logger.log(Level.WARNING, "XA End [ setAutoCommit ] failure ", e); connectionErrorOccurred(e, null); } } /** * This method is called by a Connection Handle to check if it is * the active Connection Handle. If it is not the active Connection * Handle, this method throws an SQLException. Else, it * returns setting the active Connection Handle to the calling * Connection Handle object to this object if the active Connection * Handle is null. * * @param ch ConnectionHolder that requests this * ManagedConnection instance whether * it can be active or not * @throws SQLException in case the physical is not valid or * there is already an active connection handle */ public void checkIfActive(ConnectionHolder ch) throws SQLException { if (isDestroyed || !isUsable) { String i18nMsg = localStrings.getString( "jdbc.conn_not_usable"); throw new SQLException(i18nMsg); } } /** * sets the connection type of this connection. This method is called * by the MCF while creating this ManagedConnection. Saves us a costly * instanceof operation in the getConnectionType */ public void initializeConnectionType(int _connectionType) { connectionType = _connectionType; } public void incrementCount() { connectionCount++; } public void decrementCount() { connectionCount--; } @Override public void dissociateConnections() { if(myLogicalConnection != null){ myLogicalConnection.dissociateConnection(); myLogicalConnection = null; } } public boolean getLastAutoCommitValue() { return lastAutoCommitValue; } /** * To keep track of last auto commit value. * Helps to reset the auto-commit-value while giving new connection-handle. * * @param lastAutoCommitValue */ public void setLastAutoCommitValue(boolean lastAutoCommitValue) { this.lastAutoCommitValue = lastAutoCommitValue; } public void markForRemoval(boolean flag) { markedForRemoval = flag; } public jakarta.resource.spi.ManagedConnectionFactory getMcf() { return mcf; } public int getStatementTimeout() { return statementTimeout; } public void setLastTransactionIsolationLevel(int isolationLevel) { lastTransactionIsolationLevel = isolationLevel; } /** * Returns the cached DatabaseMetaData. * * @return DatabaseMetaData */ public DatabaseMetaData getCachedDatabaseMetaData() throws ResourceException { if(cachedDatabaseMetaData == null){ try { cachedDatabaseMetaData = getActualConnection().getMetaData(); } catch (SQLException sqle) { throw new ResourceException(sqle.getMessage(), sqle); } } return cachedDatabaseMetaData; } public Boolean isClientInfoSupported() { return isClientInfoSupported; } public void setClientInfoSupported(Boolean isClientInfoSupported) { this.isClientInfoSupported = isClientInfoSupported; } private void logFine(String logMessage){ if(_logger.isLoggable(Level.FINE)) { _logger.log(Level.FINE, logMessage); } } public PreparedStatement prepareCachedStatement( ConnectionWrapper conWrapper, String sql, int resultSetType, int resultSetConcurrency) throws SQLException { if (statementCaching) { CacheObjectKey key = new CacheObjectKey(sql, CacheObjectKey.PREPARED_STATEMENT, resultSetType, resultSetConcurrency); //TODO-SC should a null check be done for statementCache? //TODO-SC refactor this method. PreparedStatementWrapper ps = (PreparedStatementWrapper) statementCache.checkAndUpdateCache(key); //TODO-SC-DEFER can the usability (isFree()) check be done by the cache itself and make sure that only a free stmt is returned if (ps != null) { if (isFree(ps)) { //Find if this ps is a valid one. If invalid, remove it //from the cache and prepare a new stmt & add it to cache if(!ps.isValid()) { statementCache.purge(ps); ps = conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, true); ps.setBusy(true); statementCache.addToCache(key, ps, false); } else { //Valid ps ps.setBusy(true); } } else { return conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, false); } } else { ps = conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, true); ps.setBusy(true); statementCache.addToCache(key, ps, false); } return ps; } else return conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, false); } public PreparedStatement prepareCachedStatement( ConnectionWrapper conWrapper, String sql, int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException { if (statementCaching) { CacheObjectKey key = new CacheObjectKey(sql, CacheObjectKey.PREPARED_STATEMENT, resultSetType, resultSetConcurrency, resultSetHoldability); //TODO-SC should a null check be done for statementCache? PreparedStatementWrapper ps = (PreparedStatementWrapper) statementCache.checkAndUpdateCache(key); //TODO-SC-DEFER can the usability (isFree()) check be done by the cache itself and make sure that only a free stmt is returned if (ps != null) { if (isFree(ps)) { //Find if this ps is a valid one. If invalid, remove it //from the cache and prepare a new stmt & add it to cache if(!ps.isValid()) { statementCache.purge(ps); ps = conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, true); ps.setBusy(true); statementCache.addToCache(key, ps, false); } else { //Valid ps ps.setBusy(true); } } else { return conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, false); } } else { ps = conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, true); statementCache.addToCache(key, ps, false); ps.setBusy(true); } return ps; } else return conWrapper.prepareCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, false); } public PreparedStatement prepareCachedStatement( ConnectionWrapper conWrapper, String sql, String[] columnNames) throws SQLException { if (statementCaching) { CacheObjectKey key = new CacheObjectKey(sql, CacheObjectKey.PREPARED_STATEMENT, columnNames); //TODO-SC should a null check be done for statementCache? PreparedStatementWrapper ps = (PreparedStatementWrapper) statementCache.checkAndUpdateCache(key); //TODO-SC-DEFER can the usability (isFree()) check be done by the cache itself and make sure that only a free stmt is returned if (ps != null) { if (isFree(ps)) { //Find if this ps is a valid one. If invalid, remove it //from the cache and prepare a new stmt & add it to cache if(!ps.isValid()) { statementCache.purge(ps); ps = conWrapper.prepareCachedStatement(sql, columnNames, true); ps.setBusy(true); statementCache.addToCache(key, ps, false); } else { //Valid ps ps.setBusy(true); } } else { return conWrapper.prepareCachedStatement(sql, columnNames, false); } } else { ps = conWrapper.prepareCachedStatement(sql, columnNames, true); statementCache.addToCache(key, ps, false); ps.setBusy(true); } return ps; } else return conWrapper.prepareCachedStatement(sql, columnNames, false); } public PreparedStatement prepareCachedStatement( ConnectionWrapper conWrapper, String sql, int[] columnIndexes) throws SQLException { if (statementCaching) { CacheObjectKey key = new CacheObjectKey(sql, CacheObjectKey.PREPARED_STATEMENT, columnIndexes); //TODO-SC should a null check be done for statementCache? PreparedStatementWrapper ps = (PreparedStatementWrapper) statementCache.checkAndUpdateCache(key); //TODO-SC-DEFER can the usability (isFree()) check be done by the cache itself and make sure that only a free stmt is returned if (ps != null) { if (isFree(ps)) { //Find if this ps is a valid one. If invalid, remove it //from the cache and prepare a new stmt & add it to cache if(!ps.isValid()) { statementCache.purge(ps); ps = conWrapper.prepareCachedStatement(sql, columnIndexes, true); ps.setBusy(true); statementCache.addToCache(key, ps, false); } else { //Valid ps ps.setBusy(true); } } else { return conWrapper.prepareCachedStatement(sql, columnIndexes, false); } } else { ps = conWrapper.prepareCachedStatement(sql, columnIndexes, true); statementCache.addToCache(key, ps, false); ps.setBusy(true); } return ps; } else return conWrapper.prepareCachedStatement(sql, columnIndexes, false); } public PreparedStatement prepareCachedStatement( ConnectionWrapper conWrapper, String sql, int autoGeneratedKeys) throws SQLException { if (statementCaching) { CacheObjectKey key = new CacheObjectKey(sql, CacheObjectKey.PREPARED_STATEMENT, autoGeneratedKeys); //TODO-SC should a null check be done for statementCache? PreparedStatementWrapper ps = (PreparedStatementWrapper) statementCache.checkAndUpdateCache(key); //TODO-SC-DEFER can the usability (isFree()) check be done by the cache itself and make sure that only a free stmt is returned if (ps != null) { if (isFree(ps)) { //Find if this ps is a valid one. If invalid, remove it //from the cache and prepare a new stmt & add it to cache if(!ps.isValid()) { statementCache.purge(ps); ps = conWrapper.prepareCachedStatement(sql, autoGeneratedKeys, true); ps.setBusy(true); statementCache.addToCache(key, ps, false); } else { //Valid ps ps.setBusy(true); } } else { return conWrapper.prepareCachedStatement(sql, autoGeneratedKeys, false); } } else { ps = conWrapper.prepareCachedStatement(sql, autoGeneratedKeys, true); statementCache.addToCache(key, ps, false); ps.setBusy(true); } return ps; } else return conWrapper.prepareCachedStatement(sql, autoGeneratedKeys, false); } public CallableStatement prepareCachedCallableStatement( ConnectionWrapper conWrapper, String sql, int resultSetType, int resultSetConcurrency) throws SQLException { if (statementCaching) { //Adding the sql as well as the Statement type "CS" to the CacheObjectKey object CacheObjectKey key = new CacheObjectKey(sql, CacheObjectKey.CALLABLE_STATEMENT, resultSetType, resultSetConcurrency); CallableStatementWrapper cs = (CallableStatementWrapper) statementCache.checkAndUpdateCache(key); //TODO-SC-DEFER can the usability (isFree()) check be done by the cache //itself and make sure that only a free stmt is returned if (cs != null) { if (isFree(cs)) { //Find if this cs is a valid one. If invalid, remove it //from the cache and prepare a new stmt & add it to cache if(!cs.isValid()) { statementCache.purge(cs); cs = conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, true); cs.setBusy(true); statementCache.addToCache(key, cs, false); } else { //Valid ps cs.setBusy(true); } } else { return conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, false); } } else { cs = conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, true); statementCache.addToCache(key, cs, false); cs.setBusy(true); } return cs; } else return conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, false); } public CallableStatement prepareCachedCallableStatement( ConnectionWrapper conWrapper, String sql, int resultSetType, int resultSetConcurrency, int resultSetHoldability) throws SQLException { if (statementCaching) { //Adding the sql as well as the Statement type "CS" to the CacheObjectKey object CacheObjectKey key = new CacheObjectKey(sql, CacheObjectKey.CALLABLE_STATEMENT, resultSetType, resultSetConcurrency, resultSetHoldability); CallableStatementWrapper cs = (CallableStatementWrapper) statementCache.checkAndUpdateCache(key); //TODO-SC-DEFER can the usability (isFree()) check be done by the cache //itself and make sure that only a free stmt is returned if (cs != null) { if (isFree(cs)) { //Find if this cs is a valid one. If invalid, remove it //from the cache and prepare a new stmt & add it to cache if(!cs.isValid()) { statementCache.purge(cs); cs = conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, true); cs.setBusy(true); statementCache.addToCache(key, cs, false); } else { //Valid ps cs.setBusy(true); } } else { return conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, false); } } else { cs = conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, true); statementCache.addToCache(key, cs, false); cs.setBusy(true); } return cs; } else return conWrapper.callableCachedStatement(sql, resultSetType, resultSetConcurrency, resultSetHoldability, false); } boolean isFree(PreparedStatementWrapper cachedps) { return !cachedps.isBusy(); } public void setAborted(boolean flag) { aborted = flag; } public boolean isAborted() { return aborted; } public void purgeStatementFromCache(PreparedStatement preparedStatement) { //TODO isValid check for preparedStatement? statementCache.purge(preparedStatement); } private static JdbcConnectionPool getJdbcConnectionPool(jakarta.resource.spi.ManagedConnectionFactory mcf) { if(Globals.getDefaultHabitat().getService(ProcessEnvironment.class).getProcessType() != ProcessEnvironment.ProcessType.Server) { // this is only applicatble in the server environment, // otherwise we bave no domain to draw upon return null; } ManagedConnectionFactoryImpl spiMCF = (ManagedConnectionFactoryImpl) mcf; Resources resources = Globals.getDefaultHabitat().getService(Domain.class).getResources(); ResourcePool pool = (ResourcePool) ConnectorsUtil.getResourceByName(resources, ResourcePool.class, spiMCF.getPoolName()); if (pool instanceof JdbcConnectionPool) { return (JdbcConnectionPool) pool; } return null; } private boolean isSlowQueryLoggingEnabled() { return getSlowQueryThresholdInMillis() > 0; } private long getSlowQueryThresholdInMillis() { if (connectionPool == null) { return -1; } String thresholdInSeconds = connectionPool.getSlowQueryThresholdInSeconds(); if (thresholdInSeconds == null || thresholdInSeconds.isEmpty()) { return -1; } return Math.round(parseDouble(thresholdInSeconds) * 1000); } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy