问题描述
数据库用的postgresql,有一个字段的类型是text[]类型,需要查出该字段包含某个值的数据条目。
好吧sql查还是很简单的:@> '{value}'包含查询就ok了,但是刚把框架切换到了jpa并且定义了一套基于jpa specification的查询逻辑,怎么在这样的环境下实现呢?
遇到的问题
- jpa并不支持pg的数组类型,所以也理所当然的没有数组的@>包含运算符
- 百度、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));
};
}
后记
这个问题的解决确实费了不少功夫,把解决过程记录一下方便大家。