Java EntityManagerFactoryBuilder类代码示例

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Override
public StoredPerProvider addingService(ServiceReference<PersistenceProvider> reference) {
    String providerName = (String)reference.getProperty(JAVAX_PERSISTENCE_PROVIDER);
    // FIXME should be set when creating the EMF was successful
    if (punit.getPersistenceProviderClassName() == null) {
        punit.setProviderClassName(providerName);
    }
    StoredPerProvider stored = new StoredPerProvider();
    LOGGER.info("Found provider for " + punit.getPersistenceUnitName() + " " + punit.getPersistenceProviderClassName());
    
    PersistenceProvider provider = context.getService(reference);

    createAndCloseDummyEMF(provider);

    stored.builder = new AriesEntityManagerFactoryBuilder(context, provider, reference.getBundle(), punit);
    Dictionary<String, ?> props = AriesEntityManagerFactoryBuilder.createBuilderProperties(punit, punit.getBundle());
    stored.reg = context.registerService(EntityManagerFactoryBuilder.class, stored.builder , props);
    return stored;
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
private boolean incompatibleExtender(Bundle bundle) {
   	
	List<BundleWire> requiredWires = bundle.adapt(BundleWiring.class)
			.getRequiredWires(OSGI_EXTENDER_NS);
	
	for(BundleWire bw : requiredWires) {
		BundleCapability capability = bw.getCapability();
		if(EntityManagerFactoryBuilder.JPA_CAPABILITY_NAME.equals(
				capability.getAttributes().get(OSGI_EXTENDER_NS))) {
			
			// If the persistence bundle requires a different revision for the 
			// JPA extender then we are incompatible, otherwise we are
			return !capability.getRevision().equals(wiring.getRevision());
		}
	}
	
	// If there is no requirement then we must assume that it's safe
	return false;
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
  public void testEntityManagerFactoryBuilderWithIncompletePersistenceUnitAddPlugin() throws Exception {
  	EntityManagerFactoryBuilder builder = getService(EntityManagerFactoryBuilder.class, "(osgi.unit.name=incompleteTestUnit2)", 1000);
  	
  	DataSourceFactory dsf = getService(DataSourceFactory.class, 
  			"(" + OSGI_JDBC_DRIVER_CLASS + "=org.apache.derby.jdbc.EmbeddedDriver)");
     
  	Properties jdbcProps = new Properties();
  	jdbcProps.setProperty("url", "jdbc:derby:memory:DSFTEST;create=true");
  	
  	Map<String, Object> props = new HashMap<String, Object>();
  	props.put("javax.persistence.nonJtaDataSource", dsf.createDataSource(jdbcProps));
  	props.put("javax.persistence.transactionType", RESOURCE_LOCAL.name());
  	
  	props.put("org.apache.aries.jpa.eclipselink.plugin.types", SessionCustomizerImpl.class);
  	props.put("eclipselink.session.customizer", SessionCustomizerImpl.class.getName());
  	
  	EntityManagerFactory emf = builder.createEntityManagerFactory(props);
  	emf.createEntityManager();
assertEquals("invoked", emf
  			.getProperties().get(CUSTOMIZER_CALLED));
  	
  }
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testCarEMFBuilderExternalDS() throws Exception {
	DataSourceFactory dsf = getService(DataSourceFactory.class, 
			"(" + OSGI_JDBC_DRIVER_CLASS + "=org.apache.derby.jdbc.EmbeddedDriver)");
   
	EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
			"(osgi.unit.name=" + EXTERNAL_TEST_UNIT + ")");
	
    assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
	
	Properties jdbcProps = new Properties();
	jdbcProps.setProperty("url", "jdbc:derby:memory:DSFTEST;create=true");
	
	Map<String, Object> props = new HashMap<String, Object>();
	props.put("javax.persistence.nonJtaDataSource", dsf.createDataSource(jdbcProps));
	props.put("javax.persistence.transactionType", RESOURCE_LOCAL.name());
	
	EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
	carLifecycleRL(emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testTruckEMFBuilderExternalDS() throws Exception {
	DataSourceFactory dsf = getService(DataSourceFactory.class, 
			"(" + OSGI_JDBC_DRIVER_CLASS + "=org.apache.derby.jdbc.EmbeddedDriver)");
	
	EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
			"(osgi.unit.name=" + EXTERNAL_TEST_UNIT + ")");
	 
    assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
    
	Properties jdbcProps = new Properties();
	jdbcProps.setProperty("url", "jdbc:derby:memory:DSFTEST;create=true");
	
	Map<String, Object> props = new HashMap<String, Object>();
	props.put("javax.persistence.nonJtaDataSource", dsf.createDataSource(jdbcProps));
	props.put("javax.persistence.transactionType", RESOURCE_LOCAL.name());
	
	EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
	truckLifecycleRL(emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testCarEMFBuilderExternalDSXA() throws Exception {
	EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
			"(osgi.unit.name=" + EXTERNAL_TEST_UNIT + ")");
	 
    assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
    
	Map<String, Object> props = new HashMap<String, Object>();
	props.put("javax.persistence.jtaDataSource", ds);
	props.put("javax.persistence.transactionType", JTA.name());
	//EclipseLink also needs a non-jta-datasource
	props.put("javax.persistence.nonJtaDataSource", ds);
	EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
	carLifecycleXA(ut, emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testTruckEMFBuilderExternalDSXA() throws Exception {
	EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
			"(osgi.unit.name=" + EXTERNAL_TEST_UNIT + ")");
	 
    assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
    
	Map<String, Object> props = new HashMap<String, Object>();
	props.put("javax.persistence.jtaDataSource", ds);
	props.put("javax.persistence.transactionType", JTA.name());
	//EclipseLink also needs a non-jta-datasource
	props.put("javax.persistence.nonJtaDataSource", ds);
	EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
	truckLifecycleXA(ut, emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testCarEMFBuilderNoNonJTADataSource() throws Exception {
    EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
                    "(osgi.unit.name=" + EXTERNAL_TEST_UNIT + ")");
    
    assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
    
    Map<String, Object> props = new HashMap<String, Object>();
    Properties jdbcProps = new Properties();
    jdbcProps.setProperty("url", "jdbc:derby:memory:TESTNOJTA;create=true");
    props.put("javax.persistence.dataSource", dsf.createDataSource(jdbcProps));
    props.put("javax.persistence.transactionType", RESOURCE_LOCAL.name());
    
    EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
    carLifecycleRL(emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testTruckEMFBuilderNoNonJTADataSource() throws Exception {
	EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
			"(osgi.unit.name=" + EXTERNAL_TEST_UNIT + ")");
	 
    assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
    
	Map<String, Object> props = new HashMap<String, Object>();
	Properties jdbcProps = new Properties();
	jdbcProps.setProperty("url", "jdbc:derby:memory:TESTNOJTA;create=true");
	props.put("javax.persistence.dataSource", dsf.createDataSource(jdbcProps));
	props.put("javax.persistence.transactionType", RESOURCE_LOCAL.name());
	
	EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
	truckLifecycleRL(emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
private void create(Bundle bundle, Context context) {
    Map.Entry<String, PersistenceProvider> pp = provider.get(context.definition.provider);
    // Do the registration of the service asynchronously. Since it may imply all kinds of
    // listening actions performed as a result of it, it may block the bundle handling.
    // However, indicate that we are in the process of the actions by
    // setting the used provider.
    context.usedProvider = pp.getKey();
    new Thread(() -> {
        synchronized (context) {
            if (context.usedProvider != null) {
                PersistenceUnitInfo info = PersistenceUnitProcessor.getPersistenceUnitInfo(bundle,
                        context.definition, pp.getValue());
                context.factory = PersistenceUnitProcessor.createFactory(pp.getValue(), info);
                Hashtable<String, Object> props = new Hashtable<>();
                props.put(EntityManagerFactoryBuilder.JPA_UNIT_NAME, context.definition.name);
                props.put(PersistenceUnitTransactionType.class.getName(), info.getTransactionType().name());
                context.registration = bundle.getBundleContext().registerService(
                        EntityManagerFactory.class, context.factory, props);
            }
        }
    }).start();
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@SuppressWarnings("unchecked")
@Test
public void testLifecycle() {
    BundleContext context = mock(BundleContext.class);
    Coordinator coordinator = mock(Coordinator.class);
    EMFTracker tracker = new EMFTracker(context, coordinator);
    ServiceReference<EntityManagerFactory> ref = mock(ServiceReference.class);
    Mockito.when(ref.getProperty(EntityManagerFactoryBuilder.JPA_UNIT_NAME)).thenReturn("testunit");
    Mockito.when(ref.getProperty(PersistenceUnitTransactionType.class.getName())).thenReturn("JTA");
    
    Bundle puBundle = mock(Bundle.class);
    BundleContext puContext = mock(BundleContext.class);
    when(puBundle.getBundleContext()).thenReturn(puContext);
    when(ref.getBundle()).thenReturn(puBundle);
    EntityManagerFactory emf = mock(EntityManagerFactory.class);
    when(puContext.getService(ref)).thenReturn(emf);
    ServiceRegistration<?> emSupplierReg = mock(ServiceRegistration.class, "emSupplierReg");
    ServiceRegistration<?> emProxyReg = mock(ServiceRegistration.class, "emProxyReg");
    when(puContext.registerService(any(Class.class), any(), any(Dictionary.class)))
        .thenReturn(emSupplierReg, emProxyReg);

    EMFTracker.TrackedEmf tracked = (TrackedEmf)tracker.addingService(ref);
    Assert.assertEquals(emf, tracked.emf);
    Assert.assertEquals(emSupplierReg, tracked.emSupplierReg);
    Assert.assertEquals(emProxyReg, tracked.emProxyReg);
    Assert.assertNotNull(tracked.tmTracker);
    Assert.assertNull(tracked.rlTxManagerReg);
    
    tracker.removedService(ref, tracked);
    verify(emSupplierReg, times(1)).unregister();
    verify(emProxyReg, times(1)).unregister();
    verify(puContext, times(1)).ungetService(ref);
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testCarEMFBuilder() throws Exception {
    EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
                                                        "(osgi.unit.name=" + DSF_TEST_UNIT + ")");
    
    assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
    
    Map<String, Object> props = new HashMap<String, Object>();
    EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
    carLifecycleRL(emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Test
public void testTruckEMFBuilder() throws Exception {
	EntityManagerFactoryBuilder emfBuilder = getService(EntityManagerFactoryBuilder.class,
			"(osgi.unit.name=" + DSF_TEST_UNIT + ")");

	assertEquals(getProviderClassName(), emfBuilder.getPersistenceProviderName());
	assertEquals(getProviderBundle(), emfBuilder.getPersistenceProviderBundle());
	
	Map<String, Object> props = new HashMap<String, Object>();
	EntityManagerFactory emf = emfBuilder.createEntityManagerFactory(props);
	truckLifecycleRL(emf.createEntityManager());
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
void registerJpaCrudRepository(String unitName, EntityManagerFactory emf) {
    Dictionary<String, String> properties = new Hashtable<>();
    properties.put(SERVICE_VENDOR, "AdeptJ");
    properties.put(SERVICE_PID, EclipseLinkCrudRepository.class.getName());
    properties.put(SERVICE_DESCRIPTION, "AdeptJ Modules JpaCrudRepository(EclipseLink)");
    properties.put(EntityManagerFactoryBuilder.JPA_UNIT_NAME, unitName);
    LOGGER.info("Registering JpaCrudRepository For PersistenceUnit: [{}]", unitName);
    this.jpaCrudRepositories.put(unitName, this.context.registerService(JpaCrudRepository.class,
            new EclipseLinkCrudRepository(emf), properties));
    this.entityManagerFactories.put(unitName, emf);
}
 

import org.osgi.service.jpa.EntityManagerFactoryBuilder; //导入依赖的package包/类
@Override
public void start(BundleContext context) {
    TransactionController.initialize(context);
    OurPersistenceProvider provider = new OurPersistenceProvider();
    Hashtable<String, Object> dict = new Hashtable<>();
    dict.put( EntityManagerFactoryBuilder.JPA_UNIT_PROVIDER, "org.eclipse.persistence.jpa.PersistenceProvider");
    context.registerService(PersistenceProvider.class, provider, dict);
}