Java 杂谈程序员

Java jpa specification postgres

2019-02-22  本文已影响0人  dVito

问题描述

数据库用的postgresql,有一个字段的类型是text[]类型,需要查出该字段包含某个值的数据条目。

好吧sql查还是很简单的:@> '{value}'包含查询就ok了,但是刚把框架切换到了jpa并且定义了一套基于jpa specification的查询逻辑,怎么在这样的环境下实现呢?

遇到的问题

  1. jpa并不支持pg的数组类型,所以也理所当然的没有数组的@>包含运算符
  2. 百度、Google、Stack Overflow 都搜索了居然没有找到直接的解决方案??

v1版解决方案:

搜不到好的解决方案只能自己解决了。一开始也没想到好的解决方案,但是有一个方肯定行,当字符串处理!,直接按字符串比较的方式用like运算。

cb.like(root.get("tags").as(String.class), "%" + tag+ "%");

ok了能正常运行。虽然功能确实是能满足要求,并且在目前的场景下也是不会出现bug的,但是这么low的方法是不能接受的,还是要继续掉头发。

最终版解决方案:

继续查资料发现有类似的场景(筛选JSON类型下的某一字段)用到了CriteriaBuilder.function调用sql函数。没错就是要这个功能。
这里完全可以通过function调用arraycontains实现包含判断,到这应该没问题了吧。

cb.isTrue(cb.function("arraycontains", Boolean.class, root.get("tags"), cb.literal("{tag}")

报错了,还是类型错误。

Caused by: org.postgresql.util.PSQLException: ERROR: function arraycontains(text[], character varying) does not exist
  建议:No function matches the given name and argument types. You might need to add explicit type casts.

arraycontains的参数是(array, array)通过jpa转换出来的是array,string。
好吧调用参数类型不对那就额外调用一次将字符串转数组,pg还有string_to_array这个函数,再用这个试一下呢,但是还是有类型问题,返回值类型写什么呢,姑且先用String吧。

Expression<String> exp = cb.function("string_to_array", String.class, cb.literal(tag), cb.literal(","));
predicates.add(cb.isTrue(cb.function("arraycontains", Boolean.class, root.get("tags"), exp)));

运行正常,没问题了!

下面附上完整的代码

public Specification<EpProject> getSpecification() {
        return (Specification<EpProject>) (root, criteriaQuery, cb) -> {
            List<Predicate> predicates = new ArrayList<>();

            if (tag!= null) {
                Expression<String> exp = cb.function("string_to_array", String.class, cb.literal(tag), cb.literal(","));
                predicates.add(cb.isTrue(cb.function("arraycontains", Boolean.class, root.get("tags"), exp)));
            }

            Predicate[] ps = new Predicate[predicates.size()];
            return cb.and(predicates.toArray(ps));
        };
    }

后记

这个问题的解决确实费了不少功夫,把解决过程记录一下方便大家。

上一篇 下一篇

猜你喜欢

热点阅读