org.wisepersist.AbstractDao Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of wise-persist Show documentation

WisePersist is a simple JPA wrapper which provides @Transactional and @NonTransactional annotations and pre-configured Guice module.

The newest version!

/*
 * Copyright (c) 2014 WisePersist
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.wisepersist;

import javax.persistence.EntityManager;

/**
 * @author jiakuanwang
 */
public abstract class AbstractDao {

  private static final ThreadLocal threadLocal = new ThreadLocal<>();

  protected EntityManager em() {
    if (threadLocal.get() == null) {
      throw new IllegalStateException(
          "All public DAO methods should be annotated with @Transactional or @NonTransactional");
    }
    return threadLocal.get();
  }

  public void setEntityManager(EntityManager entityManager) {
    if (threadLocal.get() != null && entityManager != null) {
      throw new DaoException(
          "DAO methods annotated with @Transactional or @NonTransactional cannot be nested");
    }
    threadLocal.set(entityManager);
  }

// TODO(Jake): Do we need to support nested methods?
//  public void setEntityManager(EntityManager entityManager, boolean useTransaction) {
//    EntityManager currentEm = threadLocal.get();
//    if (currentEm != null) {
//      if (useTransaction && currentEm.getTransaction().isActive()) {
//        try {
//          currentEm.getTransaction().commit();
//        } catch (Exception e) {
//          log.error(e.getMessage(), e);
//          currentEm.getTransaction().rollback();
//        }
//      }
//      if (currentEm.isOpen()) {
//        currentEm.close();
//      }
//    }
//    threadLocal.set(entityManager);
//  }
}