其他分享
首页 > 其他分享> > Spring Data JPA 连表动态条件查询

Spring Data JPA 连表动态条件查询

作者:互联网

多表查询在spring data jpa中有两种实现方式,第一种是利用hibernate的级联查询来实现(使用较为复杂,查询不够灵活),第二种是使用原生sql查询。

JPA原生SQL连表查询

@Repository
public class SqlQueryRepository implements BaseQueryRepository {
    private static final String COUNT_REPLACEMENT_TEMPLATE = "select count(%s) $5$6$7";
    /**
     * 匹配第三组,换成count(*),目前只用到simple
     */
    private static final String SIMPLE_COUNT_VALUE = "$3*";
    /**
     * 复杂查询,count(主对象)
     */
    private static final String COMPLEX_COUNT_VALUE = "$3$6";

    private static final Pattern COUNT_MATCH;
    private static final String IDENTIFIER = "[\\p{Alnum}._$]+";
    private static final String IDENTIFIER_GROUP = String.format("(%s)", IDENTIFIER);

    @PersistenceContext
    private EntityManager entityManager;

    // (select\s+((distinct )?(.+?)?)\s+)?(from\s+[\p{Alnum}._$]+(?:\s+as)?\s+)([\p{Alnum}._$]+)(.*)
    static {
        StringBuilder builder = new StringBuilder();
        // from as starting delimiter
        builder.append("(?<=from)");
        // at least one space separating
        builder.append("(?: )+");
        // Entity name, can be qualified (any
        builder.append(IDENTIFIER_GROUP);
        // exclude possible "as" keyword
        builder.append("(?: as)*");
        // at least one space separating
        builder.append("(?: )+");
        // the actual alias
        builder.append("(\\w*)");

        builder = new StringBuilder();
        builder.append("(select\\s+((distinct )?(.+?)?)\\s+)?(from\\s+");
        builder.append(IDENTIFIER);
        builder.append("(?:\\s+as)?\\s+)");
        builder.append(IDENTIFIER_GROUP);
        builder.append("(.*)");

        COUNT_MATCH = compile(builder.toString(), CASE_INSENSITIVE);
    }


    /**
     * 封装原生sql分页查询,自动生成countSql
     *
     * @param pageable 分页参数
     * @param querySql 查询sql,不包含排序
     * @param orderSql 排序sql
     * @param paramMap 参数列表
     * @param clazz    返回对象class
     * @param <T>      返回对象
     * @return PageImpl
     */
    @Override
    public <T> Page<T> queryPageable(String querySql, String orderSql, Map<String, Object> paramMap, Pageable pageable, Class<T> clazz) {
        String countSql = createCountQuery(querySql);
        Query countQuery = (Query)this.entityManager.createNativeQuery(countSql);
        Query query = (Query)this.entityManager.createNativeQuery(querySql + orderSql);

        // 设置参数
        if (paramMap != null && paramMap.size() > 0) {
            for (Map.Entry<String, Object> entry : paramMap.entrySet()) {
                countQuery.setParameter(entry.getKey(), entry.getValue());
                query.setParameter(entry.getKey(), entry.getValue());
            }
        }

        BigInteger totalCount = (BigInteger) countQuery.getSingleResult();

        query.setFirstResult((int) pageable.getOffset());
        query.setMaxResults(pageable.getPageSize());
        // 不使用hibernate转bean,存在数据类型问题
        //query.setResultTransformer(Transformers.aliasToBean(clazz));
        query.setResultTransformer(Transformers.ALIAS_TO_ENTITY_MAP);
        List<T> resultList = JSON.parseArray(JSON.toJSONString(query.getResultList(), SerializerFeature.WriteMapNullValue), clazz);

        return new PageImpl<>(resultList, pageable, totalCount.longValue());
    }


    /**
     * 根据查询sql自动生成countSql,正则匹配
     *
     * @param sql 查询sql
     * @return countSql
     */
    public String createCountQuery(String sql) {
        Matcher matcher = COUNT_MATCH.matcher(sql);
        return matcher.replaceFirst(String.format(COUNT_REPLACEMENT_TEMPLATE, SIMPLE_COUNT_VALUE));
    }

}

使用示例

@Repository
public class RiskChangeDaoImpl implements RiskChangeNativeDao {
    @Autowired
    private BaseQueryRepository baseQueryRepository;


    @Override
    public Page<RiskChange> getRiskChangePageable(String tenantId, String userId, RiskMonitorDTO riskMonitorDTO, PageRequest pageable) {
        // 拼接查询sql
        StringBuilder selectSql = new StringBuilder();
        selectSql.append("SELECT rc.id id, rc.pk_risk_change pkRiskChange, rc.pk_risk pkRisk, rc.change_date changeDate, rc.company_id companyId, ");
        selectSql.append("rc.company_name companyName, rc.risk_level riskLevel, rc.risk_type riskType, rc.risk_content riskContent ");
        selectSql.append("FROM dv_risk_sub rs,dv_risk_change rc  ");
        selectSql.append("WHERE  rs.user_id = :userId  AND rs.tenant_id = :tenantId AND rs.dr = 0 ");
        selectSql.append("AND rs.company_id = rc.company_id  AND rc.change_date >= rs.sub_date ");

        HashMap<String, Object> paramMap = new HashMap<>(16);
        paramMap.put("userId", userId);
        paramMap.put("tenantId", tenantId);

        StringBuilder whereSql = new StringBuilder();
        // 企业名称模糊筛选
        if (StringUtils.isNotBlank(riskMonitorDTO.getCompanyName())) {
            whereSql.append(" AND rs.company_name like :companyName ");
            paramMap.put("companyName", "%" + riskMonitorDTO.getCompanyName() + "%");
        }

        // 风险类型筛选 in
        if (!CollectionUtils.isEmpty(riskMonitorDTO.getRiskTypes())) {
            whereSql.append(" AND rc.risk_type in :riskType ");
            paramMap.put("riskType",  riskMonitorDTO.getRiskTypes());
        }

        if (StringUtils.isNotBlank(riskMonitorDTO.getStartChangeDate())) {
            whereSql.append(" AND rc.change_date >= :startChangeDate ");
            paramMap.put("startChangeDate",  riskMonitorDTO.getStartChangeDate());
        }
        if (StringUtils.isNotBlank(riskMonitorDTO.getEndChangeDate())) {
            whereSql.append(" AND rc.change_date <= :endChangeDate ");
            paramMap.put("endChangeDate",  riskMonitorDTO.getEndChangeDate());
        }

        // 添加排序
        String orderSql = " ORDER BY changeDate desc, companyId desc ";
        String querySql = selectSql.append(whereSql).toString();

        return baseQueryRepository.queryPageable(querySql, orderSql, paramMap, pageable, RiskChange.class);
    }
}

标签:riskMonitorDTO,String,JPA,Spring,rc,连表,paramMap,append,risk
来源: https://www.cnblogs.com/iiot/p/13446835.html