使用弹簧数据规范构建动态选择查询

Question

我正在尝试使用 Spring Data 规范构建一个选择查询。有问题的查询如下：

SELECT * FROM Product WHERE id IN (SELECT Product_id FROM Product_tags WHERE Tags IN ('GRADUATION', 'BIRTHDAY'));

用户应该提供一组与子查询中的 IN 运算符匹配的标签，例如 BIRTHDAY 和 GRADUATION。我尝试根据这个答案构建我的解决方案，但遇到了一些麻烦。

    public static Specification<Product> withTags(Set<Tags> tags) {
        return tags == null ?
                null :
                (root, query, criteriaBuilder) -> {
                    List<Predicate> predicates = new ArrayList<>();
                    Subquery<Tags> subquery = query.subquery(Tags.class);

                    Root<Tags> subqueryRoot = subquery.from(Tags.class);
                    subquery.select(subqueryRoot.get("product_tags")
                        .get("product_id"));
                    subquery.where(criteriaBuilder.trim(subqueryRoot.get("product")
                        .get("id")).in(tags));

                    predicates.add(subqueryRoot.get("*").in(subquery));
                    return criteriaBuilder.and(predicates.toArray(new Predicate[0]));

                };
    }

这里的问题是我试图从 Tags 创建一个子查询，它没有注册为实体，而是一个枚举。因此，执行代码会出现错误（这是迄今为止我遇到的唯一错误，请指出代码中可能导致其他错误的部分）。

public enum Tags {

    BIRTHDAY("birthday"),
    GRADUATION("graduation"),
    GET_WELL_SOON("get well soon"),
    RIBBON("ribbon"),
    WRAPPING_PAPER("wrapping paper");

    final String tagName;

    private Tags(String tagName) {
        this.tagName = tagName;
    }

    public String getTagName() {
        return tagName;
    }
}

不确定这是否有帮助，但在 Product 类中，有一个用 @ElementCollection 表示的字段 tags。 Spring 自动创建一个名为“product_tags”的表，子查询从此表中进行选择。

    @ElementCollection(fetch = FetchType.EAGER)
    @Enumerated(EnumType.STRING)
    private Set<Tags> tags;

如果可能，我想翻译此查询而不是第一个查询

SELECT * FROM Product WHERE id IN (SELECT Product_id FROM Product_tags WHERE Tags = ANY(ARRAY['GRADUATION', 'GET_WELL_SOON']));

更新

我已经编辑了我的代码，

    public static Specification<Product> withTags(Set<Tags> tags) {
        return tags == null ?
                null :
                (root, query, criteriaBuilder) -> {

            List<Predicate> predicates = new ArrayList<>();
            Subquery<Long> subquery = query.subquery(Long.class);
            Root<Product> subroot = subquery.from(Product.class);

            subquery.select(subroot.get("id").get("tags"));

            subquery.where(criteriaBuilder.trim(subroot.join("tags")
                .get("id")).in(tags));

            predicates.add(root.get("id").in(subquery));

            return criteriaBuilder.and(predicates.toArray(new Predicate[0]));
        };
    }

但现在我收到此错误

java.lang.IllegalStateException: Illegal attempt to dereference path source [null.id] of basic type

作为参考，我的表定义为这样的

产品：

   Column    |          Type          | Collation | Nullable | Default
-------------+------------------------+-----------+----------+---------
 id          | bigint                 |           | not null |
 category    | character varying(255) |           |          |
 date_added  | date                   |           |          |
 description | character varying(255) |           |          |
 name        | character varying(255) |           |          |
 price       | double precision       |           | not null |

product_tags：

   Column   |          Type          | Collation | Nullable | Default
------------+------------------------+-----------+----------+---------
 product_id | bigint                 |           | not null |
 tags       | character varying(255) |           |          |

Answer 1

    public static Specification<Product> withTags(Set<Tags> tags) {
        return tags.isEmpty() ?
                null:
                (root, query, criteriaBuilder) -> {
                    Subquery<Tags> subquery = query.subquery(Tags.class);
                    Root<Product> subroot = subquery.from(Product.class);

                    subquery.select(subroot.get("id")).where(subroot.join("tags").in(tags));
                    Predicate predicate = root.get("id").in(subquery);

                    return criteriaBuilder.and(predicate);
        };

我似乎找到了答案。 Tags.class 显然可以工作，从那里我只需将查询调整为连接选择。这不是我最初希望实现的目标，但它确实有效。