WebSocketScopeContext.java

//
//  ========================================================================
//  Copyright (c) 1995-2017 Mort Bay Consulting Pty. Ltd.
//  ------------------------------------------------------------------------
//  All rights reserved. This program and the accompanying materials
//  are made available under the terms of the Eclipse Public License v1.0
//  and Apache License v2.0 which accompanies this distribution.
//
//      The Eclipse Public License is available at
//      http://www.eclipse.org/legal/epl-v10.html
//
//      The Apache License v2.0 is available at
//      http://www.opensource.org/licenses/apache2.0.php
//
//  You may elect to redistribute this code under either of these licenses.
//  ========================================================================
//

package org.eclipse.jetty.cdi.websocket;

import java.lang.annotation.Annotation;
import java.util.List;
import java.util.Set;

import javax.enterprise.context.spi.Context;
import javax.enterprise.context.spi.Contextual;
import javax.enterprise.context.spi.CreationalContext;
import javax.enterprise.inject.spi.Bean;
import javax.enterprise.inject.spi.BeanManager;
import javax.inject.Inject;

import org.eclipse.jetty.cdi.core.AnyLiteral;
import org.eclipse.jetty.cdi.core.ScopedInstance;
import org.eclipse.jetty.cdi.core.SimpleBeanStore;
import org.eclipse.jetty.cdi.websocket.annotation.WebSocketScope;
import org.eclipse.jetty.util.log.Log;
import org.eclipse.jetty.util.log.Logger;
import org.eclipse.jetty.websocket.api.Session;

/**
 * WebSocket Scope Context.
 * <p>
 * A CDI Context definition for how CDI will use objects defined to belong to the WebSocketScope
 */
public class WebSocketScopeContext implements Context
{
    private static final Logger LOG = Log.getLogger(WebSocketScopeContext.class);

    private static ThreadLocal<WebSocketScopeContext> current = new ThreadLocal<>();

    public static WebSocketScopeContext current()
    {
        return current.get();
    }

    private SimpleBeanStore beanStore;

    @Inject
    private BeanManager beanManager;

    private ThreadLocal<org.eclipse.jetty.websocket.api.Session> session = new ThreadLocal<>();

    public void begin()
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} begin()",this);
        }
        current.set(this);
    }

    public void create()
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} create()",this);
        }
        current.set(this);
        beanStore = new SimpleBeanStore();
    }

    public void destroy()
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} destroy()",this);
        }

        beanStore.destroy();
    }

    public void end()
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} end()",this);
        }
        beanStore.clear();
    }

    @SuppressWarnings({ "unchecked" })
    @Override
    public <T> T get(Contextual<T> contextual)
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} get({})",this,contextual);
        }

        Bean<T> bean = (Bean<T>)contextual;

        if (bean.getBeanClass().isAssignableFrom(Session.class))
        {
            return (T)this.session;
        }
        
        if (beanStore == null)
        {
            return null;
        }

        List<ScopedInstance<?>> beans = beanStore.getBeans(contextual);

        if ((beans != null) && (!beans.isEmpty()))
        {
            return (T)beans.get(0).instance;
        }

        return null;
    }

    @SuppressWarnings("unchecked")
    @Override
    public <T> T get(Contextual<T> contextual, CreationalContext<T> creationalContext)
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} get({},{})",this,contextual,creationalContext);
        }

        Bean<T> bean = (Bean<T>)contextual;

        if (bean.getBeanClass().isAssignableFrom(Session.class))
        {
            return (T)this.session;
        }

        if (beanStore == null)
        {
            beanStore = new SimpleBeanStore();
        }

        List<ScopedInstance<?>> beans = beanStore.getBeans(contextual);

        if ((beans != null) && (!beans.isEmpty()))
        {
            for (ScopedInstance<?> instance : beans)
            {
                if (instance.bean.equals(bean))
                {
                    return (T)instance.instance;
                }
            }
        }

        // no bean found, create it
        T t = bean.create(creationalContext);
        ScopedInstance<T> customInstance = new ScopedInstance<>();
        customInstance.bean = bean;
        customInstance.creationalContext = creationalContext;
        customInstance.instance = t;
        beanStore.addBean(customInstance);
        return t;
    }

    @Override
    public Class<? extends Annotation> getScope()
    {
        return WebSocketScope.class;
    }

    @Override
    public boolean isActive()
    {
        return true;
    }

    @SuppressWarnings({ "unchecked", "rawtypes" })
    public <T> T newInstance(Class<T> clazz)
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("newInstance({})",clazz);
        }
        Set<Bean<?>> beans = beanManager.getBeans(clazz,AnyLiteral.INSTANCE);
        if (beans.isEmpty())
        {
            return null;
        }

        Bean bean = beans.iterator().next();
        CreationalContext cc = beanManager.createCreationalContext(bean);
        return (T)beanManager.getReference(bean,clazz,cc);
    }

    public void setSession(org.eclipse.jetty.websocket.api.Session sess)
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} setSession({})",this,sess);
        }
        current.set(this);
        this.session.set(sess);
    }

    public org.eclipse.jetty.websocket.api.Session getSession()
    {
        if (LOG.isDebugEnabled())
        {
            LOG.debug("{} getSession()",this);
        }
        return this.session.get();
    }

    @Override
    public String toString()
    {
        return String.format("%s@%X[%s]",this.getClass().getSimpleName(),hashCode(),beanStore);
    }
}