001    /*****************************************************************************
002     * Copyright (c) PicoContainer Organization. All rights reserved.            *
003     * ------------------------------------------------------------------------- *
004     * The software in this package is published under the terms of the BSD      *
005     * style license a copy of which has been included with this distribution in *
006     * the LICENSE.txt file.                                                     *
007     *                                                                           *
008     * Idea by Rachel Davies, Original code by Aslak Hellesoy and Paul Hammant   *
009     *****************************************************************************/
010    package org.picocontainer.injectors;
011    
012    import org.picocontainer.Characteristics;
013    import org.picocontainer.ComponentAdapter;
014    import org.picocontainer.ComponentMonitor;
015    import org.picocontainer.LifecycleStrategy;
016    import org.picocontainer.Parameter;
017    import org.picocontainer.PicoCompositionException;
018    import org.picocontainer.behaviors.AbstractBehaviorFactory;
019    
020    import java.lang.reflect.Method;
021    import java.util.HashSet;
022    import java.util.Properties;
023    import java.util.Set;
024    
025    /**
026     * A {@link org.picocontainer.InjectionFactory} for methods.
027     * The factory creates {@link MethodInjector}.
028     * 
029     *  @author Paul Hammant 
030     */
031    @SuppressWarnings("serial")
032    public class MethodInjection extends AbstractInjectionFactory {
033    
034        private final AbstractInjectionFactory delegate;
035    
036        public MethodInjection(String injectionMethodName) {
037            delegate = new MethodInjectionByName(injectionMethodName);
038        }
039    
040        public MethodInjection(String... injectionMethodNames) {
041            delegate = new MethodInjectionByName(injectionMethodNames);
042        }
043    
044        public MethodInjection() {
045            this("inject");
046        }
047    
048        public MethodInjection(Method injectionMethod) {
049            delegate = new MethodInjectionByReflectionMethod(injectionMethod);
050        }
051    
052        public <T> ComponentAdapter<T> createComponentAdapter(ComponentMonitor componentMonitor, LifecycleStrategy lifecycleStrategy, Properties componentProperties, Object componentKey,
053                                                       Class<T> componentImplementation, Parameter... parameters) throws PicoCompositionException {
054            return delegate.createComponentAdapter(componentMonitor, lifecycleStrategy, componentProperties, componentKey, componentImplementation, parameters);
055        }
056    
057        public class MethodInjectionByName extends AbstractInjectionFactory {
058            private final Set<String> injectionMethodNames = new HashSet<String>();
059    
060            public MethodInjectionByName(String... injectionMethodNames) {
061                for (String injectionMethodName : injectionMethodNames) {
062                    this.injectionMethodNames.add(injectionMethodName);
063                }
064            }
065    
066            public MethodInjectionByName(String injectionMethodName) {
067                this.injectionMethodNames.add(injectionMethodName);
068            }
069    
070            public <T> ComponentAdapter<T> createComponentAdapter(ComponentMonitor monitor, LifecycleStrategy lifecycleStrategy, Properties componentProperties, Object componentKey, Class<T> componentImplementation, Parameter... parameters) throws PicoCompositionException {
071                boolean useNames = AbstractBehaviorFactory.arePropertiesPresent(componentProperties, Characteristics.USE_NAMES, true);
072                return wrapLifeCycle(new MethodInjector.ByMethodName(componentKey, componentImplementation, parameters, monitor, injectionMethodNames, useNames), lifecycleStrategy);
073            }
074        }
075    
076        public class MethodInjectionByReflectionMethod extends AbstractInjectionFactory {
077            private final Method injectionMethod;
078    
079            public MethodInjectionByReflectionMethod(Method injectionMethod) {
080                this.injectionMethod = injectionMethod;
081            }
082    
083            public <T> ComponentAdapter<T> createComponentAdapter(ComponentMonitor monitor, LifecycleStrategy lifecycleStrategy, Properties componentProperties, Object componentKey, Class<T> componentImplementation, Parameter... parameters) throws PicoCompositionException {
084                boolean useNames = AbstractBehaviorFactory.arePropertiesPresent(componentProperties, Characteristics.USE_NAMES, true);
085                if (injectionMethod.getDeclaringClass().isAssignableFrom(componentImplementation)) {
086                    return wrapLifeCycle(monitor.newInjector(new MethodInjector.ByReflectionMethod(componentKey, componentImplementation, parameters, monitor, injectionMethod, useNames)), lifecycleStrategy);
087                } else {
088                    throw new PicoCompositionException("method [" + injectionMethod + "] not on impl " + componentImplementation.getName());
089                }
090            }
091        }
092    
093    }