是否可以在QueryDSL查询中使用枚举方法

时间:2015-06-22 21:44:31

标签: java jpa enums querydsl

我有一个包含枚举字段的实体类型。 我使用JPA 2.1 AttributeConverter将枚举实例/值映射到数据库列。 Enum值包含我想在QueryDSL查询中使用的其他信息,就好像该信息将存储在数据库中一样。

实施例: 包含messageType的消息实体。类型是枚举值。枚举确定消息是否为全局消息。在数据库中,枚举表示为int值。如果全球或非全球信息是枚举的一部分。

是否可以写这样的东西?

import static com.example.QMessage.message;

@Repository
public class MessageDao {

    @PersistenceContext
    protected EntityManager em;

    public List<Message> getGlobalMessages() {
        return new JPAQuery(em)
            .from(message)
            .where(message.messageType.global.isTrue())
            .list(message);
    }
}

查询应仅返回消息类型为global==true的消息。但生成的QMessage类不提供该功能。 message.messageType提供EnumPath,但无法调用枚举函数。

枚举

public enum MessageType {
    WELCOME(0, true),
    REMINDER(1),
    IMPORTANT(0, true);

    private final int id;
    private final boolean global;

    private MessageType(int id) {
        this(id, false);
    }

    private MessageType(int id, boolean global) {
        this.id = id;
        this.global = global;
    }

    public int getId() {
        return this.id;
    }

    public boolean isGlobal() {
        return this.global;
    }
}

实体

@Entity
public class Message {

    @Id
    private long messageId;

    @Column(name = "MESSAGE")
    private String message;

    @Column(name = "MESSAGE_TYPE")
    private MessageType messageType;

    // Getter and Setter
    // ... 
}

转换器

@Converter(autoApply = true)
public class MessageTypeConverter implements AttributeConverter<Integer, String> {

    @Override
    public Integer convertToDatabaseColumn(MessageType messageType) {
        return messageType.getId();
    }

    @Override
    public Locale convertToEntityAttribute(Integer dbValue) {
        for(MessageType type : values()) {
            if(type.getId() == dbValue) {
                return type;
            }
        }
        throw new IllegalArgumentException("Unknown id");
    }
}

2 个答案:

答案 0 :(得分:0)

一种可能的解决方案/解决方法是在枚举类型中创建一个静态方法,该方法返回匹配枚举值的集合(global==true):

return new JPAQuery(em)
        .from(message)
        .where(message.messageType.in(MessageType.getGlobalMessageTypes())
        .list(message);

答案 1 :(得分:0)

在JPA中将枚举视为值/文字类型,因此没有通过属性/方法引用来处理它的干净方法。