| /******************************************************************************* |
| * Copyright (c) 2010, 2013 Sonatype, Inc. |
| * All rights reserved. This program and the accompanying materials |
| * are made available under the terms of the Eclipse Public License v1.0 |
| * which accompanies this distribution, and is available at |
| * http://www.eclipse.org/legal/epl-v10.html |
| * |
| * Contributors: |
| * Stuart McCulloch (Sonatype, Inc.) - initial API and implementation |
| *******************************************************************************/ |
| package org.eclipse.sisu.inject; |
| |
| import java.util.ArrayList; |
| import java.util.Map; |
| import java.util.concurrent.ConcurrentMap; |
| |
| import javax.inject.Inject; |
| import javax.inject.Singleton; |
| |
| import org.eclipse.sisu.BeanEntry; |
| import org.eclipse.sisu.Mediator; |
| |
| import com.google.inject.Injector; |
| import com.google.inject.Key; |
| import com.google.inject.TypeLiteral; |
| |
| /** |
| * Default {@link MutableBeanLocator} that locates qualified beans across a dynamic group of {@link BindingPublisher}s. |
| */ |
| @Singleton |
| @SuppressWarnings( { "rawtypes", "unchecked" } ) |
| public final class DefaultBeanLocator |
| implements MutableBeanLocator |
| { |
| // ---------------------------------------------------------------------- |
| // Implementation fields |
| // ---------------------------------------------------------------------- |
| |
| private final RankedSequence<BindingPublisher> publishers = new RankedSequence<BindingPublisher>(); |
| |
| private final ConcurrentMap<TypeLiteral, RankedBindings> cachedBindings = Soft.concurrentValues( 256, 8 ); |
| |
| // reverse mapping; can't use watcher as key since it may not be unique |
| private final Map<WatchedBeans, Object> cachedWatchers = Weak.values(); |
| |
| private final ImplicitBindings implicitBindings = new ImplicitBindings( publishers ); |
| |
| // ---------------------------------------------------------------------- |
| // Public methods |
| // ---------------------------------------------------------------------- |
| |
| public Iterable<BeanEntry> locate( final Key key ) |
| { |
| final TypeLiteral type = key.getTypeLiteral(); |
| RankedBindings bindings = cachedBindings.get( type ); |
| if ( null == bindings ) |
| { |
| synchronized ( cachedBindings ) // perform new lookup |
| { |
| bindings = new RankedBindings( type, publishers ); |
| final RankedBindings oldBindings = cachedBindings.putIfAbsent( type, bindings ); |
| if ( null != oldBindings ) |
| { |
| bindings = oldBindings; |
| } |
| } |
| } |
| final boolean isImplicit = key.getAnnotationType() == null && TypeArguments.isImplicit( type ); |
| return new LocatedBeans( key, bindings, isImplicit ? implicitBindings : null ); |
| } |
| |
| public synchronized void watch( final Key key, final Mediator mediator, final Object watcher ) |
| { |
| final WatchedBeans beans = new WatchedBeans( key, mediator, watcher ); |
| for ( final BindingPublisher p : publishers() ) |
| { |
| p.subscribe( beans ); |
| } |
| cachedWatchers.put( beans, watcher ); |
| } |
| |
| public synchronized boolean add( final BindingPublisher publisher, final int rank ) |
| { |
| if ( publishers.contains( publisher ) ) |
| { |
| return false; |
| } |
| Logs.trace( "Add publisher: {}", publisher, null ); |
| synchronized ( cachedBindings ) // block new lookup while we update the cache |
| { |
| publishers.insert( publisher, rank ); |
| for ( final RankedBindings bindings : cachedBindings.values() ) |
| { |
| bindings.add( publisher, rank ); |
| } |
| } |
| // take defensive copy in case publisher.subscribe has side-effect that triggers 'watch' |
| for ( final WatchedBeans beans : new ArrayList<WatchedBeans>( cachedWatchers.keySet() ) ) |
| { |
| publisher.subscribe( beans ); |
| } |
| return true; |
| } |
| |
| public synchronized boolean remove( final BindingPublisher publisher ) |
| { |
| final BindingPublisher oldPublisher; |
| synchronized ( cachedBindings ) // block new lookup while we update the cache |
| { |
| oldPublisher = publishers.remove( publisher ); |
| if ( null == oldPublisher ) |
| { |
| return false; |
| } |
| Logs.trace( "Remove publisher: {}", oldPublisher, null ); |
| for ( final RankedBindings bindings : cachedBindings.values() ) |
| { |
| bindings.remove( oldPublisher ); |
| } |
| } |
| for ( final WatchedBeans beans : cachedWatchers.keySet() ) |
| { |
| oldPublisher.unsubscribe( beans ); |
| } |
| return true; |
| } |
| |
| public Iterable<BindingPublisher> publishers() |
| { |
| return publishers.snapshot(); |
| } |
| |
| public synchronized void clear() |
| { |
| for ( final BindingPublisher p : publishers() ) |
| { |
| remove( p ); |
| } |
| } |
| |
| public void add( final Injector injector, final int rank ) |
| { |
| add( new InjectorPublisher( injector, new DefaultRankingFunction( rank ) ), rank ); |
| } |
| |
| public void remove( final Injector injector ) |
| { |
| remove( new InjectorPublisher( injector, null /* unused */) ); |
| } |
| |
| // ---------------------------------------------------------------------- |
| // Implementation methods |
| // ---------------------------------------------------------------------- |
| |
| /** |
| * Automatically publishes any {@link Injector} that contains a binding to this {@link BeanLocator}. |
| * |
| * @param injector The injector |
| */ |
| @Inject |
| void autoPublish( final Injector injector ) |
| { |
| staticAutoPublish( this, injector ); |
| } |
| |
| @Inject |
| static void staticAutoPublish( final MutableBeanLocator locator, final Injector injector ) |
| { |
| final RankingFunction function = injector.getInstance( RankingFunction.class ); |
| locator.add( new InjectorPublisher( injector, function ), function.maxRank() ); |
| } |
| } |