package com.trendmicro.grid.acl.ds.jpa;
   
   import com.trendmicro.grid.acl.ds.ReportReceiver;
   import com.trendmicro.grid.acl.ds.jpa.entities.JpaFileDetails;
   import com.trendmicro.grid.acl.ds.jpa.entities.JpaFileReport;
   import com.trendmicro.grid.acl.ds.jpa.util.FileQueryConfigurator;
   import com.trendmicro.grid.acl.l0.datatypes.FileReport;
   import net.sf.tinyjee.config.PropertySection;
   import net.sf.tinyjee.util.Hex;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.stereotype.Repository;
  import org.springframework.transaction.annotation.Transactional;
  
  import javax.annotation.PostConstruct;
  import javax.annotation.Resource;
  import javax.persistence.EntityManager;
  import javax.persistence.PersistenceContext;
  import javax.persistence.Query;
  import java.util.ArrayList;
  import java.util.Collection;
  import java.util.Iterator;
  
  import static com.trendmicro.grid.acl.ds.config.DataSourcePropertySection.KEY_IGNORE_STATISTICS_RECORDING;
  import static com.trendmicro.grid.acl.ds.config.DataSourcePropertySection.getPropertySection;
  import static com.trendmicro.grid.acl.ds.jpa.AbstractReceivedResultsHandler.getValue;
  
  /**
   * Implements ReportReceiver using JPA.
   *
   * @author juergen_kellerer, 2010-06-08
   * @version 1.0
   */
  @Repository
  @Transactional(readOnly = false)
  public class JpaReportReceiver implements ReportReceiver {
  
  	private static final Logger log = LoggerFactory.getLogger(JpaReportReceiver.class);
  
  	@PersistenceContext(unitName = "CoreDB")
  	EntityManager em;
  
  	@Resource
  	JpaFileRepository fileRepository;
  
  	boolean ignoreStatisticsRecording;
  
  	@PostConstruct
  	void configure() {
  		PropertySection section = getPropertySection();
  		ignoreStatisticsRecording = getValue(section, KEY_IGNORE_STATISTICS_RECORDING);
  	}
  
  	@Override
  	public void receiveFileReports(Collection<FileReport> fileReports) {
  		final boolean debug = log.isDebugEnabled();
  		final FileQueryConfigurator<Integer> queryConfigurator = fileRepository.getReferenceQueryConfigurator();
  		final Query updateQuery = em.createNamedQuery("FileReport.UpdateAccessCount");
  
  		// Fetch all required references first to allow hibernate to
  		// do better batch updates in the next section.
  		final Collection<JpaFileDetails> contentReferences = new ArrayList<JpaFileDetails>(fileReports.size());
  		for (FileReport report : fileReports)
  			contentReferences.add(fileRepository.getReference(queryConfigurator, report));
  
  		final Iterator<JpaFileDetails> referenceIterator = contentReferences.iterator();
  		for (FileReport report : fileReports) {
  			boolean entrySkipped = false;
  			final byte[] sha1 = report.getSHA1Hash();
  			final JpaFileDetails fileDetails = referenceIterator.next();
  
  			if (fileDetails != null) {
  				final JpaFileReport jpaReport = new JpaFileReport(fileDetails, report);
  				int updateCount = updateQuery.setParameter("accessCount", jpaReport.getAccessCount()).
  						setParameter("reportId", jpaReport.getReportId()).executeUpdate();
  
  				if (updateCount == 0) {
  					if (debug)
  						log.debug("Creating a new report entry '{}' for file '{}'", jpaReport.getTimePeriod(), Hex.encode(sha1));
  
  					if (!ignoreStatisticsRecording) {
  						em.persist(jpaReport);
  						em.flush();
  						em.detach(jpaReport);
  					}
  
  				} else {
  					if (log.isTraceEnabled()) {
  						log.trace("Incremented the access count of the existing report entry '{}' for file '{}', by {}", new Object[]{
  								jpaReport.getTimePeriod(), Hex.encode(sha1), report.getAmount()});
  					}
  				}
  			} else
  				entrySkipped = true;
  
  			if (entrySkipped) {
  				if (debug)
  					log.debug("Omitting storage of report for unknown file '{}' as the MD5 hash is missing.", Hex.encode(sha1));
  			}
 		}
 		if (!ignoreStatisticsRecording)
 			em.flush();
 	}
 }