DaoFactory.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package  org.turro.jpa;
 
import java.io.File;
import java.util.Date;
import java.util.HashMap;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.persistence.EntityManager;
import javax.persistence.EntityManagerFactory;
import javax.persistence.Persistence;
import javax.persistence.PersistenceUnit;
import javax.persistence.Query;
import org.amic.util.file.FileSelector;
import org.amic.util.file.FileUtil;
import org.turro.string.Strings;
import org.turro.action.IElephantCloseable;
import org.turro.annotation.ElephantCloseable;
import org.turro.elephant.context.ElephantContext;
import org.turro.jpa.config.JpaConfig;
 
@ElephantCloseable
public class DaoFactory implements IElephantCloseable {
 
  private static final HashMap<String, EntityManager> cachedEMs= new HashMap<>();
  private static final HashMap<String, Long> lastAccessEMs= new HashMap<>();
  private static final long CLEAR_CACHE = 2000L, CLOSE_CACHE = 360000L;
  
  @PersistenceUnit
  private EntityManagerFactory emf = null;
  
  private String pu, conf;
 
  public EntityManager createEntityManager(String pu, String conf) {
    EntityManagerFactory factory = getFactory(pu, conf);
    if(factory != null) {
      return factory.createEntityManager();
    }
    return null;
  }
 
  //TODO: return to EM per session
  @Deprecated
  public synchronized EntityManager createCachedEntityManager(String pu, String conf) {
    EntityManager cachedEM = cachedEMs.get(pu);
    Long lastAccess = lastAccessEMs.get(pu);
    Long time = new Date().getTime();
    if(lastAccess == null) lastAccess = time;
    EntityManagerFactory factory = getFactory(pu, conf);
    if(factory != null) {
      if(cachedEM == null || !(cachedEM.isOpen())) {
        cachedEM = factory.createEntityManager();
      } else if(time > lastAccess + CLEAR_CACHE) {
        cachedEM.clear();
      } else if(time > lastAccess + CLOSE_CACHE) {
        try {
          cachedEM.close();
        } finally {
          cachedEM = factory.createEntityManager();
        }
      }
      lastAccessEMs.put(pu, time);
      cachedEMs.put(pu, cachedEM);
      return cachedEM;
    }
    return null;
  }
 
  public EntityManagerFactory getEmf() {
    return emf;
  }
  
  private synchronized EntityManagerFactory getFactory(String pu, String conf) {
    if(emf == null) {
      doCreateFactory(pu, conf);
    }
    return emf;
  }
 
  private void doCreateFactory(String pu1, String conf1) {
    this.pu = pu1;
    this.conf = conf1;
//    Properties properties = new Properties();
//    HibernateConfig hc = (HibernateConfig) HeadlessApplication.getInstance()
//            .getImplementation("IHibernateConfig_" + this.conf);
//    if(hc != null) hc.updateProperties(properties);
    emf = Persistence.createEntityManagerFactory(this.pu, JpaConfig.from(pu1, conf1));
    FileSelector fs = new FileSelector(ElephantContext.getRealPath("/WEB-INF/_autoupdate"), conf1 + "\\_.*\\.sql");
    if(fs.getFiles() != null && fs.getFiles().length > 0) {
      EntityManager em = emf.createEntityManager();
      try {
        for(File f : fs.getFiles()) {
          em.getTransaction().begin();
          try {
            // skip comments
            for(String sql : FileUtil.getContent(f, "UTF-8", ";", new String[] {"--","/*"})) {
              if(!Strings.isBlank(sql)) {
                // remove final ;
                String execSql = sql.substring(0, sql.length() - 1);
                try {
                  // ensure \' escaping
                  Query q = em.createNativeQuery(execSql.replaceAll("\\\\'", "''"));
                  q.executeUpdate();
                } catch(Exception ex) {
                  Logger.getLogger(DaoFactory.class.getName()).log(Level.FINEST, ElephantContext.logMsg("AutoUpdate:" + f.getName() + ":" + sql), ex);
                }
              }
            }
            em.getTransaction().commit();
          } catch (Exception ex) {
            if(em.getTransaction().isActive()) {
              em.getTransaction().rollback();
            }
            Logger.getLogger(DaoFactory.class.getName()).log(Level.FINEST, ElephantContext.logMsg("AutoUpdate:" + f.getName()), ex);
          } finally {
            FileUtil.deleteFile(f);
          }
        }
      } finally {
        em.close();
      }
    }
  }
  
  public void close() {
    if(emf != null) emf.close();
  }
 
  @Override
  public void closeElephant() {
    close();
    cachedEMs.values().forEach((em) -> {
      em.close();
    });
  }
 
}