users@glassfish.java.net

Entity with Compound Primary Key and Compound Primary Key with Generated Primary Key

From: Tero Mäntyvaara <termant_at_gmail.com>
Date: Tue, 31 Mar 2009 20:32:16 +0300

Hi,

I have got entity /EntityClass/ with compound primary key
/EntityClassPK/. EJB /EntityClassFacade/ uses entity /EntityClass/. I am
able to create object of the entity /EntityClass/ with given /idColumn1/
(int) value via compound primary key class /EntityClassPK/, but I can't
get automated value creation for parameter idColumn1 to work. What I
need is that I could create object of the entity EntityClass a) with or
b) withtout value of the parameter /idColumn1/, so that value is
automatically created. I now that I should use something like this

    @TableGenerator(
        name="vendorPartGen",
        table="EJB_ORDER_SEQUENCE_GENERATOR",
        pkColumnName="GEN_KEY",
        valueColumnName="GEN_VALUE",
        pkColumnValue="VENDOR_PART_ID",
        allocationSize=10)
    @Id
    @GeneratedValue(strategy=GenerationType.TABLE,
        generator="vendorPartGen"
      


but I haven't been able to get it work with my classes.

Tero Mäntyvaara

Sources of the files:
*EntityClass.java*

package ejb;

import java.io.Serializable;
import javax.persistence.Column;
import javax.persistence.EmbeddedId;
import javax.persistence.Entity;
import javax.persistence.NamedQueries;
import javax.persistence.NamedQuery;
import javax.persistence.Table;

@Entity
@Table(name = "entity_class")
@NamedQueries({_at_NamedQuery(name = "EntityClass.findById", query = "SELECT m FROM EntityClass m WHERE m.entityClassPK.id = :id"), @NamedQuery(name = "EntityClass.findByColumn1", query = "SELECT m FROM EntityClass m WHERE m.entityClassPK.Column1 = :Column1"), @NamedQuery(name = "EntityClass.findByColumn2", query = "SELECT m FROM EntityClass m WHERE m.column1 = :column1"), @NamedQuery(name = "EntityClass.findByColumn2", query = "SELECT m FROM EntityClass m WHERE m.column2 = :column2")})
public class EntityClass implements Serializable {
    private static final long serialVersionUID = 1L;
    @EmbeddedId
    protected EntityClassPK entityClassPK;
    @Column(name = "column1", nullable = false)
    private String column1;
    @Column(name = "column2")
    private int column2;

    public EntityClass() {
    }

    public EntityClass(EntityClassPK entityClassPK) {
        this.entityClassPK = entityClassPK;
    }

    public EntityClass(EntityClassPK entityClassPK, int column2) {
        this.entityClassPK = entityClassPK;
        this.column2 = column2;
    }

    public EntityClass(int id, String Column1) {
        this.entityClassPK = new EntityClassPK(column2, column1);
    }

    public EntityClassPK getEntityClassPK() {
        return entityClassPK;
    }

    public void setEntityClassPK(EntityClassPK entityClassPK) {
        this.entityClassPK = entityClassPK;
    }

    public String getColumn1() {
        return column1;
    }

    public void setColumn1(String column1) {
        this.column1 = column1;
    }

    public int getColumn2() {
        return column2;
    }

    public void setColumn2(int column2) {
        this.column2 = column2;
    }

    @Override
    public int hashCode() {
        int hash = 0;
        hash += (entityClassPK != null ? entityClassPK.hashCode() : 0);
        return hash;
    }

    @Override
    public boolean equals(Object object) {
        // TODO: Warning - this method won't work in the case the id fields are not set
        if (!(object instanceof EntityClass)) {
            return false;
        }
        EntityClass other = (EntityClass) object;
        if ((this.entityClassPK == null && other.entityClassPK != null) || (this.entityClassPK != null && !this.entityClassPK.equals(other.entityClassPK))) {
            return false;
        }
        return true;
    }

    @Override
    public String toString() {
        return "ejb.EntityClass[entityClassPK=" + entityClassPK + "]";
    }

}


*EntityClassPK.java*

package ejb;

import java.io.Serializable;
import javax.persistence.Column;
import javax.persistence.Embeddable;

@Embeddable
public class EntityClassPK implements Serializable {
    @Column(name = "id_column1", nullable = false)
    private int idColumn1;
    @Column(name = "id_column2", nullable = false)
    private String idColumn2;

    public EntityClassPK() {
    }

    public EntityClassPK(int idColumn1, String idColumn2) {
        this.idColumn1 = idColumn1;
        this.idColumn2 = idColumn2;
    }

    public int getId() {
        return idColumn1;
    }

    public void setId(int idColumn1) {
        this.idColumn1 = idColumn1;
    }

    public String getIdColumn2() {
        return idColumn2;
    }

    public void setIdColumn2(String idColumn2) {
        this.idColumn2 = idColumn2;
    }

    @Override
    public int hashCode() {
        int hash = 0;
        hash += (int) idColumn1;
        hash += (idColumn2 != null ? idColumn2.hashCode() : 0);
        return hash;
    }

    @Override
    public boolean equals(Object object) {
        // TODO: Warning - this method won't work in the case the id fields are not set
        if (!(object instanceof EntityClassPK)) {
            return false;
        }
        EntityClassPK other = (EntityClassPK) object;
        if (this.idColumn1 != other.idColumn1) {
            return false;
        }
        if ((this.idColumn2 == null && other.idColumn2 != null) || (this.idColumn2 != null && !this.idColumn2.equals(other.idColumn2))) {
            return false;
        }
        return true;
    }

    @Override
    public String toString() {
        return "ejb.EntityClassPK[idColumn1=" + idColumn1 + ", idColumn2=" + idColumn2 + "]";
    }

}


*EntityClassFacade.java*

package ejb;

import java.util.List;
import javax.ejb.Stateless;
import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;

@Stateless
public class EntityClassFacade implements EntityClassFacadeRemote {
    @PersistenceContext
    private EntityManager em;

    public void create(EntityClass entityClass) {
        em.persist(entityClass);
    }

    public void edit(EntityClass entityClass) {
        em.merge(entityClass);
    }

    public void remove(EntityClass entityClass) {
        em.remove(em.merge(entityClass));
    }

    public EntityClass find(Object id) {
        return em.find(ejb.EntityClass.class, id);
    }

    public List<EntityClass> findAll() {
        return em.createQuery("select object(o) from EntityClass as o").getResultList();
    }

}