package net.sumaris.core.extraction.dao.technical;
   
   /*-
    * #%L
    * Quadrige3 Core :: Client API
    * %%
    * Copyright (C) 2017 - 2018 Ifremer
    * %%
    * This program is free software: you can redistribute it and/or modify
   * it under the terms of the GNU 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 General Public
   * License along with this program.  If not, see
   * <http://www.gnu.org/licenses/gpl-3.0.html>.
   * #L%
   */
  
  import net.sumaris.core.config.SumarisConfiguration;
  import net.sumaris.core.dao.technical.hibernate.HibernateDaoSupport;
  import net.sumaris.core.dao.technical.schema.SumarisDatabaseMetadata;
  import net.sumaris.core.exception.SumarisTechnicalException;
  import net.sumaris.core.model.referential.IItemReferentialEntity;
  import net.sumaris.core.service.ServiceLocator;
  import net.sumaris.core.service.referential.ReferentialService;
  import org.apache.commons.collections4.CollectionUtils;
  import org.jdom2.Element;
  import org.jdom2.filter.Filters;
  import org.jdom2.xpath.XPathFactory;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.beans.factory.annotation.Autowired;
  import org.springframework.context.ApplicationContext;
  import org.springframework.dao.DataRetrievalFailureException;
  
  import javax.persistence.Query;
  import java.util.List;
  import java.util.Objects;
  import java.util.Set;
  import java.util.function.Function;
  import java.util.stream.Collectors;
  import java.util.stream.Stream;
  
  /**
   * @author Benoit Lavenier <benoit.lavenier@e-is.pro>
   */
  public abstract class ExtractionBaseDaoImpl extends HibernateDaoSupport {
  
      private static final Logger log = LoggerFactory.getLogger(ExtractionBaseDaoImpl.class);
  
      protected static final String XML_QUERY_PATH = "xmlQuery";
  
      @Autowired
      protected SumarisConfiguration configuration;
  
      @Autowired
      protected ReferentialService referentialService;
  
      @Autowired
      ApplicationContext applicationContext;
  
      @Autowired
      public ExtractionBaseDaoImpl() {
          super();
      }
  
      @SuppressWarnings("unchecked")
      protected <R> List<R> query(String query, Class<R> jdbcClass) {
          Query nativeQuery = getEntityManager().createNativeQuery(query);
          Stream<R> resultStream = (Stream<R>) nativeQuery.getResultStream().map(jdbcClass::cast);
          return resultStream.collect(Collectors.toList());
      }
  
      protected <R> List<R> query(String query, Function<Object[], R> rowMapper) {
          Query nativeQuery = getEntityManager().createNativeQuery(query);
          Stream<Object[]> resultStream = (Stream<Object[]>) nativeQuery.getResultStream();
          return resultStream.map(rowMapper).collect(Collectors.toList());
      }
  
      protected <R> List<R> query(String query, Function<Object[], R> rowMapper, int offset, int size) {
          Query nativeQuery = getEntityManager().createNativeQuery(query)
                  .setFirstResult(offset)
                  .setMaxResults(size);
          Stream<Object[]> resultStream = (Stream<Object[]>) nativeQuery.getResultStream();
          return resultStream.map(rowMapper).collect(Collectors.toList());
      }
  
  
      protected int queryUpdate(String query) {
          if (log.isDebugEnabled()) log.debug("aggregate: " + query);
          Query nativeQuery = getEntityManager().createNativeQuery(query);
          return nativeQuery.executeUpdate();
      }
 
     protected long queryCount(String query) {
         if (log.isDebugEnabled()) log.debug("aggregate: " + query);
         Query nativeQuery = getEntityManager().createNativeQuery(query);
         Object result = nativeQuery.getSingleResult();
         if (result == null)
             throw new DataRetrievalFailureException(String.format("query count result is null.\nquery: %s", query));
         if (result instanceof Number) {
             return ((Number) result).longValue();
         } else {
             throw new DataRetrievalFailureException(String.format("query count result is not a number: %s \nquery: %s", result, query));
         }
     }
 
     protected Integer getReferentialIdByUniqueLabel(Class<? extends IItemReferentialEntity> entityClass, String label) {
         return referentialService.getIdByUniqueLabel(entityClass, label);
     }
 
     /**
      * Create a new XML Query
      * @return
      */
     protected XMLQuery createXMLQuery() {
         return applicationContext.getBean("xmlQuery", XMLQuery.class);
     }
 
 
 }