使用 CriteriaBuilder 过滤数据以将枚举值与不工作的文字进行比较

Question

我有一个带有枚举字段的 java 类，

org.example.Importacion {
...
@Enumerated(EnumType.STRING)
  private EstadoImportacion estadoImportacion;

  public static enum EstadoImportacion {
      NOT_VALID, IMPORTED, ERROR, VALID
  }

}

当我使用 CriteriaBuilder 创建查询并尝试比较枚举值时，从过滤器到使用文字的标准构建器，查询的最终结果不会过滤枚举值，因此如果我发送 org.example.Importacion.EstadoImportacion .ERROR 到迭代器方法，结果将不会在最终结果列表上过滤 ERROR。

companyCod 过滤器正常，因此如果我将“COMPANY001”作为 companyCode 发送，则查询构建器会过滤最终结果。

我想知道如何比较查询中的枚举：

   public Iterator<Importacion> iterator (
     long first, 
     long count, 
     String companyCod, 
     org.example.Importacion.EstadoImportacion estado) {

    CriteriaBuilder cb = getEntityManager().getCriteriaBuilder();
    CriteriaQuery<Importacion> criteria = cb.createQuery(Importacion.class);
    Root<Importacion> desembolso = criteria.from(Importacion.class);
    criteria.select(desembolso);
    Predicate p = cb.conjunction();
    if(companyCod != null) {

        p = cb.and(p, cb.equal(desembolso.get("codigo"), companyCod));
        //This part works fine!
    }


    if (estado != null) {
        Expression<org.example.Importacion.EstadoImportacion> estadoImportacion = null;

        if (estado.equals(org.example.Importacion.EstadoImportacion.ERROR)) {
            estadoImportacion = cb.literal(org.example.Importacion.EstadoImportacion.ERROR);
        }

        if (estado.equals(org.example.Importacion.EstadoImportacion.IMPORTED)) {
            estadoImportacion = cb.literal(org.example.Importacion.EstadoImportacion.IMPORTED);
        }

        if (estado.equals(org.example.Importacion.EstadoImportacion.NOT_VALID)) {
            estadoImportacion = cb.literal(org.example.Importacion.EstadoImportacion.NOT_VALID);
        }

        if (estado.equals(org.example.Importacion.EstadoImportacion.VALID)) {
            estadoImportacion = cb.literal(org.example.Importacion.EstadoImportacion.VALID);
        }

        p = cb.and(p, cb.equal(estadoImportacion, cb.literal(estado)));
    //Doesn't seems to compare enum values

    }

    criteria.where(p);
    javax.persistence.Query query = em.createQuery(criteria);

    query.setMaxResults((int)count + (int)first + 1);
    query.setFirstResult((int)first);
    List resultList = query.getResultList();
    Iterator iterator = (Iterator) resultList.iterator();
    LOGGER.info("desembolso size: {}", resultList.size());
    return iterator;
}

Answer 1

您的标准将文字与枚举进行比较。那不是你想要的。您想将 Importacion 的 estadoImportacion 与给定的 estado 进行比较：

Predicate p = cb.conjunction();
if(companyCod != null) {
    p = cb.and(p, cb.equal(desembolso.get("codigo"), companyCod));
}
if (estado != null) {
    p = cb.and(p, cb.equal(desembolso.get("estadoImportacion"), estado));
}