PostgreSQL
 sql >> Base de données >  >> RDS >> PostgreSQL

Mappage Hibernate entre l'énumération PostgreSQL et l'énumération Java

Vous pouvez simplement obtenir ces types via Maven Central en utilisant la dépendance Hibernate Types :

<dependency>
    <groupId>com.vladmihalcea</groupId>
    <artifactId>hibernate-types-52</artifactId>
    <version>${hibernate-types.version}</version>
</dependency>

Si vous mappez facilement Java Enum à un type de colonne PostgreSQL Enum en utilisant le type personnalisé suivant :

public class PostgreSQLEnumType extends org.hibernate.type.EnumType {
     
    public void nullSafeSet(
            PreparedStatement st, 
            Object value, 
            int index, 
            SharedSessionContractImplementor session) 
        throws HibernateException, SQLException {
        if(value == null) {
            st.setNull( index, Types.OTHER );
        }
        else {
            st.setObject( 
                index, 
                value.toString(), 
                Types.OTHER 
            );
        }
    }
}

Pour l'utiliser, vous devez annoter le champ avec Hibernate @Type annotation comme illustré dans l'exemple suivant :

@Entity(name = "Post")
@Table(name = "post")
@TypeDef(
    name = "pgsql_enum",
    typeClass = PostgreSQLEnumType.class
)
public static class Post {
 
    @Id
    private Long id;
 
    private String title;
 
    @Enumerated(EnumType.STRING)
    @Column(columnDefinition = "post_status_info")
    @Type( type = "pgsql_enum" )
    private PostStatus status;
 
    //Getters and setters omitted for brevity
}

Ce mappage suppose que vous avez le post_status_info type enum dans PostgreSQL :

CREATE TYPE post_status_info AS ENUM (
    'PENDING', 
    'APPROVED', 
    'SPAM'
)

Ça y est, ça marche comme un charme. Voici un test sur GitHub qui le prouve.