hotfix/source/openejb-core/src/main/java/org/apache/openejb/cdi/CdiAppContextsService.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.util.LogCategory;
 import org.apache.openejb.util.Logger;
 import org.apache.webbeans.config.OWBLogConst;
 import org.apache.webbeans.config.WebBeansContext;
 import org.apache.webbeans.context.*;
 import org.apache.webbeans.el.ELContextStore;
 import org.apache.webbeans.spi.ContextsService;
 import org.apache.webbeans.web.context.ServletRequestContext;
 import org.apache.webbeans.web.context.SessionContextManager;

 import javax.enterprise.context.*;
 import javax.enterprise.context.spi.Context;
 import javax.inject.Singleton;
 import javax.servlet.ServletRequestEvent;
 import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpSession;
 import java.lang.annotation.Annotation;

 public class CdiAppContextsService extends AbstractContextsService implements ContextsService {

     private static final Logger logger = Logger.getInstance(LogCategory.OPENEJB.createChild("cdi"), CdiAppContextsService.class);
     private final ThreadLocal<RequestContext> requestContext = new ThreadLocal<RequestContext>();

     private final ThreadLocal<SessionContext> sessionContext = new ThreadLocal<SessionContext>();
     private final SessionContextManager sessionCtxManager = new SessionContextManager();

     /**
      * Conversation context manager
      */
     private final ThreadLocal<ConversationContext> conversationContext;

     private final DependentContext dependentContext = new DependentContext();

     private final ApplicationContext applicationContext = new ApplicationContext();

     private final SingletonContext singletonContext = new SingletonContext();

     public CdiAppContextsService() {
         this(WebBeansContext.currentInstance().getOpenWebBeansConfiguration().supportsConversation());
     }

     public CdiAppContextsService(boolean supportsConversation) {
         dependentContext.setActive(true);
         if (supportsConversation) {
             conversationContext = new ThreadLocal<ConversationContext>();
         } else {
             conversationContext = null;
         }
         applicationContext.setActive(true);
         singletonContext.setActive(true);
     }

     @Override
     public void init(Object initializeObject) {
         //Start application context
         startContext(ApplicationScoped.class, initializeObject);

         //Start signelton context
         startContext(Singleton.class, initializeObject);
     }

     public void destroy(Object destroyObject) {
 //        //Destroy application context
 //        endContext(ApplicationScoped.class, destroyObject);
 //
 //        //Destroy singleton context
 //        endContext(Singleton.class, destroyObject);


         //Remove thread locals
         //for preventing memory leaks
         requestContext.set(null);
         requestContext.remove();
         sessionContext.set(null);
         sessionContext.remove();

         if (null != conversationContext) {
             conversationContext.set(null);
             conversationContext.remove();
         }

     }

     @Override
     public void endContext(Class<? extends Annotation> scopeType, Object endParameters) {

         if (supportsContext(scopeType)) {
             if (scopeType.equals(RequestScoped.class)) {
                 destroyRequestContext();
             } else if (scopeType.equals(SessionScoped.class)) {
                 destroySessionContext((HttpSession) endParameters);
             } else if (scopeType.equals(ApplicationScoped.class)) {
                 destroyApplicationContext();
             } else if (scopeType.equals(Dependent.class)) {
                 // Do nothing
             } else if (scopeType.equals(Singleton.class)) {
                 destroySingletonContext();
             } else if (supportsConversation() && scopeType.equals(ConversationScoped.class)) {
                 destroyConversationContext();
             } else {
                 if (logger.isWarningEnabled()) {
                     logger.warning("CDI-OpenWebBeans container in OpenEJB does not support context scope "
                             + scopeType.getSimpleName()
                             + ". Scopes @Dependent, @RequestScoped, @ApplicationScoped and @Singleton are supported scope types");
                 }
             }
         }

     }

     @Override
     public Context getCurrentContext(Class<? extends Annotation> scopeType) {
         if (scopeType.equals(RequestScoped.class)) {
             return getRequestContext();
         } else if (scopeType.equals(SessionScoped.class)) {
             return getSessionContext();
         } else if (scopeType.equals(ApplicationScoped.class)) {
             return getApplicationContext();
         } else if (supportsConversation() && scopeType.equals(ConversationScoped.class)) {
             return getConversationContext();
         } else if (scopeType.equals(Dependent.class)) {
             return dependentContext;
         } else if (scopeType.equals(Singleton.class)) {
             return getSingletonContext();
         }

         return null;
     }

     @Override
     public void startContext(Class<? extends Annotation> scopeType, Object startParameter) throws ContextException {
         if (supportsContext(scopeType)) {
             if (scopeType.equals(RequestScoped.class)) {
                 initRequestContext((ServletRequestEvent) startParameter);
             } else if (scopeType.equals(SessionScoped.class)) {
                 initSessionContext((HttpSession) startParameter);
             } else if (scopeType.equals(ApplicationScoped.class)) {
             } else if (scopeType.equals(Dependent.class)) {
                 // Do nothing
             } else if (scopeType.equals(Singleton.class)) {
                 initSingletonContext();
             } else if (supportsConversation() && scopeType.equals(ConversationScoped.class)) {
                 initConversationContext((ConversationContext) startParameter);
             } else {
                 if (logger.isWarningEnabled()) {
                     logger.warning("CDI-OpenWebBeans container in OpenEJB does not support context scope "
                             + scopeType.getSimpleName()
                             + ". Scopes @Dependent, @RequestScoped, @ApplicationScoped and @Singleton are supported scope types");
                 }
             }
         }
     }

     @Override
     public boolean supportsContext(Class<? extends Annotation> scopeType) {
         if (scopeType.equals(RequestScoped.class)
                 || scopeType.equals(SessionScoped.class)
                 || scopeType.equals(ApplicationScoped.class)
                 || scopeType.equals(Dependent.class)
                 || scopeType.equals(Singleton.class)
                 || (scopeType.equals(ConversationScoped.class) && supportsConversation())) {
             return true;
         }

         return false;
     }

     private void initRequestContext(ServletRequestEvent event) {

         RequestContext rq = new ServletRequestContext();
         rq.setActive(true);

         requestContext.set(rq);// set thread local
         if (event != null) {
             HttpServletRequest request = (HttpServletRequest) event.getServletRequest();
             ((ServletRequestContext) rq).setServletRequest(request);

             if (request != null) {
                 //Re-initialize thread local for session
                 HttpSession session = request.getSession(false);

                 if (session != null) {
                     initSessionContext(session);
                 }

 //                //Init thread local application context
 //                initApplicationContext(event.getServletContext());
 //
 //                //Init thread local sigleton context
 //                initSingletonContext(event.getServletContext());
             }
         }
     }

     private void destroyRequestContext() {
         //Get context
         RequestContext context = getRequestContext();

         //Destroy context
         if (context != null) {
             context.destroy();
         }

         // clean up the EL caches after each request
         ELContextStore elStore = ELContextStore.getInstance(false);
         if (elStore != null) {
             elStore.destroyELContextStore();
         }

         //Clear thread locals
         requestContext.set(null);
         requestContext.remove();

         //Also clear application and singleton context
 //        applicationContext.set(null);
 //        applicationContext.remove();

         //Singleton context
 //        singletonContext.set(null);
 //        singletonContext.remove();

         //Conversation context
         if (null != conversationContext) {
             conversationContext.set(null);
             conversationContext.remove();
         }
     }

     /**
      * Creates the session context at the session start.
      *
      * @param session http session object
      */
     private void initSessionContext(HttpSession session) {
         if (session == null) {
             // no session -> no SessionContext
             return;
         }

         String sessionId = session.getId();
         //Current context
         SessionContext currentSessionContext = sessionCtxManager.getSessionContextWithSessionId(sessionId);

         //No current context
         if (currentSessionContext == null) {
             currentSessionContext = new SessionContext();
             sessionCtxManager.addNewSessionContext(sessionId, currentSessionContext);
         }
         //Activate
         currentSessionContext.setActive(true);

         //Set thread local
         sessionContext.set(currentSessionContext);
     }

     /**
      * Destroys the session context and all of its components at the end of the
      * session.
      *
      * @param session http session object
      */
     private void destroySessionContext(HttpSession session) {
         if (session != null) {
             //Get current session context
             SessionContext context = sessionContext.get();

             //Destroy context
             if (context != null) {
                 context.destroy();
             }

             //Clear thread locals
             sessionContext.set(null);
             sessionContext.remove();

             //Remove session from manager
             sessionCtxManager.destroySessionContextWithSessionId(session.getId());
         }
     }

     //we don't have initApplicationContext

     private void destroyApplicationContext() {
         // look for thread local
         // this can be set by initRequestContext
 //        this.currentApplicationContext.destroy();
     }

     private void initSingletonContext() {
     }

     private void destroySingletonContext() {
 //        this.currentSingletonContext.destroy();
     }

     /**
      * Initialize conversation context.
      *
      * @param context context
      */
     private void initConversationContext(ConversationContext context) {
         if (context == null) {
             if (conversationContext.get() == null) {
                 ConversationContext newContext = new ConversationContext();
                 newContext.setActive(true);

                 conversationContext.set(newContext);
             } else {
                 conversationContext.get().setActive(true);
             }

         } else {
             context.setActive(true);
             conversationContext.set(context);
         }
     }

     /**
      * Destroy conversation context.
      */
     private void destroyConversationContext() {
         ConversationContext context = getConversationContext();

         if (context != null) {
             context.destroy();
         }

         if (null != conversationContext) {
             conversationContext.set(null);
             conversationContext.remove();
         }
     }


     private RequestContext getRequestContext() {
         return requestContext.get();
     }

     private Context getSessionContext() {
         SessionContext context = sessionContext.get();
         if (context == null) {

             lazyStartSessionContext();
             context = sessionContext.get();
         }
         return context;
     }

     /**
      * Gets application context.
      *
      * @return application context
      */
     private ApplicationContext getApplicationContext() {
         return applicationContext;
     }

     /**
      * Gets singleton context.
      *
      * @return singleton context
      */
     private SingletonContext getSingletonContext() {
         return singletonContext;
     }

     /**
      * Get current conversation ctx.
      *
      * @return conversation context
      */
     private ConversationContext getConversationContext() {
         return conversationContext.get();
     }

     private Context lazyStartSessionContext() {

         if (logger.isDebugEnabled()) {
             logger.debug(">lazyStartSessionContext");
         }

         Context webContext = null;
         Context context = getCurrentContext(RequestScoped.class);
         if (context instanceof ServletRequestContext) {
             ServletRequestContext requestContext = (ServletRequestContext) context;
             HttpServletRequest servletRequest = requestContext.getServletRequest();
             if (null != servletRequest) { // this could be null if there is no active request context
                 try {
                     HttpSession currentSession = servletRequest.getSession();
                     initSessionContext(currentSession);
 //                    if (failoverService != null && failoverService.isSupportFailOver())
 //                    {
 //                        failoverService.sessionIsInUse(currentSession);
 //                    }

                     if (logger.isDebugEnabled()) {
                         logger.debug("Lazy SESSION context initialization SUCCESS");
                     }
                 } catch (Exception e) {
                     logger.error(OWBLogConst.ERROR_0013, e);
                 }

             } else {
                 logger.warning("Could NOT lazily initialize session context because NO active request context");
             }
         } else {
             logger.warning("Could NOT lazily initialize session context because of " + context + " RequestContext");
         }

         if (logger.isDebugEnabled()) {
             logger.debug("<lazyStartSessionContext " + webContext);
         }
         return webContext;
     }

     private boolean supportsConversation() {
         return conversationContext != null;
     }

 }
	/*
	* 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.util.LogCategory;
	import org.apache.openejb.util.Logger;
	import org.apache.webbeans.config.OWBLogConst;
	import org.apache.webbeans.config.WebBeansContext;
	import org.apache.webbeans.context.*;
	import org.apache.webbeans.el.ELContextStore;
	import org.apache.webbeans.spi.ContextsService;
	import org.apache.webbeans.web.context.ServletRequestContext;
	import org.apache.webbeans.web.context.SessionContextManager;

	import javax.enterprise.context.*;
	import javax.enterprise.context.spi.Context;
	import javax.inject.Singleton;
	import javax.servlet.ServletRequestEvent;
	import javax.servlet.http.HttpServletRequest;
	import javax.servlet.http.HttpSession;
	import java.lang.annotation.Annotation;

	public class CdiAppContextsService extends AbstractContextsService implements ContextsService {

	private static final Logger logger = Logger.getInstance(LogCategory.OPENEJB.createChild("cdi"), CdiAppContextsService.class);
	private final ThreadLocal<RequestContext> requestContext = new ThreadLocal<RequestContext>();

	private final ThreadLocal<SessionContext> sessionContext = new ThreadLocal<SessionContext>();
	private final SessionContextManager sessionCtxManager = new SessionContextManager();

	/**
	* Conversation context manager
	*/
	private final ThreadLocal<ConversationContext> conversationContext;

	private final DependentContext dependentContext = new DependentContext();

	private final ApplicationContext applicationContext = new ApplicationContext();

	private final SingletonContext singletonContext = new SingletonContext();

	public CdiAppContextsService() {
	this(WebBeansContext.currentInstance().getOpenWebBeansConfiguration().supportsConversation());
	}

	public CdiAppContextsService(boolean supportsConversation) {
	dependentContext.setActive(true);
	if (supportsConversation) {
	conversationContext = new ThreadLocal<ConversationContext>();
	} else {
	conversationContext = null;
	}
	applicationContext.setActive(true);
	singletonContext.setActive(true);
	}

	@Override
	public void init(Object initializeObject) {
	//Start application context
	startContext(ApplicationScoped.class, initializeObject);

	//Start signelton context
	startContext(Singleton.class, initializeObject);
	}

	public void destroy(Object destroyObject) {
	// //Destroy application context
	// endContext(ApplicationScoped.class, destroyObject);
	//
	// //Destroy singleton context
	// endContext(Singleton.class, destroyObject);


	//Remove thread locals
	//for preventing memory leaks
	requestContext.set(null);
	requestContext.remove();
	sessionContext.set(null);
	sessionContext.remove();

	if (null != conversationContext) {
	conversationContext.set(null);
	conversationContext.remove();
	}

	}

	@Override
	public void endContext(Class<? extends Annotation> scopeType, Object endParameters) {

	if (supportsContext(scopeType)) {
	if (scopeType.equals(RequestScoped.class)) {
	destroyRequestContext();
	} else if (scopeType.equals(SessionScoped.class)) {
	destroySessionContext((HttpSession) endParameters);
	} else if (scopeType.equals(ApplicationScoped.class)) {
	destroyApplicationContext();
	} else if (scopeType.equals(Dependent.class)) {
	// Do nothing
	} else if (scopeType.equals(Singleton.class)) {
	destroySingletonContext();
	} else if (supportsConversation() && scopeType.equals(ConversationScoped.class)) {
	destroyConversationContext();
	} else {
	if (logger.isWarningEnabled()) {
	logger.warning("CDI-OpenWebBeans container in OpenEJB does not support context scope "
	+ scopeType.getSimpleName()
	+ ". Scopes @Dependent, @RequestScoped, @ApplicationScoped and @Singleton are supported scope types");
	}
	}
	}

	}

	@Override
	public Context getCurrentContext(Class<? extends Annotation> scopeType) {
	if (scopeType.equals(RequestScoped.class)) {
	return getRequestContext();
	} else if (scopeType.equals(SessionScoped.class)) {
	return getSessionContext();
	} else if (scopeType.equals(ApplicationScoped.class)) {
	return getApplicationContext();
	} else if (supportsConversation() && scopeType.equals(ConversationScoped.class)) {
	return getConversationContext();
	} else if (scopeType.equals(Dependent.class)) {
	return dependentContext;
	} else if (scopeType.equals(Singleton.class)) {
	return getSingletonContext();
	}

	return null;
	}

	@Override
	public void startContext(Class<? extends Annotation> scopeType, Object startParameter) throws ContextException {
	if (supportsContext(scopeType)) {
	if (scopeType.equals(RequestScoped.class)) {
	initRequestContext((ServletRequestEvent) startParameter);
	} else if (scopeType.equals(SessionScoped.class)) {
	initSessionContext((HttpSession) startParameter);
	} else if (scopeType.equals(ApplicationScoped.class)) {
	} else if (scopeType.equals(Dependent.class)) {
	// Do nothing
	} else if (scopeType.equals(Singleton.class)) {
	initSingletonContext();
	} else if (supportsConversation() && scopeType.equals(ConversationScoped.class)) {
	initConversationContext((ConversationContext) startParameter);
	} else {
	if (logger.isWarningEnabled()) {
	logger.warning("CDI-OpenWebBeans container in OpenEJB does not support context scope "
	+ scopeType.getSimpleName()
	+ ". Scopes @Dependent, @RequestScoped, @ApplicationScoped and @Singleton are supported scope types");
	}
	}
	}
	}

	@Override
	public boolean supportsContext(Class<? extends Annotation> scopeType) {
	if (scopeType.equals(RequestScoped.class)
	\|\| scopeType.equals(SessionScoped.class)
	\|\| scopeType.equals(ApplicationScoped.class)
	\|\| scopeType.equals(Dependent.class)
	\|\| scopeType.equals(Singleton.class)
	\|\| (scopeType.equals(ConversationScoped.class) && supportsConversation())) {
	return true;
	}

	return false;
	}

	private void initRequestContext(ServletRequestEvent event) {

	RequestContext rq = new ServletRequestContext();
	rq.setActive(true);

	requestContext.set(rq);// set thread local
	if (event != null) {
	HttpServletRequest request = (HttpServletRequest) event.getServletRequest();
	((ServletRequestContext) rq).setServletRequest(request);

	if (request != null) {
	//Re-initialize thread local for session
	HttpSession session = request.getSession(false);

	if (session != null) {
	initSessionContext(session);
	}

	// //Init thread local application context
	// initApplicationContext(event.getServletContext());
	//
	// //Init thread local sigleton context
	// initSingletonContext(event.getServletContext());
	}
	}
	}

	private void destroyRequestContext() {
	//Get context
	RequestContext context = getRequestContext();

	//Destroy context
	if (context != null) {
	context.destroy();
	}

	// clean up the EL caches after each request
	ELContextStore elStore = ELContextStore.getInstance(false);
	if (elStore != null) {
	elStore.destroyELContextStore();
	}

	//Clear thread locals
	requestContext.set(null);
	requestContext.remove();

	//Also clear application and singleton context
	// applicationContext.set(null);
	// applicationContext.remove();

	//Singleton context
	// singletonContext.set(null);
	// singletonContext.remove();

	//Conversation context
	if (null != conversationContext) {
	conversationContext.set(null);
	conversationContext.remove();
	}
	}

	/**
	* Creates the session context at the session start.
	*
	* @param session http session object
	*/
	private void initSessionContext(HttpSession session) {
	if (session == null) {
	// no session -> no SessionContext
	return;
	}

	String sessionId = session.getId();
	//Current context
	SessionContext currentSessionContext = sessionCtxManager.getSessionContextWithSessionId(sessionId);

	//No current context
	if (currentSessionContext == null) {
	currentSessionContext = new SessionContext();
	sessionCtxManager.addNewSessionContext(sessionId, currentSessionContext);
	}
	//Activate
	currentSessionContext.setActive(true);

	//Set thread local
	sessionContext.set(currentSessionContext);
	}

	/**
	* Destroys the session context and all of its components at the end of the
	* session.
	*
	* @param session http session object
	*/
	private void destroySessionContext(HttpSession session) {
	if (session != null) {
	//Get current session context
	SessionContext context = sessionContext.get();

	//Destroy context
	if (context != null) {
	context.destroy();
	}

	//Clear thread locals
	sessionContext.set(null);
	sessionContext.remove();

	//Remove session from manager
	sessionCtxManager.destroySessionContextWithSessionId(session.getId());
	}
	}

	//we don't have initApplicationContext

	private void destroyApplicationContext() {
	// look for thread local
	// this can be set by initRequestContext
	// this.currentApplicationContext.destroy();
	}

	private void initSingletonContext() {
	}

	private void destroySingletonContext() {
	// this.currentSingletonContext.destroy();
	}

	/**
	* Initialize conversation context.
	*
	* @param context context
	*/
	private void initConversationContext(ConversationContext context) {
	if (context == null) {
	if (conversationContext.get() == null) {
	ConversationContext newContext = new ConversationContext();
	newContext.setActive(true);

	conversationContext.set(newContext);
	} else {
	conversationContext.get().setActive(true);
	}

	} else {
	context.setActive(true);
	conversationContext.set(context);
	}
	}

	/**
	* Destroy conversation context.
	*/
	private void destroyConversationContext() {
	ConversationContext context = getConversationContext();

	if (context != null) {
	context.destroy();
	}

	if (null != conversationContext) {
	conversationContext.set(null);
	conversationContext.remove();
	}
	}


	private RequestContext getRequestContext() {
	return requestContext.get();
	}

	private Context getSessionContext() {
	SessionContext context = sessionContext.get();
	if (context == null) {

	lazyStartSessionContext();
	context = sessionContext.get();
	}
	return context;
	}

	/**
	* Gets application context.
	*
	* @return application context
	*/
	private ApplicationContext getApplicationContext() {
	return applicationContext;
	}

	/**
	* Gets singleton context.
	*
	* @return singleton context
	*/
	private SingletonContext getSingletonContext() {
	return singletonContext;
	}

	/**
	* Get current conversation ctx.
	*
	* @return conversation context
	*/
	private ConversationContext getConversationContext() {
	return conversationContext.get();
	}

	private Context lazyStartSessionContext() {

	if (logger.isDebugEnabled()) {
	logger.debug(">lazyStartSessionContext");
	}

	Context webContext = null;
	Context context = getCurrentContext(RequestScoped.class);
	if (context instanceof ServletRequestContext) {
	ServletRequestContext requestContext = (ServletRequestContext) context;
	HttpServletRequest servletRequest = requestContext.getServletRequest();
	if (null != servletRequest) { // this could be null if there is no active request context
	try {
	HttpSession currentSession = servletRequest.getSession();
	initSessionContext(currentSession);
	// if (failoverService != null && failoverService.isSupportFailOver())
	// {
	// failoverService.sessionIsInUse(currentSession);
	// }

	if (logger.isDebugEnabled()) {
	logger.debug("Lazy SESSION context initialization SUCCESS");
	}
	} catch (Exception e) {
	logger.error(OWBLogConst.ERROR_0013, e);
	}

	} else {
	logger.warning("Could NOT lazily initialize session context because NO active request context");
	}
	} else {
	logger.warning("Could NOT lazily initialize session context because of " + context + " RequestContext");
	}

	if (logger.isDebugEnabled()) {
	logger.debug("<lazyStartSessionContext " + webContext);
	}
	return webContext;
	}

	private boolean supportsConversation() {
	return conversationContext != null;
	}

	}