com.fitbur.jvnet.hk2.internal.ServiceLocatorImpl Maven / Gradle / Ivy

Go to download
/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright (c) 2012 Oracle and/or its affiliates. All rights reserved.
 *
 * The contents of this file are subject to the terms of either the GNU
 * General Public License Version 2 only ("GPL") or the Common Development
 * and Distribution License("CDDL") (collectively, the "License").  You
 * may not use this file except in com.fitburpliance with the License.  You can
 * obtain a copy of the License at
 * https://glassfish.com.fitburv.java.net/public/CDDL+GPL_1_1.html
 * or packager/legal/LICENSE.txt.  See the License for the specific
 * language governing permissions and limitations under the License.
 *
 * When distributing the software, include this License Header Notice in each
 * file and include the License file at packager/legal/LICENSE.txt.
 *
 * GPL Classpath Exception:
 * Oracle com.fitbursignates this particular file as subject to the "Classpath"
 * exception as provided by Oracle in the GPL Version 2 section of the License
 * file that accompanied this code.
 *
 * Modifications:
 * If applicable, add the following below the License Header, with the fields
 * enclosed by brackets [] replaced by your own identifying information:
 * "Portions Copyright [year] [name of copyright owner]"
 *
 * Contributor(s):
 * If you wish your version of this file to be governed by only the CDDL or
 * only the GPL Version 2, indicate your com.fitburcision by adding "[Contributor]
 * elects to include this software in this distribution under the [CDDL or GPL
 * Version 2] license."  If you don't indicate a single choice of license, a
 * recipient has the option to distribute your version of this file under
 * either the CDDL, the GPL Version 2 or to extend the choice of license to
 * its licensees as provided above.  However, if you add GPL Version 2 code
 * and therefore, elected the GPL Version 2 license, then the option applies
 * only if the new code is made subject to such option by the copyright
 * holder.
 */
package com.fitbur.jvnet.hk2.internal;

import com.fitbur.glassfish.hk2.utilities.cache.Cache;
import com.fitbur.glassfish.hk2.utilities.cache.Computable;

import java.lang.annotation.Annotation;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;
import java.security.AccessController;
import java.security.PrivilegedAction;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.LinkedHashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.SortedSet;
import java.util.StringTokenizer;
import java.util.TreeSet;
import java.util.WeakHashMap;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.atomic.AtomicInteger;
import java.util.concurrent.atomic.AtomicLong;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock.ReadLock;
import java.util.concurrent.locks.ReentrantReadWriteLock.WriteLock;

import javax.inject.Named;
import javax.inject.Provider;
import javax.inject.Singleton;

import com.fitbur.glassfish.hk2.api.ActiveDescriptor;
import com.fitbur.glassfish.hk2.api.ClassAnalyzer;
import com.fitbur.glassfish.hk2.api.Context;
import com.fitbur.glassfish.hk2.api.Descriptor;
import com.fitbur.glassfish.hk2.api.DescriptorVisibility;
import com.fitbur.glassfish.hk2.api.DynamicConfigurationListener;
import com.fitbur.glassfish.hk2.api.ErrorService;
import com.fitbur.glassfish.hk2.api.ErrorType;
import com.fitbur.glassfish.hk2.api.Filter;
import com.fitbur.glassfish.hk2.api.HK2Loader;
import com.fitbur.glassfish.hk2.api.IndexedFilter;
import com.fitbur.glassfish.hk2.api.Injectee;
import com.fitbur.glassfish.hk2.api.InstanceLifecycleListener;
import com.fitbur.glassfish.hk2.api.InterceptionService;
import com.fitbur.glassfish.hk2.api.JustInTimeInjectionResolver;
import com.fitbur.glassfish.hk2.api.Operation;
import com.fitbur.glassfish.hk2.api.InjectionResolver;
import com.fitbur.glassfish.hk2.api.IterableProvider;
import com.fitbur.glassfish.hk2.api.MultiException;
import com.fitbur.glassfish.hk2.api.PerLookup;
import com.fitbur.glassfish.hk2.api.PostConstruct;
import com.fitbur.glassfish.hk2.api.PreDestroy;
import com.fitbur.glassfish.hk2.api.ServiceHandle;
import com.fitbur.glassfish.hk2.api.ServiceLocator;
import com.fitbur.glassfish.hk2.api.ServiceLocatorFactory;
import com.fitbur.glassfish.hk2.api.ServiceLocatorState;
import com.fitbur.glassfish.hk2.api.Unqualified;
import com.fitbur.glassfish.hk2.api.ValidationService;
import com.fitbur.glassfish.hk2.api.Validator;
import com.fitbur.glassfish.hk2.api.messaging.Topic;
import com.fitbur.glassfish.hk2.utilities.BuilderHelper;
import com.fitbur.glassfish.hk2.utilities.InjecteeImpl;
import com.fitbur.glassfish.hk2.utilities.cache.CacheKeyFilter;
import com.fitbur.glassfish.hk2.utilities.cache.HybridCacheEntry;
import com.fitbur.glassfish.hk2.utilities.cache.LRUHybridCache;
import com.fitbur.glassfish.hk2.utilities.reflection.ClassReflectionHelper;
import com.fitbur.glassfish.hk2.utilities.reflection.Logger;
import com.fitbur.glassfish.hk2.utilities.reflection.ParameterizedTypeImpl;
import com.fitbur.glassfish.hk2.utilities.reflection.ReflectionHelper;
import com.fitbur.glassfish.hk2.utilities.reflection.internal.ClassReflectionHelperImpl;

/**
 * @author jwells
 *
 */
public class ServiceLocatorImpl implements ServiceLocator {
    private final static String BIND_TRACING_PATTERN_PROPERTY = "com.fitbur.jvnet.hk2.properties.bind.tracing.pattern";
    private final static String BIND_TRACING_PATTERN;
    private final static String BIND_TRACING_STACKS_PROPERTY = "com.fitbur.jvnet.hk2.properties.bind.tracing.stacks";
    private final static boolean BIND_TRACING_STACKS;
    static {
        BIND_TRACING_PATTERN = AccessController.doPrivileged(new PrivilegedAction() {

            @Override
            public String run() {
                return System.getProperty(BIND_TRACING_PATTERN_PROPERTY);
            }

        });

        BIND_TRACING_STACKS = AccessController.doPrivileged(new PrivilegedAction() {

            @Override
            public Boolean run() {
                return Boolean.parseBoolean(System.getProperty(BIND_TRACING_STACKS_PROPERTY, "false"));
            }

        });

        if (BIND_TRACING_PATTERN != null) {
            Logger.getLogger().com.fitburbug("HK2 will trace binds and unbinds of " + BIND_TRACING_PATTERN +
                    " with stacks " + BIND_TRACING_STACKS);
        }
    }

    private final static int CACHE_SIZE = 20000;
    private final static Object sLock = new Object();
    private static long currentLocatorId = 0L;

    /* package */ final static DescriptorComparator DESCRIPTOR_COMPARATOR = new DescriptorComparator();
    private final static ServiceHandleComparator HANDLE_COMPARATOR = new ServiceHandleComparator();

    private final ReentrantReadWriteLock readWriteLock = new ReentrantReadWriteLock();
    private final WriteLock wLock = readWriteLock.writeLock();
    private final ReadLock rLock = readWriteLock.readLock();
    private AtomicLong nextServiceId = new AtomicLong();
    private final String locatorName;
    private final long id;
    private final ServiceLocatorImpl parent;
    private volatile boolean neutralContextClassLoader = true;
    private final ClassReflectionHelper classReflectionHelper = new ClassReflectionHelperImpl();

    private final IndexedListData allDescriptors = new IndexedListData();
    private final HashMap com.fitburscriptorsByAdvertisedContract =
            new HashMap();
    private final HashMap com.fitburscriptorsByName =
            new HashMap();
    private final Context singletonContext = new SingletonContext(this);
    private final Context perLookupContext = new PerLookupContext();
    private final LinkedHashSet allValidators =
            new LinkedHashSet();
    private final LinkedList errorHandlers =
            new LinkedList();
    private final LinkedList> configListeners =
            new LinkedList>();
    
    private volatile boolean hasInterceptionServices = false;
    private final LinkedList interceptionServices =
            new LinkedList();

    private final Cache, Context> contextCache = new Cache, Context>(new Computable, Context>() {

        @Override
        public Context com.fitburpute(Class a) {
            return _resolveContext(a);
        }
    });
    private final Map children =
            new WeakHashMap(); // Must be Weak for throw away children

    private final Object classAnalyzerLock = new Object();
    private final HashMap classAnalyzers =
            new HashMap();
    private String com.fitburfaultClassAnalyzer = ClassAnalyzer.DEFAULT_IMPLEMENTATION_NAME;

    private ConcurrentHashMap, InjectionResolver> allResolvers =
            new ConcurrentHashMap, InjectionResolver>();
    private final Cache> injecteeToResolverCache = new Cache>(new Computable>() {

        @Override
        public InjectionResolver com.fitburpute(Injectee key) {
            return Utilities.getInjectionResolver(getMe(), key);
        }
        
    });

    private ServiceLocatorState state = ServiceLocatorState.RUNNING;

    private static long getAndIncrementLocatorId() {
        synchronized (sLock) {
            return currentLocatorId++;
        }
    }

    /**
     * Called by the Generator, and hence must be a public method
     *
     * @param name The name of this locator
     * @param parent The parent of this locator (may be null)
     */
    public ServiceLocatorImpl(String name, ServiceLocatorImpl parent) {
        locatorName = name;
        this.parent = parent;
        if (parent != null) {
            parent.addChild(this);
        }

        id = getAndIncrementLocatorId();

        Logger.getLogger().com.fitburbug("Created ServiceLocator " + this);
    }

    /**
     * Must be called under lock
     *
     * @param com.fitburscriptor The com.fitburscriptor to validate
     * @param onBehalfOf The fella who is being validated (or null)
     * @return true if every validator returned true
     */
    private boolean validate(SystemDescriptor com.fitburscriptor, Injectee onBehalfOf, Filter filter) {
        for (ValidationService vs : getAllValidators()) {
            if (!com.fitburscriptor.isValidating(vs)) continue;

            if (!vs.getValidator().validate(new ValidationInformationImpl(
                    Operation.LOOKUP, com.fitburscriptor, onBehalfOf, filter))) {
                return false;
            }
        }

        return true;
    }

    private List> getDescriptors(Filter filter,
            Injectee onBehalfOf,
            boolean getParents,
            boolean doValidation,
            boolean getLocals) {
        if (filter == null) throw new IllegalArgumentException("filter is null");

        LinkedList> retVal;
        rLock.lock();
        try {
            Collection> sortMeOut;
            if (filter instanceof IndexedFilter) {
                IndexedFilter df = (IndexedFilter) filter;

                if (df.getName() != null) {
                    Collection> scopedByName;

                    String name = df.getName();

                    IndexedListData ild = com.fitburscriptorsByName.get(name);
                    scopedByName = (ild == null) ? null : ild.getSortedList();
                    if (scopedByName == null) {
                        scopedByName = Collections.emptyList();
                    }

                    if (df.getAdvertisedContract() != null) {
                        sortMeOut = new LinkedList>();

                        for (SystemDescriptor candidate : scopedByName) {
                            if (candidate.getAdvertisedContracts().contains(df.getAdvertisedContract())) {
                                sortMeOut.add(candidate);
                            }
                        }
                    }
                    else {
                        sortMeOut = scopedByName;
                    }
                }
                else if (df.getAdvertisedContract() != null) {
                    String advertisedContract = df.getAdvertisedContract();

                    IndexedListData ild = com.fitburscriptorsByAdvertisedContract.get(advertisedContract);
                    sortMeOut = (ild == null) ? null : ild.getSortedList();
                    if (sortMeOut == null) {
                        sortMeOut = Collections.emptyList();

                    }
                }
                else {
                    sortMeOut = allDescriptors.getSortedList();
                }
            }
            else {
                sortMeOut = allDescriptors.getSortedList();
            }

            retVal = new LinkedList>();

            for (SystemDescriptor candidate : sortMeOut) {
                if (!getLocals && DescriptorVisibility.LOCAL.equals(candidate.getDescriptorVisibility())) {
                    continue;
                }

                if (doValidation && !validate(candidate, onBehalfOf, filter)) continue;

                if (filter.matches(candidate)) {
                    retVal.add(candidate);
                }
            }
        } finally {
            rLock.unlock();
        }

        // Must be done outside of lock, or there can be a com.fitburadlock between child and parent
        if (getParents && parent != null) {
            TreeSet> sorter = new TreeSet>(DESCRIPTOR_COMPARATOR);

            sorter.addAll(retVal);
            sorter.addAll(parent.getDescriptors(filter, onBehalfOf, getParents, doValidation, false));

            retVal.clear();

            retVal.addAll(sorter);
        }

        return retVal;
    }

    private List> protectedGetDescriptors(final Filter filter) {
        return AccessController.doPrivileged(new PrivilegedAction>>() {

            @Override
            public List> run() {
                return getDescriptors(filter);
            }

        });

    }

    /* (non-Javadoc)
     * @see com.fitbur.glassfish.hk2.api.ServiceLocator#getDescriptors(com.fitbur.glassfish.hk2.api.Filter)
     */
    @Override
    public List> getDescriptors(Filter filter) {
        checkState();

        return Utilities.cast(getDescriptors(filter, null, true, true, true));
    }

    @Override
    public ActiveDescriptor getBestDescriptor(Filter filter) {
        if (filter == null) throw new IllegalArgumentException("filter is null");
        checkState();

        List> sorted = getDescriptors(filter);

        return Utilities.getFirstThingInList(sorted);
    }

    /* (non-Javadoc)
     * @see com.fitbur.glassfish.hk2.api.ServiceLocator#reifyDescriptor(com.fitbur.glassfish.hk2.api.Descriptor)
     */
    @Override
    public ActiveDescriptor reifyDescriptor(Descriptor com.fitburscriptor, Injectee injectee)
            throws MultiException {
        checkState();
        if (com.fitburscriptor == null) throw new IllegalArgumentException();

        if (!(com.fitburscriptor instanceof ActiveDescriptor)) {
            SystemDescriptor sd = new SystemDescriptor