/*
* 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.OpenEJBRuntimeException;
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 OpenEJBRuntimeException("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() {
return get();
}
public static WebBeansContext get()
{
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) return getWebBeansContext(appContexts);
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();
}
@Override
public void clear(Object key) {
contextMessage(getContext(), "clearing ");
getContext().clear();
}
}