hotfix/source/openejb-core/src/main/java/org/apache/openejb/cdi/ThreadSingletonServiceImpl.java - virgo/org.eclipse.virgo.eclipse-mirror - 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.openejb.cdi;

 import org.apache.openejb.AppContext;
 import org.apache.openejb.assembler.classic.AppInfo;
 import org.apache.openejb.assembler.classic.EjbJarInfo;
 import org.apache.openejb.loader.SystemInstance;
 import org.apache.openejb.spi.ContainerSystem;
 import org.apache.openejb.util.LogCategory;
 import org.apache.openejb.util.Logger;
 import org.apache.openejb.util.classloader.MultipleClassLoader;
 import org.apache.webbeans.config.OpenWebBeansConfiguration;
 import org.apache.webbeans.config.WebBeansContext;
 import org.apache.webbeans.spi.ContainerLifecycle;
 import org.apache.webbeans.spi.ContextsService;
 import org.apache.webbeans.spi.ConversationService;
 import org.apache.webbeans.spi.LoaderService;
 import org.apache.webbeans.spi.ResourceInjectionService;
 import org.apache.webbeans.spi.ScannerService;
 import org.apache.webbeans.spi.SecurityService;
 import org.apache.webbeans.spi.TransactionService;
 import org.apache.webbeans.spi.adaptor.ELAdaptor;

 import java.util.Collections;
 import java.util.Comparator;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 import java.util.Properties;

 /**
  * @version $Rev:$ $Date:$
  */
 public class ThreadSingletonServiceImpl implements ThreadSingletonService {

     public static final Logger logger = Logger.getInstance(LogCategory.OPENEJB_STARTUP, ThreadSingletonServiceImpl.class);
     //this needs to be static because OWB won't tell us what the existing SingletonService is and you can't set it twice.
     private static final ThreadLocal<WebBeansContext> contexts = new ThreadLocal<WebBeansContext>();

     public ThreadSingletonServiceImpl() {

     }


     @Override
     public void initialize(StartupObject startupObject) {
         AppContext appContext = startupObject.getAppContext();

         appContext.setCdiEnabled(hasBeans(startupObject.getAppInfo()));

         //initialize owb context, cf geronimo's OpenWebBeansGBean
         Properties properties = new Properties();
         Map<Class<?>, Object> services = new HashMap<Class<?>, Object>();
         properties.setProperty(OpenWebBeansConfiguration.APPLICATION_IS_JSP, "true");
         properties.setProperty(OpenWebBeansConfiguration.USE_EJB_DISCOVERY, "true");
         //from CDI builder
         properties.setProperty(OpenWebBeansConfiguration.INTERCEPTOR_FORCE_NO_CHECKED_EXCEPTIONS, "false");
         properties.setProperty(SecurityService.class.getName(), ManagedSecurityService.class.getName());
         properties.setProperty(OpenWebBeansConfiguration.CONVERSATION_PERIODIC_DELAY, "1800000");
         properties.setProperty(OpenWebBeansConfiguration.APPLICATION_SUPPORTS_CONVERSATION, "true");
         properties.setProperty(OpenWebBeansConfiguration.IGNORED_INTERFACES, "org.apache.aries.proxy.weaving.WovenProxy");

         services.put(AppContext.class, appContext);
         services.put(TransactionService.class, new OpenEJBTransactionService());
         services.put(ELAdaptor.class,new CustomELAdapter(appContext));
         services.put(ContextsService.class, new CdiAppContextsService(true));
         services.put(ResourceInjectionService.class, new CdiResourceInjectionService());
         services.put(ScannerService.class, new CdiScanner());
         services.put(LoaderService.class, new OptimizedLoaderService());

         optional(services, ConversationService.class, "org.apache.webbeans.jsf.DefaultConversationService");

         ClassLoader oldClassLoader = Thread.currentThread().getContextClassLoader();
         ClassLoader cl;
         if (oldClassLoader != ThreadSingletonServiceImpl.class.getClassLoader()) {
             cl = new MultipleClassLoader(oldClassLoader, ThreadSingletonServiceImpl.class.getClassLoader());
         } else {
             cl = oldClassLoader;
         }
         Thread.currentThread().setContextClassLoader(cl);
         WebBeansContext webBeansContext;
         Object old = null;
         try {
             webBeansContext = new WebBeansContext(services, properties);
             appContext.set(WebBeansContext.class, webBeansContext);
             old = contextEntered(webBeansContext);
             setConfiguration(webBeansContext.getOpenWebBeansConfiguration());
             try {
                 webBeansContext.getService(ContainerLifecycle.class).startApplication(startupObject);
             } catch (Exception e) {
                 throw new RuntimeException("couldn't start owb context", e);
             }
         } finally {
             contextExited(old);
             Thread.currentThread().setContextClassLoader(oldClassLoader);
         }
     }

     private boolean hasBeans(AppInfo appInfo) {
         for (EjbJarInfo ejbJar : appInfo.ejbJars) {
             if (ejbJar.beans != null) return true;
         }
         return false;
     }

     private <T> void optional(Map<Class<?>, Object> services, Class<T> type, String implementation) {
         try {
             Class clazz = this.getClass().getClassLoader().loadClass(implementation);
             services.put(type, type.cast(clazz.newInstance()));

             logger.debug(String.format("CDI Service Installed: %s = %s", type.getName(), implementation));
         } catch (ClassNotFoundException e) {
             logger.debug(String.format("CDI Service not installed: %s", type.getName()));
         } catch (Exception e) {
             throw new IllegalStateException(e);
         }
     }

     //not sure what openejb will need

     private void setConfiguration(OpenWebBeansConfiguration configuration) {
         //from CDI builder
         configuration.setProperty(SecurityService.class.getName(), ManagedSecurityService.class.getName());
         configuration.setProperty(OpenWebBeansConfiguration.INTERCEPTOR_FORCE_NO_CHECKED_EXCEPTIONS, "false");
 //        configuration.setProperty(OpenWebBeansConfiguration.APPLICATION_IS_JSP, "true");

         configuration.setProperty(OpenWebBeansConfiguration.CONTAINER_LIFECYCLE, OpenEJBLifecycle.class.getName());
         configuration.setProperty(OpenWebBeansConfiguration.TRANSACTION_SERVICE, OpenEJBTransactionService.class.getName());
         configuration.setProperty(OpenWebBeansConfiguration.SCANNER_SERVICE, CdiScanner.class.getName());
         configuration.setProperty(OpenWebBeansConfiguration.CONTEXTS_SERVICE, CdiAppContextsService.class.getName());
         configuration.setProperty(OpenWebBeansConfiguration.VALIDATOR_SERVICE, OpenEJBValidatorService.class.getName());
         configuration.setProperty(ResourceInjectionService.class.getName(), CdiResourceInjectionService.class.getName());
     }

     @Override
     public Object contextEntered(WebBeansContext newOWBContext) {
         return enter(newOWBContext);
     }

     public static WebBeansContext enter(WebBeansContext newOWBContext) {
         WebBeansContext oldContext = contexts.get();
         contexts.set(newOWBContext);
         contextMessage(newOWBContext, "Enter:");
         return oldContext;
     }

     private static void contextMessage(WebBeansContext newOWBContext, String prefix) {
     }

     @Override
     public void contextExited(Object oldContext) {
         exit(oldContext);
     }

     public static void exit(Object oldContext) {
         if (oldContext != null && !(oldContext instanceof WebBeansContext)) throw new IllegalArgumentException("ThreadSingletonServiceImpl can only be used with WebBeansContext, not " + oldContext.getClass().getName());
         contexts.set((WebBeansContext) oldContext);
     }

     private WebBeansContext getContext(Object key) {
         return get((ClassLoader)key);
     }

     public static WebBeansContext get(ClassLoader classLoader)
     {
         WebBeansContext context = contexts.get();
         if (context == null) {
             // Fallback strategy is to just grab the first AppContext and assume it is the right one
             // This kind of algorithm could be greatly improved
             final ContainerSystem containerSystem = SystemInstance.get().getComponent(ContainerSystem.class);

             final List<AppContext> appContexts = containerSystem.getAppContexts();

             if (appContexts.size() > 0) {
             	for(AppContext currentAppContext: appContexts) {
             		if(classLoader == currentAppContext.getClassLoader())
             			return currentAppContext.getWebBeansContext();
             	}
             }

             throw new IllegalStateException("On a thread without an initialized context");
         }
         return context;
     }

     private static WebBeansContext getWebBeansContext(List<AppContext> appContexts) {
         Collections.sort(appContexts, new Comparator<AppContext>() {
             @Override
             public int compare(AppContext appContext, AppContext appContext1) {
                 return appContext1.getWebBeansContext().getBeanManagerImpl().getBeans().size() - appContext.getWebBeansContext().getBeanManagerImpl().getBeans().size();
             }
         });
         return appContexts.get(0).getWebBeansContext();
     }

     @Override
     public WebBeansContext get(Object key) {
         return getContext(key);
     }

     @Override
     public void clear(Object key) {
         contextMessage(getContext(key), "clearing ");
         getContext(key).clear();
     }

 }
	/*
	* 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.openejb.cdi;

	import org.apache.openejb.AppContext;
	import org.apache.openejb.assembler.classic.AppInfo;
	import org.apache.openejb.assembler.classic.EjbJarInfo;
	import org.apache.openejb.loader.SystemInstance;
	import org.apache.openejb.spi.ContainerSystem;
	import org.apache.openejb.util.LogCategory;
	import org.apache.openejb.util.Logger;
	import org.apache.openejb.util.classloader.MultipleClassLoader;
	import org.apache.webbeans.config.OpenWebBeansConfiguration;
	import org.apache.webbeans.config.WebBeansContext;
	import org.apache.webbeans.spi.ContainerLifecycle;
	import org.apache.webbeans.spi.ContextsService;
	import org.apache.webbeans.spi.ConversationService;
	import org.apache.webbeans.spi.LoaderService;
	import org.apache.webbeans.spi.ResourceInjectionService;
	import org.apache.webbeans.spi.ScannerService;
	import org.apache.webbeans.spi.SecurityService;
	import org.apache.webbeans.spi.TransactionService;
	import org.apache.webbeans.spi.adaptor.ELAdaptor;

	import java.util.Collections;
	import java.util.Comparator;
	import java.util.HashMap;
	import java.util.List;
	import java.util.Map;
	import java.util.Properties;

	/**
	* @version $Rev:$ $Date:$
	*/
	public class ThreadSingletonServiceImpl implements ThreadSingletonService {

	public static final Logger logger = Logger.getInstance(LogCategory.OPENEJB_STARTUP, ThreadSingletonServiceImpl.class);
	//this needs to be static because OWB won't tell us what the existing SingletonService is and you can't set it twice.
	private static final ThreadLocal<WebBeansContext> contexts = new ThreadLocal<WebBeansContext>();

	public ThreadSingletonServiceImpl() {

	}



	@Override
	public void initialize(StartupObject startupObject) {
	AppContext appContext = startupObject.getAppContext();

	appContext.setCdiEnabled(hasBeans(startupObject.getAppInfo()));

	//initialize owb context, cf geronimo's OpenWebBeansGBean
	Properties properties = new Properties();
	Map<Class<?>, Object> services = new HashMap<Class<?>, Object>();
	properties.setProperty(OpenWebBeansConfiguration.APPLICATION_IS_JSP, "true");
	properties.setProperty(OpenWebBeansConfiguration.USE_EJB_DISCOVERY, "true");
	//from CDI builder
	properties.setProperty(OpenWebBeansConfiguration.INTERCEPTOR_FORCE_NO_CHECKED_EXCEPTIONS, "false");
	properties.setProperty(SecurityService.class.getName(), ManagedSecurityService.class.getName());
	properties.setProperty(OpenWebBeansConfiguration.CONVERSATION_PERIODIC_DELAY, "1800000");
	properties.setProperty(OpenWebBeansConfiguration.APPLICATION_SUPPORTS_CONVERSATION, "true");
	properties.setProperty(OpenWebBeansConfiguration.IGNORED_INTERFACES, "org.apache.aries.proxy.weaving.WovenProxy");

	services.put(AppContext.class, appContext);
	services.put(TransactionService.class, new OpenEJBTransactionService());
	services.put(ELAdaptor.class,new CustomELAdapter(appContext));
	services.put(ContextsService.class, new CdiAppContextsService(true));
	services.put(ResourceInjectionService.class, new CdiResourceInjectionService());
	services.put(ScannerService.class, new CdiScanner());
	services.put(LoaderService.class, new OptimizedLoaderService());

	optional(services, ConversationService.class, "org.apache.webbeans.jsf.DefaultConversationService");

	ClassLoader oldClassLoader = Thread.currentThread().getContextClassLoader();
	ClassLoader cl;
	if (oldClassLoader != ThreadSingletonServiceImpl.class.getClassLoader()) {
	cl = new MultipleClassLoader(oldClassLoader, ThreadSingletonServiceImpl.class.getClassLoader());
	} else {
	cl = oldClassLoader;
	}
	Thread.currentThread().setContextClassLoader(cl);
	WebBeansContext webBeansContext;
	Object old = null;
	try {
	webBeansContext = new WebBeansContext(services, properties);
	appContext.set(WebBeansContext.class, webBeansContext);
	old = contextEntered(webBeansContext);
	setConfiguration(webBeansContext.getOpenWebBeansConfiguration());
	try {
	webBeansContext.getService(ContainerLifecycle.class).startApplication(startupObject);
	} catch (Exception e) {
	throw new RuntimeException("couldn't start owb context", e);
	}
	} finally {
	contextExited(old);
	Thread.currentThread().setContextClassLoader(oldClassLoader);
	}
	}

	private boolean hasBeans(AppInfo appInfo) {
	for (EjbJarInfo ejbJar : appInfo.ejbJars) {
	if (ejbJar.beans != null) return true;
	}
	return false;
	}

	private <T> void optional(Map<Class<?>, Object> services, Class<T> type, String implementation) {
	try {
	Class clazz = this.getClass().getClassLoader().loadClass(implementation);
	services.put(type, type.cast(clazz.newInstance()));

	logger.debug(String.format("CDI Service Installed: %s = %s", type.getName(), implementation));
	} catch (ClassNotFoundException e) {
	logger.debug(String.format("CDI Service not installed: %s", type.getName()));
	} catch (Exception e) {
	throw new IllegalStateException(e);
	}
	}

	//not sure what openejb will need

	private void setConfiguration(OpenWebBeansConfiguration configuration) {
	//from CDI builder
	configuration.setProperty(SecurityService.class.getName(), ManagedSecurityService.class.getName());
	configuration.setProperty(OpenWebBeansConfiguration.INTERCEPTOR_FORCE_NO_CHECKED_EXCEPTIONS, "false");
	// configuration.setProperty(OpenWebBeansConfiguration.APPLICATION_IS_JSP, "true");

	configuration.setProperty(OpenWebBeansConfiguration.CONTAINER_LIFECYCLE, OpenEJBLifecycle.class.getName());
	configuration.setProperty(OpenWebBeansConfiguration.TRANSACTION_SERVICE, OpenEJBTransactionService.class.getName());
	configuration.setProperty(OpenWebBeansConfiguration.SCANNER_SERVICE, CdiScanner.class.getName());
	configuration.setProperty(OpenWebBeansConfiguration.CONTEXTS_SERVICE, CdiAppContextsService.class.getName());
	configuration.setProperty(OpenWebBeansConfiguration.VALIDATOR_SERVICE, OpenEJBValidatorService.class.getName());
	configuration.setProperty(ResourceInjectionService.class.getName(), CdiResourceInjectionService.class.getName());
	}

	@Override
	public Object contextEntered(WebBeansContext newOWBContext) {
	return enter(newOWBContext);
	}

	public static WebBeansContext enter(WebBeansContext newOWBContext) {
	WebBeansContext oldContext = contexts.get();
	contexts.set(newOWBContext);
	contextMessage(newOWBContext, "Enter:");
	return oldContext;
	}

	private static void contextMessage(WebBeansContext newOWBContext, String prefix) {
	}

	@Override
	public void contextExited(Object oldContext) {
	exit(oldContext);
	}

	public static void exit(Object oldContext) {
	if (oldContext != null && !(oldContext instanceof WebBeansContext)) throw new IllegalArgumentException("ThreadSingletonServiceImpl can only be used with WebBeansContext, not " + oldContext.getClass().getName());
	contexts.set((WebBeansContext) oldContext);
	}

	private WebBeansContext getContext(Object key) {
	return get((ClassLoader)key);
	}

	public static WebBeansContext get(ClassLoader classLoader)
	{
	WebBeansContext context = contexts.get();
	if (context == null) {
	// Fallback strategy is to just grab the first AppContext and assume it is the right one
	// This kind of algorithm could be greatly improved
	final ContainerSystem containerSystem = SystemInstance.get().getComponent(ContainerSystem.class);

	final List<AppContext> appContexts = containerSystem.getAppContexts();

	if (appContexts.size() > 0) {
	for(AppContext currentAppContext: appContexts) {
	if(classLoader == currentAppContext.getClassLoader())
	return currentAppContext.getWebBeansContext();
	}
	}

	throw new IllegalStateException("On a thread without an initialized context");
	}
	return context;
	}

	private static WebBeansContext getWebBeansContext(List<AppContext> appContexts) {
	Collections.sort(appContexts, new Comparator<AppContext>() {
	@Override
	public int compare(AppContext appContext, AppContext appContext1) {
	return appContext1.getWebBeansContext().getBeanManagerImpl().getBeans().size() - appContext.getWebBeansContext().getBeanManagerImpl().getBeans().size();
	}
	});
	return appContexts.get(0).getWebBeansContext();
	}

	@Override
	public WebBeansContext get(Object key) {
	return getContext(key);
	}

	@Override
	public void clear(Object key) {
	contextMessage(getContext(key), "clearing ");
	getContext(key).clear();
	}

	}