net.snowflake.client.pooling.SnowflakePooledConnection Maven / Gradle / Ivy

/*
 * Copyright (c) 2012-2019 Snowflake Computing Inc. All rights reserved.
 */
package net.snowflake.client.pooling;

import javax.sql.ConnectionEvent;
import javax.sql.ConnectionEventListener;
import javax.sql.PooledConnection;
import javax.sql.StatementEventListener;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.HashSet;
import java.util.Set;

/**
 * Snowflake implementation of pooled connection
 */
public class SnowflakePooledConnection implements PooledConnection
{
  /**
   * physical connection, an instance of SnowflakeConnectionV1 class
   */
  private Connection physicalConnection;

  /**
   * list of event listener registered to listen for connection event
   */
  private final Set eventListeners;

  SnowflakePooledConnection(Connection physicalConnection)
  {
    this.physicalConnection = physicalConnection;
    this.eventListeners = new HashSet<>();
  }

  @Override
  public Connection getConnection() throws SQLException
  {
    return new LogicalConnection(this);
  }

  Connection getPhysicalConnection()
  {
    return physicalConnection;
  }

  /**
   * Fire a connection has been closed event to event listener
   */
  void fireConnectionCloseEvent()
  {
    for (ConnectionEventListener connectionEventListener : eventListeners)
    {
      connectionEventListener.connectionClosed(new ConnectionEvent(this));
    }
  }

  void fireConnectionErrorEvent(SQLException e)
  {
    for (ConnectionEventListener connectionEventListener : eventListeners)
    {
      connectionEventListener.connectionErrorOccurred(
          new ConnectionEvent(this, e));
    }
  }

  @Override
  public void addConnectionEventListener(ConnectionEventListener eventListener)
  {
    this.eventListeners.add(eventListener);
  }

  @Override
  public void close() throws SQLException
  {
    if (this.physicalConnection != null)
    {
      this.physicalConnection.close();
      this.physicalConnection = null;
    }

    eventListeners.clear();
  }

  @Override
  public void removeConnectionEventListener(ConnectionEventListener eventListener)
  {
    this.eventListeners.remove(eventListener);
  }

  @Override
  public void addStatementEventListener(StatementEventListener eventListener)
  {
    // do nothing for now
  }

  @Override
  public void removeStatementEventListener(StatementEventListener eventListener)
  {
    // do nothing for now
  }
}