package fr.ifremer.quadrige2.synchro.dao;
   
   /*-
    * #%L
    * Quadrige2 Core :: Quadrige2 Client Core
    * $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 com.google.common.collect.ArrayListMultimap;
  import com.google.common.collect.Multimap;
  import fr.ifremer.quadrige2.core.exception.Quadrige2TechnicalException;
  import fr.ifremer.quadrige2.core.config.Quadrige2Configuration;
  import fr.ifremer.quadrige2.core.dao.technical.Daos;
  import fr.ifremer.quadrige2.core.dao.technical.jdbc.OptionalDataSourceJdbcDaoSupport;
  import org.apache.commons.collections4.CollectionUtils;
  import org.apache.commons.logging.Log;
  import org.apache.commons.logging.LogFactory;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.context.annotation.Lazy;
  import org.springframework.stereotype.Repository;
  
  import javax.annotation.Resource;
  import javax.sql.DataSource;
  import java.sql.Connection;
  import java.sql.DatabaseMetaData;
  import java.sql.ResultSet;
  import java.sql.SQLException;
  import java.util.Properties;
  import java.util.Set;
  
  /**
   * Created by blavenie on 20/08/15.
   */
  @Repository("synchroClientDao")
  @Lazy
  public class SynchroClientDaoImpl extends OptionalDataSourceJdbcDaoSupport implements SynchroClientDao {
  
  	/** Logger. */
  	private static final Log log =
  			LogFactory.getLog(SynchroClientDaoImpl.class);
  
  	@Resource
  	private Quadrige2Configuration config;
  
  	/**
  	 * Constructor used by Spring
  	 * 
  	 * @param dataSource
  	 *            a {@link javax.sql.DataSource} object.
  	 */
  	@Autowired
  	public SynchroClientDaoImpl(DataSource dataSource) {
  		super(dataSource);
  	}
  
  	/**
  	 * Constructor to use when Spring not started
  	 */
  	public SynchroClientDaoImpl() {
  		super();
  		config = Quadrige2Configuration.getInstance();
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	public boolean isAllTablesEmpty(Set<String> tableNames) {
  		return isAllTablesEmpty(null, tableNames);
  	}
  
  	/** {@inheritDoc} */
  	@Override
  	public boolean isAllTablesEmpty(Properties connectionProperties, Set<String> tableNames) {
  		if (CollectionUtils.isEmpty(tableNames)) {
  			return true;
  		}
  
  		Connection connection = null;
  		try {
  			connection = createConnection(connectionProperties);
  
  			for (String tableName : tableNames) {
  				long count = Daos.countTableRows(connection, tableName);
 				if (count > 0) {
 					return false;
 				}
 			}
 
 			return true;
 		} catch (SQLException e) {
 			throw new Quadrige2TechnicalException("Error while getting if all tables are empty");
 		} finally {
 			closeSilently(connection);
 		}
 	}
 
 	/** {@inheritDoc} */
 	@Override
 	public Multimap<String, String> getExportedKeys(String pkTableName, String pkColumnName) {
 
 		// Execute query and fetch into result list
 		Multimap<String, String> result = ArrayListMultimap.create();
 
 		// Exported keys (primary keys referenced by another table)
 		Connection connection = null;
 		ResultSet rs = null;
 		try {
 			connection = createConnection(null);
 			DatabaseMetaData dbMeta = connection.getMetaData();
 
 			rs = dbMeta.getExportedKeys(config.getJdbcCatalog(), config.getJdbcSchema(), pkTableName);
 			while (rs.next()) {
 				String columnName = rs.getString("PKCOLUMN_NAME").toLowerCase();
 				if (columnName.equalsIgnoreCase(pkColumnName)) {
 					String fkTableName = rs.getString("FKTABLE_NAME").toUpperCase();
 					String fkColumnName = rs.getString("FKCOLUMN_NAME").toLowerCase();
 					result.put(fkTableName, fkColumnName);
 				}
 			}
 		} catch (SQLException e) {
 			throw new Quadrige2TechnicalException(String.format("Error while getting exported keys for table [%s]", pkTableName));
 		} finally {
 			fr.ifremer.common.synchro.dao.Daos.closeSilently(rs);
 		}
 
 		return result;
 	}
 
 	/* -- Internal methods -- */
 
 }