appserver/tests/appserv-tests/cciblackbox-tx/src/main/java/com/sun/connector/cciblackbox/CciManagedConnection.java

/*
 * Copyright (c) 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.connector.cciblackbox;

import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;

import jakarta.resource.NotSupportedException;
import jakarta.resource.ResourceException;
import jakarta.resource.spi.ConnectionEventListener;
import jakarta.resource.spi.ConnectionRequestInfo;
import jakarta.resource.spi.IllegalStateException;
import jakarta.resource.spi.ManagedConnection;
import jakarta.resource.spi.ManagedConnectionFactory;
import jakarta.resource.spi.ManagedConnectionMetaData;
import jakarta.resource.spi.SecurityException;
import jakarta.resource.spi.security.PasswordCredential;
import javax.security.auth.Subject;
import javax.sql.XAConnection;
import javax.transaction.xa.XAResource;

/**
 * This class represents a physical connection to an underlying EIS.
 * @author Sheetal Vartak
 */
public class CciManagedConnection implements ManagedConnection {

  private XAConnection xacon;

  private java.sql.Connection con;

  private CciConnectionEventListener cciListener;

  private PasswordCredential passCred;

  private ManagedConnectionFactory mcf;

  private PrintWriter logWriter;

  private boolean supportsXA;

  private boolean supportsLocalTx;

  private boolean destroyed;

  private Set connectionSet; // set of CciConnection

  CciManagedConnection(ManagedConnectionFactory mcf, PasswordCredential passCred,
      XAConnection xacon, Connection con, boolean supportsXA, boolean supportsLocalTx) {
    this.mcf = mcf;
    this.passCred = passCred;
    this.xacon = xacon;
    this.con = con;
    this.supportsXA = supportsXA;
    this.supportsLocalTx = supportsLocalTx;
    connectionSet = new HashSet();
    cciListener = new CciConnectionEventListener(this);
    if (xacon != null) {
      xacon.addConnectionEventListener(cciListener);
    }

  }

  public void setXAConnection(XAConnection xa) {
    this.xacon = xa;
  }

  public void setConnection(java.sql.Connection con) {
    this.con = con;
  }

  public void setSupportsXA(boolean xa) {
    this.supportsXA = xa;
  }

  public void setSupportsLocalTx(boolean xa) {
    this.supportsLocalTx = xa;
  }

  public void setManagedConnectionFactory(ManagedConnectionFactory xa) {
    this.mcf = xa;
  }

  public XAConnection getXAConnection() {
    return this.xacon;
  }

  public java.sql.Connection getConnection() {
    return this.con;
  }

  public boolean getSupportsXA() {
    return this.supportsXA;
  }

  public boolean getSupportsLocalTx() {
    return this.supportsLocalTx;
  }

  public ManagedConnectionFactory getManagedConnectionFactory() {
    return this.mcf;
  }

  private void throwResourceException(SQLException ex) throws ResourceException {

    ResourceException re = new ResourceException("SQLException: " + ex.getMessage());
    re.setLinkedException(ex);
    throw re;
  }

  public Object getConnection(Subject subject, ConnectionRequestInfo connectionRequestInfo)
      throws ResourceException {

    PasswordCredential pc = Util.getPasswordCredential(mcf, subject, connectionRequestInfo);
    if (!Util.isPasswordCredentialEqual(pc, passCred)) {
      throw new SecurityException("Principal does not match." + "Reauthentication not supported");
    }
    checkIfDestroyed();
    CciConnection cciCon = new CciConnection(this);
    addCciConnection(cciCon);
    return cciCon;
  }

  public void destroy() throws ResourceException {
    try {
      if (destroyed) return;
      destroyed = true;
      Iterator it = connectionSet.iterator();
      while (it.hasNext()) {
        CciConnection cciCon = (CciConnection) it.next();
        cciCon.invalidate();
      }
      connectionSet.clear();
      con.close();
      if (xacon != null) xacon.close();
    }
    catch (SQLException ex) {
      throwResourceException(ex);
    }
  }

  public void cleanup() throws ResourceException {
    try {
      checkIfDestroyed();
      Iterator it = connectionSet.iterator();
      while (it.hasNext()) {
        CciConnection cciCon = (CciConnection) it.next();
        cciCon.invalidate();
      }
      connectionSet.clear();
      if (xacon != null) {
        con.close();
        con = xacon.getConnection();
      } else {
        con.setAutoCommit(true);
      }
    }
    catch (SQLException ex) {
      throwResourceException(ex);
    }
  }

  public void associateConnection(Object connection) throws ResourceException {

    checkIfDestroyed();
    if (connection instanceof CciConnection) {
      CciConnection cciCon = (CciConnection) connection;
      cciCon.associateConnection(this);
    } else {
      throw new IllegalStateException("Invalid connection object: " + connection);
    }
  }

  public void addConnectionEventListener(ConnectionEventListener listener) {
    cciListener.addConnectorListener(listener);
  }

  public void removeConnectionEventListener(ConnectionEventListener listener) {

    cciListener.removeConnectorListener(listener);
  }

  public XAResource getXAResource() throws ResourceException {
    if (!supportsXA) {
      throw new NotSupportedException("XA transaction not supported");
    }
    try {
      checkIfDestroyed();
      return xacon.getXAResource();
    }
    catch (SQLException ex) {
      throwResourceException(ex);
      return null;
    }
  }

  public jakarta.resource.spi.LocalTransaction getLocalTransaction() throws ResourceException {
    if (!supportsLocalTx) {
      throw new NotSupportedException("Local transaction not supported");
    } else {
      checkIfDestroyed();
      return new SpiLocalTransactionImpl(this);
    }
  }

  public ManagedConnectionMetaData getMetaData() throws ResourceException {
    checkIfDestroyed();
    return new CciManagedConnectionMetaDataImpl(this);
  }

  public void setLogWriter(PrintWriter out) throws ResourceException {
    this.logWriter = out;
  }

  public PrintWriter getLogWriter() throws ResourceException {
    return logWriter;
  }

  public Connection getJdbcConnection() throws ResourceException {
    checkIfDestroyed();
    return con;
  }

  boolean isDestroyed() {
    return destroyed;
  }

  PasswordCredential getPasswordCredential() {
    return passCred;
  }

  public void sendEvent(int eventType, Exception ex) {
    cciListener.sendEvent(eventType, ex, null);
  }

  public void sendEvent(int eventType, Exception ex, Object connectionHandle) {
    cciListener.sendEvent(eventType, ex, connectionHandle);
  }

  public void removeCciConnection(CciConnection cciCon) {
    connectionSet.remove(cciCon);
  }

  public void addCciConnection(CciConnection cciCon) {
    connectionSet.add(cciCon);
  }

  private void checkIfDestroyed() throws ResourceException {
    if (destroyed) {
      throw new IllegalStateException("Managed connection is closed");
    }
  }
}