eu.drus.jpa.unit.cdi.EntityManagerHolder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jpa-unit-cdi Show documentation

Makes JPA Unit usable in CDI test environment

There is a newer version: 0.5.0

package eu.drus.jpa.unit.cdi;

import javax.persistence.EntityManager;

class EntityManagerHolder {

    public static final EntityManagerHolder INSTANCE = new EntityManagerHolder();
    private static final ThreadLocal CONTEXT = new ThreadLocal<>();

    private EntityManagerHolder() {}

    public EntityManager getEntityManager() {
        return CONTEXT.get();
    }

    public void setEntityManager(final EntityManager value) {
        if (value != null) {
            CONTEXT.set(value);
        } else {
            CONTEXT.remove();
        }
    }
}