package fr.ifremer.quadrige3.core.dao.technical.hibernate;
   
   /*-
    * #%L
    * Quadrige3 Core :: Quadrige3 Core Shared
    * $Id:$
    * $HeadURL:$
    * %%
    * Copyright (C) 2017 Ifremer
   * %%
   * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU Affero General Public License as published by
   * the Free Software Foundation, either version 3 of the License, or
   * (at your option) any later version.
   * 
   * This program is distributed in the hope that it will be useful,
   * but WITHOUT ANY WARRANTY; without even the implied warranty of
   * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   * GNU General Public License for more details.
   * 
   * You should have received a copy of the GNU Affero General Public License
   * along with this program.  If not, see <http://www.gnu.org/licenses/>.
   * #L%
   */
  
  
  import org.hibernate.HibernateException;
  import org.hibernate.engine.jdbc.connections.spi.ConnectionProvider;
  import org.hibernate.exception.spi.Configurable;
  import org.hibernate.service.UnknownUnwrapTypeException;
  import org.hibernate.service.spi.Stoppable;
  
  import javax.sql.DataSource;
  import java.sql.Connection;
  import java.sql.SQLException;
  import java.util.Properties;
  
  /**
   * <p>HibernateConnectionProvider class.</p>
   */
  public class HibernateConnectionProvider implements ConnectionProvider,Configurable,Stoppable {
  
      private static final long serialVersionUID = 6463355546534159477L;
      
  	private static DataSource dataSource = null;
  
      /** {@inheritDoc} */
      @Override
      public void configure(Properties props) throws HibernateException {
          if (dataSource == null) {
              throw new HibernateException("DataSource must be set before using ConnectionProvider.");
          }
      }
  
      /** {@inheritDoc} */
      @Override
      public Connection getConnection() throws SQLException {
          return dataSource.getConnection();
      }
  
      /** {@inheritDoc} */
      @Override
      public void closeConnection(Connection conn) throws SQLException {
          if (conn != null && !conn.isClosed()) {
              conn.close();
          }
      }
  
      /** {@inheritDoc} */
      @Override
      public void stop(){
          // Release datasource
          dataSource = null;
      }
  
      /** {@inheritDoc} */
      @Override
      public boolean supportsAggressiveRelease() {
          // TODO A quoi cela sert il ?
          return false;
      }
  
      /**
       * <p>Getter for the field <code>dataSource</code>.</p>
       *
       * @return a {@link javax.sql.DataSource} object.
       */
      public static DataSource getDataSource() {
          return dataSource;
      }
  
      /**
       * <p>Setter for the field <code>dataSource</code>.</p>
       *
       * @param dataSource a {@link javax.sql.DataSource} object.
       */
      public static void setDataSource(DataSource dataSource) {
          HibernateConnectionProvider.dataSource = dataSource;
      }
 
     /** {@inheritDoc} */
     @SuppressWarnings("rawtypes")
 	@Override
     public boolean isUnwrappableAs(Class unwrapType) {
 		return ConnectionProvider.class.equals(unwrapType) ||
 		HibernateConnectionProvider.class.isAssignableFrom(unwrapType);
     }
 
 	/** {@inheritDoc} */
 	@SuppressWarnings("unchecked")
     @Override
     public <T> T unwrap(Class<T> unwrapType) {
 		if (ConnectionProvider.class.equals(unwrapType) ||
 				HibernateConnectionProvider.class.isAssignableFrom(unwrapType)) {
 			return (T) this;
 		} else {
 			throw new UnknownUnwrapTypeException(unwrapType);
 		}
     }
 
 }