JPA/Hibernate自定义查询和Postgres枚举列表

Question

我有一个实体Company，其类型由enumCompanyType表示。数据库是Postgres，在那里表示为枚举类型。我使用JPA/Hibernate及其存储库。请注意，我是JPA、Hibernate和Kotlin的新手。

我试图创建自定义@Query，我需要选择类型（枚举）在可能类型列表中的公司。我，但是，遇到各种错误类型铸造在SQL和/或语法的@Query。

Kotlin中数据类公司的主要部分（未复制任何其他属性，包括id）：

@Entity(name = "Company")
@Table(name = "company")
data class Company(

    @Enumerated(EnumType.STRING)
    @NotNull
    @Column(name = "type", nullable = false, columnDefinition = "company_type")
    val type: CompanyType = CompanyType.OTHER
) : Serializable

枚举CompanyType的静态编程语言：

enum class CompanyType(value: Int) {
    BUSINESS(1),
    TOWN(2),
    NONPROFIT(3),
    RESEARCH(4),
    OTHER(5)
}

在Postgres中枚举company_type：

CREATE TYPE public.company_type AS ENUM (
    'BUSINESS',
    'TOWN',
    'NONPROFIT',
    'RESEARCH',
    'OTHER'
);
CREATE CAST (character varying AS public.company_type) WITH INOUT AS ASSIGNMENT;

JPA存储库与我的初始尝试：

@Repository
interface CompanyDAO : PagingAndSortingRepository<Company> {

    @Query("SELECT c FROM #{#entityName} c " +
        "WHERE c.type IN ?1"
    )
    fun findAllByTypeIn(types: List<CompanyType>, pageable: Pageable): Page<Company>
}

但一旦执行，就会发生以下错误：

ERROR: operator does not exist: company_type = character varying

所以我试图铸造它，但不知道具体如何...

    @Query("SELECT c FROM #{#entityName} c " +
        "WHERE c.type IN cast(?1 AS company_type[])"
    )

导致编译时出错：

antlr.MismatchedTokenException: expecting EOF, found ')'

并试图

    @Query("SELECT c FROM #{#entityName} c " +
        "WHERE c.type IN ?1\\:\\:company_type[]"
    )

导致

org.hibernate.QueryException: unexpected char: '\'

如何创建这样的查询，以获取枚举列表并返回值等于列表中任何项的实体？




Answer 1

将强制转换创建为隐式而不是赋值是否有助于将其用于比较？

使用INOUT作为隐式创建演员阵容（角色随天数变化）；

Answer 2

此错误来自Postgres：

错误：运算符不存在：company_type=字符变化

要解决这个问题，您可以在Postgres中为company_类型创建一个自定义运算符，而不更改代码中的任何内容。这样地：

CREATE FUNCTION ctype_compare(company_type, text)
RETURNS boolean
AS '
select cast($1 as text) = $2;
'
LANGUAGE sql IMMUTABLE;

CREATE OPERATOR = (
    leftarg = company_type,
    rightarg = text,
    procedure = company_type_compare
);

有了这个，你实际上可以删除@Query，Hibernate会做正确的事情。如果您不能创建自定义运算符，可能是因为您没有正确的权限，那么您必须将查询更改为以下内容：

    @Query("SELECT c FROM #{#entityName} c " +
        "WHERE cast (c.type as text) IN ?1")

然后必须将参数类型固定为String。

    Page<Company> findAllByTypeIn(List<String> types, Pageable pageable);

要调用DAO方法，需要传递正确的类型：

    List<String> types = new ArrayList();
    types.add(CompanyType.OTHER.toString());
    types.add(CompanyType.BUSINESS.toString());

    Page<Company> companies = dao.findAllByTypeIn(types, Pageable.unpaged());

我是用Java做的，不是Kotlin。但它应该对你有用。