spring boot 自定義sql分頁查詢


 @Override
    public <T> Page<T> pageSQL(@Nonnull String sql, @Nonnull Pageable pageable, @Nonnull Class<T> clazz) {
        Long totalCount = totalCount(sql);
        NativeQuery nativeQuery = (NativeQuery) entityManager.createNativeQuery(sql);
        this.addScalar(nativeQuery, clazz);
        Query query = nativeQuery.unwrap(NativeQueryImpl.class).setResultTransformer(Transformers.aliasToBean(clazz));
        query.setFirstResult(pageable.getPageNumber() * pageable.getPageSize());
        query.setMaxResults(pageable.getPageSize());
        List<T> list = query.getResultList();
        return new PageImpl<>(list, pageable, totalCount);
    }

    /**
     * 添加 字段類型
     * @param nativeQuery sql查詢語句
     * @param clazz 類
     */
    private void addScalar(NativeQuery nativeQuery, @Nonnull Class clazz) {
        for (Field field : clazz.getDeclaredFields()) {
            nativeQuery.addScalar(field.getName(), CLASS_TYPE_MAP.get(field.getType()));
        }
        if (clazz.getSuperclass() != null) {
            addScalar(nativeQuery, clazz.getSuperclass());
        }
    }

    @Override
    public Long totalCount(@Nonnull String sql) {
        sql = "select count(1) from (" + sql + ") countAlias";
        NativeQuery query = (NativeQuery) entityManager.createNativeQuery(sql);
        BigInteger bigInteger = (BigInteger) query.uniqueResult();
        return bigInteger.longValue();
    }

    public static final Map<Class<?>, Type> CLASS_TYPE_MAP = new HashMap<>();

    static {
        CLASS_TYPE_MAP.put(Integer.class, StandardBasicTypes.INTEGER);
        CLASS_TYPE_MAP.put(Long.class, StandardBasicTypes.LONG);
        CLASS_TYPE_MAP.put(Float.class, StandardBasicTypes.FLOAT);
        CLASS_TYPE_MAP.put(Double.class, StandardBasicTypes.DOUBLE);
        CLASS_TYPE_MAP.put(BigDecimal.class, StandardBasicTypes.BIG_DECIMAL);
        CLASS_TYPE_MAP.put(String.class, StandardBasicTypes.STRING);
        CLASS_TYPE_MAP.put(LocalDate.class, LocalDateType.INSTANCE);
        CLASS_TYPE_MAP.put(LocalDateTime.class, LocalDateTimeType.INSTANCE);
        CLASS_TYPE_MAP.put(LocalTime.class, LocalTimeType.INSTANCE);
    }

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM