| /* |
| * Licensed to the Apache Software Foundation (ASF) under one |
| * or more contributor license agreements. See the NOTICE file |
| * distributed with this work for additional information |
| * regarding copyright ownership. The ASF licenses this file |
| * to you 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.apache.ode.store.jpa; |
| |
| import org.apache.commons.logging.Log; |
| import org.apache.commons.logging.LogFactory; |
| import org.apache.ode.dao.jpa.JpaTxMgrProvider; |
| import org.apache.ode.store.ConfStoreConnection; |
| import org.apache.ode.store.ConfStoreConnectionFactory; |
| |
| import javax.persistence.EntityManagerFactory; |
| import javax.persistence.Persistence; |
| import javax.sql.DataSource; |
| import javax.transaction.TransactionManager; |
| |
| import java.util.HashMap; |
| import java.util.Properties; |
| |
| /** |
| * @author Matthieu Riou <mriou at apache dot org> |
| */ |
| public class DbConfStoreConnectionFactory implements ConfStoreConnectionFactory { |
| private static final Log __log = LogFactory.getLog(DbConfStoreConnectionFactory.class); |
| |
| private DataSource _ds; |
| private EntityManagerFactory _emf; |
| |
| private TransactionManager _txMgr; |
| |
| @SuppressWarnings("unchecked") |
| public DbConfStoreConnectionFactory(DataSource ds, boolean createDatamodel, String txFactoryClassName) { |
| this(ds,new Properties(), createDatamodel, txFactoryClassName); |
| } |
| |
| public DbConfStoreConnectionFactory(DataSource ds, Properties initialProps, boolean createDatamodel, String txFactoryClassName) { |
| _ds = ds; |
| initTxMgr(txFactoryClassName); |
| |
| HashMap<String, Object> propMap = new HashMap<String,Object>(); |
| propMap.put("openjpa.Log", "commons"); |
| propMap.put("openjpa.ManagedRuntime", new JpaTxMgrProvider(_txMgr)); |
| propMap.put("openjpa.ConnectionFactory", _ds); |
| propMap.put("openjpa.ConnectionFactoryMode", "managed"); |
| propMap.put("openjpa.FlushBeforeQueries", "false"); |
| propMap.put("openjpa.FetchBatchSize", 1000); |
| |
| //dirty hack for ODE-1015 |
| String skipIsolation = System.getProperty("openjpa.connection.isolation.skip", "N"); |
| if(skipIsolation.equalsIgnoreCase("N")) |
| propMap.put("openjpa.jdbc.TransactionIsolation", "read-committed"); |
| |
| propMap.put("javax.persistence.provider", "org.apache.openjpa.persistence.PersistenceProviderImpl"); |
| |
| if (createDatamodel) propMap.put("openjpa.jdbc.SynchronizeMappings", "buildSchema(ForeignKeys=false)"); |
| |
| _emf = Persistence.createEntityManagerFactory("ode-store", propMap); |
| } |
| |
| @SuppressWarnings("unchecked") |
| public ConfStoreConnection getConnection() { |
| HashMap propMap2 = new HashMap(); |
| propMap2.put("openjpa.TransactionMode", "managed"); |
| return new ConfStoreConnectionJpa(_emf.createEntityManager(propMap2)); |
| } |
| |
| public void beginTransaction() { |
| try { |
| if(__log.isDebugEnabled()) __log.debug("begin transaction on " + _txMgr); |
| _txMgr.begin(); |
| } catch (Exception e) { |
| throw new RuntimeException(e); |
| } |
| } |
| |
| public void commitTransaction() { |
| try { |
| if(__log.isDebugEnabled()) __log.debug("commit transaction on " + _txMgr); |
| _txMgr.commit(); |
| } catch (Exception e) { |
| throw new RuntimeException(e); |
| } |
| } |
| |
| public void rollbackTransaction() { |
| try { |
| if(__log.isDebugEnabled()) __log.debug("rollback transaction on " + _txMgr); |
| _txMgr.rollback(); |
| } catch (Exception e) { |
| throw new RuntimeException(e); |
| } |
| } |
| |
| @SuppressWarnings("unchecked") |
| private void initTxMgr(String txFactoryClassName) { |
| __log.info("ProcessStore initializing transaction manager using " + txFactoryClassName); |
| try { |
| Class txFactClass = getClass().getClassLoader().loadClass(txFactoryClassName); |
| Object txFact = txFactClass.newInstance(); |
| _txMgr = (TransactionManager) txFactClass.getMethod("getTransactionManager", (Class[]) null).invoke(txFact); |
| } catch (Exception e) { |
| __log.fatal("Couldn't initialize a transaction manager with factory: " + txFactoryClassName, e); |
| throw new RuntimeException("Couldn't initialize a transaction manager with factory: " + txFactoryClassName, e); |
| } |
| } |
| } |