package com.trendmicro.grid.acl.ds.jpa;
   
   import com.trendmicro.grid.acl.ds.SourceDomainRepository;
   import com.trendmicro.grid.acl.ds.datatypes.SharedSourceDomain;
   import com.trendmicro.grid.acl.ds.datatypes.SharedSourceDomainListPage;
   import com.trendmicro.grid.acl.ds.jpa.entities.JpaSourceDomain;
   import com.trendmicro.grid.acl.l0.datatypes.SourceDomain;
   import org.slf4j.Logger;
   import org.slf4j.LoggerFactory;
  import org.springframework.stereotype.Repository;
  import org.springframework.transaction.annotation.Transactional;
  
  import javax.persistence.EntityManager;
  import javax.persistence.PersistenceContext;
  import javax.persistence.TypedQuery;
  import java.util.List;
  
  /**
   * Implements SourceDomainRepository using JPA.
   *
   * @author juergen_kellerer, 2010-06-01
   * @version 1.0
   */
  @Repository
  @Transactional(readOnly = true)
  public class JpaSourceDomainRepository implements SourceDomainRepository {
  
  	private static final Logger log = LoggerFactory.getLogger(JpaSourceDomainRepository.class);
  	private static int pageSize = 25;
  
  	public static int getPageSize() {
  		return pageSize;
  	}
  
  	public static void setPageSize(int pageSize) {
  		JpaSourceDomainRepository.pageSize = pageSize;
  	}
  
  	@PersistenceContext(unitName = "CoreDB")
  	EntityManager em;
  
  	@Override
  	public SharedSourceDomainListPage getAllPaged(int pageNumber) {
  		final TypedQuery<SharedSourceDomain> q = em.createNamedQuery("SourceDomain.SelectAll", SharedSourceDomain.class);
  
  		q.setFirstResult(pageNumber * pageSize).setMaxResults(pageSize + 1);
  		List<SharedSourceDomain> domains = q.getResultList();
  
  		boolean hasMore = domains.size() > pageSize;
  
  		if (hasMore) {
  			if (log.isTraceEnabled()) log.trace("Removing the indicator row that created the hasMore=true result.");
  			domains.remove(domains.size() - 1);
  		}
  
  		return new SharedSourceDomainListPage(pageNumber, !hasMore, domains);
  	}
  
  	@Override
  	public SharedSourceDomain getByName(String domainName) {
  		List<SharedSourceDomain> result = em.createNamedQuery("SourceDomain.SelectDomainByName", SharedSourceDomain.class).
  				setParameter("name", domainName).getResultList();
  
  		return result.isEmpty() ? null : result.get(0);
  	}
  
  	@Override
  	@Transactional(readOnly = false)
  	public SharedSourceDomain getOrCreate(String domainName) {
  		SharedSourceDomain domain = getByName(domainName);
  
  		if (domain == null) {
  			domain = new JpaSourceDomain(domainName);
  			em.persist(domain);
  		}
  
  		return domain;
  	}
  
  	@Override
  	@Transactional(readOnly = false)
  	public void createOrUpdate(SourceDomain domain) {
  		JpaSourceDomain jpaDomain = new JpaSourceDomain(domain);
  
  		if (jpaDomain.getPrimaryKey() == 0) {
  			final List<Integer> result = em.createNamedQuery("SourceDomain.SelectIdByName", Integer.class).
  					setParameter("name", domain.getName()).getResultList();
  
  			if (result.isEmpty())
  				em.persist(jpaDomain);
  			else
  				jpaDomain.setPrimaryKey(result.get(0));
  		}
  
  		em.merge(jpaDomain);
  	}
  }