org.glassfish.hk2.configuration.internal.ConfiguredByContext Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2014, 2024 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package org.glassfish.hk2.configuration.internal;

import java.lang.annotation.Annotation;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.locks.ReentrantLock;

import jakarta.inject.Singleton;

import org.glassfish.hk2.api.ActiveDescriptor;
import org.glassfish.hk2.api.Context;
import org.glassfish.hk2.api.DescriptorVisibility;
import org.glassfish.hk2.api.MultiException;
import org.glassfish.hk2.api.ServiceHandle;
import org.glassfish.hk2.api.Visibility;
import org.glassfish.hk2.configuration.api.ConfiguredBy;

/**
 * @author jwells
 *
 */
@Singleton
@Visibility(DescriptorVisibility.LOCAL)
public class ConfiguredByContext implements Context {

    private final static ThreadLocal> workingOn = ThreadLocal.withInitial(() -> null);

    private final ReentrantLock lock = new ReentrantLock();
    private final Map, Object> db = new HashMap<>();

    /* (non-Javadoc)
     * @see org.glassfish.hk2.api.Context#getScope()
     */
    @Override
    public Class getScope() {
        return ConfiguredBy.class;
    }

    /* (non-Javadoc)
     * @see org.glassfish.hk2.api.Context#findOrCreate(org.glassfish.hk2.api.ActiveDescriptor, org.glassfish.hk2.api.ServiceHandle)
     */
    @SuppressWarnings("unchecked")
    @Override
    public  U findOrCreate(ActiveDescriptor activeDescriptor,
            ServiceHandle root) {
        ActiveDescriptor previousValue = (ActiveDescriptor) workingOn.get();
        workingOn.set(activeDescriptor);
        try {
            return internalFindOrCreate(activeDescriptor, root);
        }
        finally {
            workingOn.set(previousValue);
        }
    }

    /* (non-Javadoc)
     * @see org.glassfish.hk2.api.Context#findOrCreate(org.glassfish.hk2.api.ActiveDescriptor, org.glassfish.hk2.api.ServiceHandle)
     */
    @SuppressWarnings("unchecked")
    private  U internalFindOrCreate(ActiveDescriptor activeDescriptor,
            ServiceHandle root) {
        lock.lock();
        try {
            U retVal = (U) db.get(activeDescriptor);
            if (retVal != null) return retVal;

            if (activeDescriptor.getName() == null) {
                throw new MultiException(new IllegalStateException("ConfiguredBy services without names are templates and cannot be created directly"));
            }

            retVal = activeDescriptor.create(root);
            db.put(activeDescriptor, retVal);

            return retVal;
        } finally {
            lock.unlock();
        }
    }

    /* (non-Javadoc)
     * @see org.glassfish.hk2.api.Context#containsKey(org.glassfish.hk2.api.ActiveDescriptor)
     */
    @Override
    public boolean containsKey(ActiveDescriptor descriptor) {
        lock.lock();
        try {
            return db.containsKey(descriptor);
        } finally {
            lock.unlock();
        }
    }

    /* (non-Javadoc)
     * @see org.glassfish.hk2.api.Context#destroyOne(org.glassfish.hk2.api.ActiveDescriptor)
     */
    @SuppressWarnings("unchecked")
    @Override
    public void destroyOne(ActiveDescriptor descriptor) {
        lock.lock();
        try {
            Object destroyMe = db.remove(descriptor);
            if (destroyMe == null) return;

            ((ActiveDescriptor