package com.trendmicro.grid.acl.ds.jpa;
   
   import com.trendmicro.grid.acl.ds.ProcessingResultReceiver;
   import com.trendmicro.grid.acl.ds.config.DataSourcePropertySection;
   import com.trendmicro.grid.acl.l0.datatypes.ProcessPackageDataSet;
   import net.sf.tinyjee.config.PropertySection;
   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 java.util.Collection;
  
  import static com.trendmicro.grid.acl.ds.config.DataSourcePropertySection.*;
  import static com.trendmicro.grid.acl.ds.jpa.AbstractReceivedResultsHandler.StorageContext;
  import static com.trendmicro.grid.acl.ds.jpa.AbstractReceivedResultsHandler.getValue;
  
  /**
   * Implements ProcessingResultReceiver using JPA.
   * <p/>
   * Note: This is the primary implementation that stores content inside the CoreDB.
   *
   * @author Juergen_Kellerer, 2010-06-12
   * @version 1.0
   */
  @Repository
  @Transactional
  public class JpaProcessingResultReceiver implements ProcessingResultReceiver {
  
  	private static final Logger log = LoggerFactory.getLogger(JpaProcessingResultReceiver.class);
  
  	@PersistenceContext(unitName = "CoreDB")
  	EntityManager em;
  
  	@Resource
  	JpaVendorRepository vendorRepository;
  	@Resource
  	JpaSourceRepository sourceRepository;
  	@Resource
  	JpaPackageFamilyRepository packageFamilyRepository;
  	@Resource
  	JpaPackageRepository packageRepository;
  	@Resource
  	JpaFileRepository fileRepository;
  
  	ReceivedDataSetsHandler dataSetsHandler;
  	boolean ignoreNullMetadataUpdates, ignoreAllUpdates, enableHistoryRecording;
  
  	@PostConstruct
  	void configure() {
  		PropertySection section = getPropertySection();
  
  		ignoreNullMetadataUpdates = getValue(section, KEY_IGNORE_NULL_METADATA_UPDATES);
  		ignoreAllUpdates = getValue(section, KEY_IGNORE_ALL_UPDATES);
  		enableHistoryRecording = getValue(section, KEY_ENABLE_HISTORY_RECORDING);
  
  		// Building a handler chain.
  		final ReceivedSourcesHandler sourcesHandler = new ReceivedSourcesHandler(sourceRepository);
  		final ReceivedPackageFamiliesHandler familiesHandler = new ReceivedPackageFamiliesHandler(
  			vendorRepository, packageFamilyRepository);
  		final ReceivedPreparedPackagesHandler packagesHandler = new ReceivedPreparedPackagesHandler(
  			fileRepository, packageRepository, familiesHandler);
  
  		dataSetsHandler = new ReceivedDataSetsHandler(
  			fileRepository, packageRepository, packagesHandler, sourcesHandler);
  	}
  
  	public void setEnableHistoryRecording(boolean enableHistoryRecording) {
  		this.enableHistoryRecording = enableHistoryRecording;
  	}
  
  	/**
  	 * {@inheritDoc}
  	 */
  	public void receive(Collection<ProcessPackageDataSet> dataSets) {
  		StorageContext context = new StorageContext(em, dataSets, ignoreAllUpdates, ignoreNullMetadataUpdates, enableHistoryRecording);
  		dataSetsHandler.handle(context);
  	}
  
  	/**
  	 * {@inheritDoc}
  	 */
  	public void checkpoint() {
  		// nothing to do as receive(..) is already transactional.
  	}
  }