jpa工具类

import org.apache.shiro.util.Assert;
import org.springframework.data.domain.Page;
import org.springframework.data.domain.Pageable;
import org.springframework.data.jpa.repository.query.QueryUtils;
import org.springframework.data.repository.support.PageableExecutionUtils;
import org.springframework.stereotype.Repository;
import org.springframework.transaction.annotation.Transactional;

import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import javax.persistence.Query;
import javax.persistence.TypedQuery;
import java.util.List;
import java.util.Map;

/**
 * @Author wangyu
 * @Description: 
 * @Date 2019/3/15
 */
@Transactional(readOnly=true)
@Repository
public class JpaUtil{
    @PersistenceContext
    private EntityManager em;
    /**
     * 获取列表
     * @param sql
     * @param params
     * @param requiredType
     * @return
     */
    public List list(String sql, Map params, Class requiredType) {
        //String hql = "select o.uuid,o.name from UserModel o where 1=1 and o.uuid=:uuid";
        Query query = em.createQuery(sql);
        if (params != null) {
            for (String key : params.keySet()) {
                query.setParameter(key, params.get(key));
            }
        }
        return query.getResultList();
    }

    public List list(String sql, Map params) {
        return list(sql,params,null);
    }
    /**
     * 获取分页数据
     * @param sql
     * @param params
     * @param pageable
     * @param requiredType
     * @return
     */
    @SuppressWarnings("unchecked")
    public Page page(String sql, Map params, Pageable pageable, Class requiredType) {
        Query query = em.createQuery(sql);
        if (params != null) {
            for (String key : params.keySet()) {
                query.setParameter(key, params.get(key));
            }
        }
        if (pageable.isPaged()) {
            query.setFirstResult((int) pageable.getOffset());
            query.setMaxResults(pageable.getPageSize());
        }
        /**
         * 生成获取总数的sql
         */
        TypedQuery cQuery = (TypedQuery) em.createQuery(QueryUtils.createCountQueryFor(sql));
        if (params != null) {
            for (String key : params.keySet()) {
                cQuery.setParameter(key, params.get(key));
            }
        }
        return PageableExecutionUtils.getPage(query.getResultList(), pageable, ()->executeCountQuery(cQuery));
        //return new PageImpl(query.getResultList(), pageable, executeCountQuery(cQuery));
    }

    /**
     * 计算数量
     * @param sql
     * @param params
     * @return
     */
    public int count(String sql, Map params) {
        TypedQuery cQuery = (TypedQuery) em.createQuery(QueryUtils.createCountQueryFor(sql));
        if (params != null) {
            for (String key : params.keySet()) {
                cQuery.setParameter(key, params.get(key));
            }
        }
        long countL =  executeCountQuery(cQuery);
        return Integer.parseInt(countL+"");
    }

    /**
     * Executes a count query and transparently sums up all values returned.
     *
     * @param query must not be {@literal null}.
     * @return
     */
    private static Long executeCountQuery(TypedQuery query) {
        Assert.notNull(query, "TypedQuery must not be null!");
        List totals = query.getResultList();
        Long total = 0L;
        for (Long element : totals) {
            total += element == null ? 0 : element;
        }
        return total;
    }
}

你可能感兴趣的:(springcloud)