package com.test.cms.dao.repository;

import org.springframework.stereotype.Repository;

import javax.persistence.EntityManager;

import javax.persistence.PersistenceContext;

import javax.persistence.Query;

import java.math.BigInteger;

import java.util.*;

/**

* 构建复杂查询

* @author lchb

*/

@Repository

public class NativeQueryRepository{

@PersistenceContext

private EntityManager entityManager;

/**

* 根据sql查询

* @param sql

* @param resultClass

* @param

* @return

*/

public List findBySql(String sql,Class resultClass){

Query nativeQuery=null;

if(resultClass!=null && !resultClass.equals(Object.class)) {

nativeQuery = entityManager.createNativeQuery(sql, resultClass);

}else{

nativeQuery = entityManager.createNativeQuery(sql);

}

List rows = nativeQuery.getResultList();

List resultList=new ArrayList();

for (Object obj : rows) {

T row=(T)obj;

resultList.add(row);

}

return resultList;

}

/**

* 根据参数查询

* @param sql

* @param parameters

* @param resultClass

* @param

* @return

*/

public List findBySql(String sql,Map parameters,Class resultClass){

Query nativeQuery=null;

if(resultClass!=null && !resultClass.equals(Object.class)) {

nativeQuery = entityManager.createNativeQuery(sql, resultClass);

}else{

nativeQuery = entityManager.createNativeQuery(sql);

}

Iterator> entries = parameters.entrySet().iterator();

while (entries.hasNext()) {

Map.Entry entry = entries.next();

nativeQuery.setParameter(entry.getKey(),entry.getValue());

}

List rows = nativeQuery.getResultList();

List resultList=new ArrayList();

for (Object obj : rows) {

T row=(T)obj;

resultList.add(row);

}

return resultList;

}

public Integer findCountBySql(String sql,Map parameters){

List list=this.findBySql(sql,parameters,Object.class);

BigInteger total=(BigInteger)list.get(0);

return total.intValue();

}

}