如何修改Specification以支持IN子句传递订单类型列表

现有一段用于构建查询条件的Java Specification代码，当前方法接收单个orderType参数与refNum参数，需要将orderType参数替换为订单类型列表（List<String>），实现IN子句查询逻辑。该Specification已加入现有搜索查询的WHERE子句中，请问如何修改代码满足需求？

原代码：

public static Specification<OrderEntity> addReference(String orderType, String refNum) {
  return (root, query, builder) -> {
    if (null != refType && null != refNum) { // 注：原代码此处疑似笔误，应为判断orderType
      Join<OrderEntity, OrderDetailEntity> orderSearchJoin =
                root.join(OrderEntity_.orderDetailEntityList);
      query.distinct(true);      
      return builder.and(
                builder.equal(orderSearchJoin.get(OrderDetailEntity_.orderType), orderType),
                builder.equal(orderSearchJoin.get(OrderDetailEntity_.referenceNum), refNum));
    } else {
      return null;
    }
  };
}

按以下要点修改即可实现需求：

• 调整方法入参：将单个String orderType替换为List<String> orderTypes，同时更新入参校验逻辑
• 替换IN查询条件：把原本判断订单类型的builder.equal语句，换成builder.in来匹配列表中的所有值
• 完善空值校验：新增对订单类型列表的有效性判断（非null且非空），避免生成无效查询条件

修改后的完整代码：

public static Specification<OrderEntity> addReference(List<String> orderTypes, String refNum) {
  return (root, query, builder) -> {
    // 校验条件：订单类型列表非空且refNum不为null
    if (orderTypes != null && !orderTypes.isEmpty() && refNum != null) {
      Join<OrderEntity, OrderDetailEntity> orderSearchJoin =
                root.join(OrderEntity_.orderDetailEntityList);
      query.distinct(true);      
      return builder.and(
                // 使用in子句匹配订单类型列表中的所有值
                orderSearchJoin.get(OrderDetailEntity_.orderType).in(orderTypes),
                builder.equal(orderSearchJoin.get(OrderDetailEntity_.referenceNum), refNum));
    } else {
      // 条件不满足时返回null，不会加入WHERE子句
      return null;
    }
  };
}

关键说明 #

1. 修正了原代码中refType的笔误，统一校验orderTypes和refNum的有效性
2. 使用Path.in(Collection)方法构建IN查询，这是JPA Criteria Builder实现IN子句的标准方式
3. 保留query.distinct(true)，避免关联查询时出现重复结果

内容的提问来源于stack exchange，提问作者kalaparthi