ejb/openejb-extender/src/main/java/org/apache/aries/ejb/openejb/extender/RunningApplication.java - aries - Git at Google

 /**
  *  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.aries.ejb.openejb.extender;

 import java.util.ArrayList;
 import java.util.Collection;
 import java.util.Dictionary;
 import java.util.HashMap;
 import java.util.HashSet;
 import java.util.Hashtable;
 import java.util.List;
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.ConcurrentMap;

 import javax.naming.Context;
 import javax.naming.NamingException;
 import javax.persistence.EntityManagerFactory;

 import org.apache.aries.jpa.container.PersistenceUnitConstants;
 import org.apache.aries.jpa.container.context.PersistenceContextProvider;
 import org.apache.aries.util.AriesFrameworkUtil;
 import org.apache.aries.util.manifest.ManifestHeaderProcessor;
 import org.apache.aries.util.manifest.ManifestHeaderProcessor.NameValuePair;
 import org.apache.openejb.AppContext;
 import org.apache.openejb.BeanContext;
 import org.apache.openejb.ContainerType;
 import org.apache.openejb.assembler.classic.EnterpriseBeanInfo;
 import org.apache.openejb.assembler.classic.PersistenceContextReferenceInfo;
 import org.apache.openejb.assembler.classic.PersistenceUnitReferenceInfo;
 import org.apache.openejb.assembler.classic.ProxyInterfaceResolver;
 import org.apache.openejb.assembler.classic.ReferenceLocationInfo;
 import org.apache.openejb.jee.EnterpriseBean;
 import org.apache.openejb.persistence.JtaEntityManager;
 import org.osgi.framework.Bundle;
 import org.osgi.framework.ServiceReference;
 import org.osgi.framework.ServiceRegistration;
 import org.osgi.util.tracker.ServiceTracker;
 import org.osgi.util.tracker.ServiceTrackerCustomizer;

 public class RunningApplication implements ServiceTrackerCustomizer {

   private static final String NONE = "NONE";

   private final AppContext ctx;
   private final Bundle bundle;
   private final Collection<ServiceRegistration<?>> regs =
     new ArrayList<ServiceRegistration<?>>();

   private ServiceTracker tracker;

   private final ConcurrentMap<String, ConcurrentMap<Context, PersistenceUnitReferenceInfo>>
     unitRegistrations = new ConcurrentHashMap<String, ConcurrentMap<Context, PersistenceUnitReferenceInfo>>();

   private final ConcurrentMap<String, ConcurrentMap<Context, PersistenceContextReferenceInfo>>
     contextRegistrations = new ConcurrentHashMap<String, ConcurrentMap<Context, PersistenceContextReferenceInfo>>();

   public RunningApplication(AppContext context, Bundle bundle, List<EnterpriseBeanInfo> enterpriseBeans) {
     this.ctx = context;
     this.bundle = bundle;

     for(EnterpriseBeanInfo bean : enterpriseBeans) {
       for(PersistenceUnitReferenceInfo pui : bean.jndiEnc.persistenceUnitRefs) {
         ConcurrentMap<Context, PersistenceUnitReferenceInfo> map = unitRegistrations.
            get(pui.persistenceUnitName);

         if(map == null) {
           map = new ConcurrentHashMap<Context, PersistenceUnitReferenceInfo>();
           unitRegistrations.put(pui.persistenceUnitName, map);
         }

         for(BeanContext eb : ctx.getBeanContexts()) {
           if(eb.getEjbName().equals(bean.ejbName)){
             map.put(eb.getJndiContext(), pui);
             continue;
           }
         }
       }

       for(PersistenceContextReferenceInfo pci : bean.jndiEnc.persistenceContextRefs) {
         ConcurrentMap<Context, PersistenceContextReferenceInfo> map = contextRegistrations.
             get(pci.persistenceUnitName);

         if(map == null) {
           map = new ConcurrentHashMap<Context, PersistenceContextReferenceInfo>();
           contextRegistrations.put(pci.persistenceUnitName, map);
         }

         for(BeanContext eb : ctx.getBeanContexts()) {
           if(eb.getEjbName().equals(bean.ejbName)){
             map.put(eb.getJndiContext(), pci);
             continue;
           }
         }
       }
     }
   }

   public AppContext getCtx() {
     return ctx;
   }

   public void init() {

     tracker = new ServiceTracker(bundle.getBundleContext(),
         EntityManagerFactory.class.getName(), this);
     tracker.open();

     registerEJBs();
   }

   public void destroy() {
     tracker.close();
     for(ServiceRegistration<?> reg : regs) {
       AriesFrameworkUtil.safeUnregisterService(reg);
     }
   }


   private void registerEJBs() {

     Collection<String> names = new HashSet<String>();

     Dictionary<String, String> d = bundle.getHeaders();
     String valueOfExportEJBHeader = d.get("Export-EJB");

     if((valueOfExportEJBHeader == null)){
       return;
     }

     if(names.contains(NONE)){
       return;
     }

     List<NameValuePair> contentsOfExportEJBHeader = ManifestHeaderProcessor.parseExportString(valueOfExportEJBHeader);
     for(NameValuePair nvp:contentsOfExportEJBHeader){
       names.add(nvp.getName());
     }

     if(valueOfExportEJBHeader.trim().equals("")){
         names = new AllCollection<String>();
       }

     //Register our session beans
     for (BeanContext beanContext : ctx.getDeployments()) {
       String ejbName = beanContext.getEjbName();
       //Skip if not a Singleton or stateless bean
       ContainerType type = beanContext.getContainer().getContainerType();
       boolean register = type == ContainerType.SINGLETON || type == ContainerType.STATELESS;

       //Skip if not allowed name
       register &= names.contains(ejbName);

       if(!register) {
         continue;
       }

       if (beanContext.isLocalbean()) {

         BeanContext.BusinessLocalBeanHome home = beanContext.getBusinessLocalBeanHome();

         Dictionary<String, Object> props = new Hashtable<String, Object>();

         props.put("ejb.name", ejbName);
         props.put("ejb.type", getCasedType(type));
         regs.add(bundle.getBundleContext().registerService(beanContext.getBeanClass().getName(),
             new EJBServiceFactory(home), props));
       }


       for (Class<?> interfce : beanContext.getBusinessLocalInterfaces()) {

         BeanContext.BusinessLocalHome home = beanContext.getBusinessLocalHome(interfce);

         Dictionary<String, Object> props = new Hashtable<String, Object>();

         props.put("ejb.name", ejbName);
         props.put("ejb.type", getCasedType(type));
         regs.add(bundle.getBundleContext().registerService(interfce.getName(),
             new EJBServiceFactory(home), props));
       }

       for (Class<?> interfce : beanContext.getBusinessRemoteInterfaces()) {

         List<Class> interfaces = ProxyInterfaceResolver.getInterfaces(beanContext.getBeanClass(),
             interfce, beanContext.getBusinessRemoteInterfaces());
         BeanContext.BusinessRemoteHome home = beanContext.getBusinessRemoteHome(interfaces, interfce);

         Dictionary<String, Object> props = new Hashtable<String, Object>();

         props.put("sevice.exported.interfaces", interfce.getName());
         props.put("ejb.name", ejbName);
         props.put("ejb.type", getCasedType(type));
         regs.add(bundle.getBundleContext().registerService(interfce.getName(),
             new EJBServiceFactory(home), props));
       }
     }
   }

   private String getCasedType(ContainerType type) {
     String s = type.toString().substring(0,1).toUpperCase();
     s += type.toString().substring(1).toLowerCase();
     return s;
   }

   public Object addingService(ServiceReference reference) {

     if(isTrue(reference, PersistenceUnitConstants.CONTAINER_MANAGED_PERSISTENCE_UNIT) &&
        !!!isTrue(reference, PersistenceContextProvider.PROXY_FACTORY_EMF_ATTRIBUTE)) {

       Map<Context, PersistenceUnitReferenceInfo> pUnitRefs = unitRegistrations.
              get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));
       Map<Context, PersistenceContextReferenceInfo> pCtxRefs = contextRegistrations.
              get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));

       if(pUnitRefs == null) {
         pUnitRefs = new HashMap<Context, PersistenceUnitReferenceInfo>();
       }
       if(pCtxRefs == null) {
         pCtxRefs = new HashMap<Context, PersistenceContextReferenceInfo>();
       }

       if(pUnitRefs.size() > 0 || pCtxRefs.size() > 0) {

         EntityManagerFactory emf = (EntityManagerFactory)bundle.getBundleContext().getService(reference);

         for(Entry<Context, PersistenceUnitReferenceInfo> e : pUnitRefs.entrySet()) {
           try {
             e.getKey().bind(e.getValue().referenceName, emf);
           } catch (NamingException ex) {
             // TODO Auto-generated catch block
             ex.printStackTrace();
           }
         }

         for(Entry<Context, PersistenceContextReferenceInfo> e : pCtxRefs.entrySet()) {
           PersistenceContextReferenceInfo pci = e.getValue();
           try {
             e.getKey().bind(pci.referenceName, new JtaEntityManager((String)reference.getProperty(
                 PersistenceUnitConstants.OSGI_UNIT_NAME), AriesPersistenceContextIntegration.get(),
                 emf, pci.properties, pci.extended));
           } catch (NamingException ex) {
             // TODO Auto-generated catch block
             ex.printStackTrace();
           }
         }
         return emf;
       }
     }
     return null;
   }

   private boolean isTrue(ServiceReference reference,
       String key) {
     return Boolean.parseBoolean(String.valueOf(reference.getProperty(key)));
   }

   public void modifiedService(ServiceReference reference, Object service) {
     //No op
   }

   public void removedService(ServiceReference reference, Object service) {

     Map<Context, PersistenceUnitReferenceInfo> pUnitRefs = unitRegistrations.
         get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));
     Map<Context, PersistenceContextReferenceInfo> pCtxRefs = contextRegistrations.
         get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));

     if(pUnitRefs == null) {
       pUnitRefs = new HashMap<Context, PersistenceUnitReferenceInfo>();
     }
     if(pCtxRefs == null) {
       pCtxRefs = new HashMap<Context, PersistenceContextReferenceInfo>();
     }

     if(pUnitRefs.size() > 0 || pCtxRefs.size() > 0) {

       for(Entry<Context, PersistenceUnitReferenceInfo> e : pUnitRefs.entrySet()) {
         try {
           e.getKey().unbind(e.getValue().referenceName);
         } catch (NamingException ex) {
           // TODO Auto-generated catch block
           ex.printStackTrace();
         }
       }

       for(Entry<Context, PersistenceContextReferenceInfo> e : pCtxRefs.entrySet()) {
         PersistenceContextReferenceInfo pci = e.getValue();
         try {
           e.getKey().unbind(pci.referenceName);
         } catch (NamingException ex) {
           // TODO Auto-generated catch block
           ex.printStackTrace();
         }
       }
     }
   }
 }
	/**
	* 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.aries.ejb.openejb.extender;

	import java.util.ArrayList;
	import java.util.Collection;
	import java.util.Dictionary;
	import java.util.HashMap;
	import java.util.HashSet;
	import java.util.Hashtable;
	import java.util.List;
	import java.util.Map;
	import java.util.Map.Entry;
	import java.util.concurrent.ConcurrentHashMap;
	import java.util.concurrent.ConcurrentMap;

	import javax.naming.Context;
	import javax.naming.NamingException;
	import javax.persistence.EntityManagerFactory;

	import org.apache.aries.jpa.container.PersistenceUnitConstants;
	import org.apache.aries.jpa.container.context.PersistenceContextProvider;
	import org.apache.aries.util.AriesFrameworkUtil;
	import org.apache.aries.util.manifest.ManifestHeaderProcessor;
	import org.apache.aries.util.manifest.ManifestHeaderProcessor.NameValuePair;
	import org.apache.openejb.AppContext;
	import org.apache.openejb.BeanContext;
	import org.apache.openejb.ContainerType;
	import org.apache.openejb.assembler.classic.EnterpriseBeanInfo;
	import org.apache.openejb.assembler.classic.PersistenceContextReferenceInfo;
	import org.apache.openejb.assembler.classic.PersistenceUnitReferenceInfo;
	import org.apache.openejb.assembler.classic.ProxyInterfaceResolver;
	import org.apache.openejb.assembler.classic.ReferenceLocationInfo;
	import org.apache.openejb.jee.EnterpriseBean;
	import org.apache.openejb.persistence.JtaEntityManager;
	import org.osgi.framework.Bundle;
	import org.osgi.framework.ServiceReference;
	import org.osgi.framework.ServiceRegistration;
	import org.osgi.util.tracker.ServiceTracker;
	import org.osgi.util.tracker.ServiceTrackerCustomizer;

	public class RunningApplication implements ServiceTrackerCustomizer {

	private static final String NONE = "NONE";

	private final AppContext ctx;
	private final Bundle bundle;
	private final Collection<ServiceRegistration<?>> regs =
	new ArrayList<ServiceRegistration<?>>();

	private ServiceTracker tracker;

	private final ConcurrentMap<String, ConcurrentMap<Context, PersistenceUnitReferenceInfo>>
	unitRegistrations = new ConcurrentHashMap<String, ConcurrentMap<Context, PersistenceUnitReferenceInfo>>();

	private final ConcurrentMap<String, ConcurrentMap<Context, PersistenceContextReferenceInfo>>
	contextRegistrations = new ConcurrentHashMap<String, ConcurrentMap<Context, PersistenceContextReferenceInfo>>();

	public RunningApplication(AppContext context, Bundle bundle, List<EnterpriseBeanInfo> enterpriseBeans) {
	this.ctx = context;
	this.bundle = bundle;

	for(EnterpriseBeanInfo bean : enterpriseBeans) {
	for(PersistenceUnitReferenceInfo pui : bean.jndiEnc.persistenceUnitRefs) {
	ConcurrentMap<Context, PersistenceUnitReferenceInfo> map = unitRegistrations.
	get(pui.persistenceUnitName);

	if(map == null) {
	map = new ConcurrentHashMap<Context, PersistenceUnitReferenceInfo>();
	unitRegistrations.put(pui.persistenceUnitName, map);
	}

	for(BeanContext eb : ctx.getBeanContexts()) {
	if(eb.getEjbName().equals(bean.ejbName)){
	map.put(eb.getJndiContext(), pui);
	continue;
	}
	}
	}

	for(PersistenceContextReferenceInfo pci : bean.jndiEnc.persistenceContextRefs) {
	ConcurrentMap<Context, PersistenceContextReferenceInfo> map = contextRegistrations.
	get(pci.persistenceUnitName);

	if(map == null) {
	map = new ConcurrentHashMap<Context, PersistenceContextReferenceInfo>();
	contextRegistrations.put(pci.persistenceUnitName, map);
	}

	for(BeanContext eb : ctx.getBeanContexts()) {
	if(eb.getEjbName().equals(bean.ejbName)){
	map.put(eb.getJndiContext(), pci);
	continue;
	}
	}
	}
	}
	}

	public AppContext getCtx() {
	return ctx;
	}

	public void init() {

	tracker = new ServiceTracker(bundle.getBundleContext(),
	EntityManagerFactory.class.getName(), this);
	tracker.open();

	registerEJBs();
	}

	public void destroy() {
	tracker.close();
	for(ServiceRegistration<?> reg : regs) {
	AriesFrameworkUtil.safeUnregisterService(reg);
	}
	}



	private void registerEJBs() {

	Collection<String> names = new HashSet<String>();

	Dictionary<String, String> d = bundle.getHeaders();
	String valueOfExportEJBHeader = d.get("Export-EJB");

	if((valueOfExportEJBHeader == null)){
	return;
	}

	if(names.contains(NONE)){
	return;
	}

	List<NameValuePair> contentsOfExportEJBHeader = ManifestHeaderProcessor.parseExportString(valueOfExportEJBHeader);
	for(NameValuePair nvp:contentsOfExportEJBHeader){
	names.add(nvp.getName());
	}

	if(valueOfExportEJBHeader.trim().equals("")){
	names = new AllCollection<String>();
	}

	//Register our session beans
	for (BeanContext beanContext : ctx.getDeployments()) {
	String ejbName = beanContext.getEjbName();
	//Skip if not a Singleton or stateless bean
	ContainerType type = beanContext.getContainer().getContainerType();
	boolean register = type == ContainerType.SINGLETON \|\| type == ContainerType.STATELESS;

	//Skip if not allowed name
	register &= names.contains(ejbName);

	if(!register) {
	continue;
	}

	if (beanContext.isLocalbean()) {

	BeanContext.BusinessLocalBeanHome home = beanContext.getBusinessLocalBeanHome();

	Dictionary<String, Object> props = new Hashtable<String, Object>();

	props.put("ejb.name", ejbName);
	props.put("ejb.type", getCasedType(type));
	regs.add(bundle.getBundleContext().registerService(beanContext.getBeanClass().getName(),
	new EJBServiceFactory(home), props));
	}


	for (Class<?> interfce : beanContext.getBusinessLocalInterfaces()) {

	BeanContext.BusinessLocalHome home = beanContext.getBusinessLocalHome(interfce);

	Dictionary<String, Object> props = new Hashtable<String, Object>();

	props.put("ejb.name", ejbName);
	props.put("ejb.type", getCasedType(type));
	regs.add(bundle.getBundleContext().registerService(interfce.getName(),
	new EJBServiceFactory(home), props));
	}

	for (Class<?> interfce : beanContext.getBusinessRemoteInterfaces()) {

	List<Class> interfaces = ProxyInterfaceResolver.getInterfaces(beanContext.getBeanClass(),
	interfce, beanContext.getBusinessRemoteInterfaces());
	BeanContext.BusinessRemoteHome home = beanContext.getBusinessRemoteHome(interfaces, interfce);

	Dictionary<String, Object> props = new Hashtable<String, Object>();

	props.put("sevice.exported.interfaces", interfce.getName());
	props.put("ejb.name", ejbName);
	props.put("ejb.type", getCasedType(type));
	regs.add(bundle.getBundleContext().registerService(interfce.getName(),
	new EJBServiceFactory(home), props));
	}
	}
	}

	private String getCasedType(ContainerType type) {
	String s = type.toString().substring(0,1).toUpperCase();
	s += type.toString().substring(1).toLowerCase();
	return s;
	}

	public Object addingService(ServiceReference reference) {

	if(isTrue(reference, PersistenceUnitConstants.CONTAINER_MANAGED_PERSISTENCE_UNIT) &&
	!!!isTrue(reference, PersistenceContextProvider.PROXY_FACTORY_EMF_ATTRIBUTE)) {

	Map<Context, PersistenceUnitReferenceInfo> pUnitRefs = unitRegistrations.
	get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));
	Map<Context, PersistenceContextReferenceInfo> pCtxRefs = contextRegistrations.
	get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));

	if(pUnitRefs == null) {
	pUnitRefs = new HashMap<Context, PersistenceUnitReferenceInfo>();
	}
	if(pCtxRefs == null) {
	pCtxRefs = new HashMap<Context, PersistenceContextReferenceInfo>();
	}

	if(pUnitRefs.size() > 0 \|\| pCtxRefs.size() > 0) {

	EntityManagerFactory emf = (EntityManagerFactory)bundle.getBundleContext().getService(reference);

	for(Entry<Context, PersistenceUnitReferenceInfo> e : pUnitRefs.entrySet()) {
	try {
	e.getKey().bind(e.getValue().referenceName, emf);
	} catch (NamingException ex) {
	// TODO Auto-generated catch block
	ex.printStackTrace();
	}
	}

	for(Entry<Context, PersistenceContextReferenceInfo> e : pCtxRefs.entrySet()) {
	PersistenceContextReferenceInfo pci = e.getValue();
	try {
	e.getKey().bind(pci.referenceName, new JtaEntityManager((String)reference.getProperty(
	PersistenceUnitConstants.OSGI_UNIT_NAME), AriesPersistenceContextIntegration.get(),
	emf, pci.properties, pci.extended));
	} catch (NamingException ex) {
	// TODO Auto-generated catch block
	ex.printStackTrace();
	}
	}
	return emf;
	}
	}
	return null;
	}

	private boolean isTrue(ServiceReference reference,
	String key) {
	return Boolean.parseBoolean(String.valueOf(reference.getProperty(key)));
	}

	public void modifiedService(ServiceReference reference, Object service) {
	//No op
	}

	public void removedService(ServiceReference reference, Object service) {

	Map<Context, PersistenceUnitReferenceInfo> pUnitRefs = unitRegistrations.
	get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));
	Map<Context, PersistenceContextReferenceInfo> pCtxRefs = contextRegistrations.
	get(reference.getProperty(PersistenceUnitConstants.OSGI_UNIT_NAME));

	if(pUnitRefs == null) {
	pUnitRefs = new HashMap<Context, PersistenceUnitReferenceInfo>();
	}
	if(pCtxRefs == null) {
	pCtxRefs = new HashMap<Context, PersistenceContextReferenceInfo>();
	}

	if(pUnitRefs.size() > 0 \|\| pCtxRefs.size() > 0) {

	for(Entry<Context, PersistenceUnitReferenceInfo> e : pUnitRefs.entrySet()) {
	try {
	e.getKey().unbind(e.getValue().referenceName);
	} catch (NamingException ex) {
	// TODO Auto-generated catch block
	ex.printStackTrace();
	}
	}

	for(Entry<Context, PersistenceContextReferenceInfo> e : pCtxRefs.entrySet()) {
	PersistenceContextReferenceInfo pci = e.getValue();
	try {
	e.getKey().unbind(pci.referenceName);
	} catch (NamingException ex) {
	// TODO Auto-generated catch block
	ex.printStackTrace();
	}
	}
	}
	}
	}