| /* |
| * Copyright (c) 2008, Rickard Öberg. All Rights Reserved. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0 |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| * |
| */ |
| |
| package org.apache.zest.runtime.composite; |
| |
| import java.lang.reflect.Method; |
| import java.util.LinkedHashMap; |
| import org.apache.zest.api.composite.MissingMethodException; |
| import org.apache.zest.functional.HierarchicalVisitor; |
| import org.apache.zest.functional.Iterables; |
| import org.apache.zest.functional.VisitableHierarchy; |
| import org.apache.zest.runtime.injection.Dependencies; |
| import org.apache.zest.runtime.injection.DependencyModel; |
| import org.apache.zest.spi.module.ModuleSpi; |
| |
| import static org.apache.zest.functional.Iterables.map; |
| |
| /** |
| * Model for Composite methods. This includes both private and public methods. |
| */ |
| public final class CompositeMethodsModel |
| implements VisitableHierarchy<Object, Object> |
| { |
| private final LinkedHashMap<Method, CompositeMethodModel> methods; |
| private final MixinsModel mixinsModel; |
| |
| public CompositeMethodsModel( MixinsModel mixinsModel ) |
| { |
| methods = new LinkedHashMap<>(); |
| this.mixinsModel = mixinsModel; |
| } |
| |
| public Iterable<DependencyModel> dependencies() |
| { |
| return Iterables.flattenIterables( map( Dependencies.DEPENDENCIES_FUNCTION, methods.values() ) ); |
| } |
| |
| // Context |
| public Object invoke( MixinsInstance mixins, |
| Object proxy, |
| Method method, |
| Object[] args, |
| ModuleSpi moduleInstance |
| ) |
| throws Throwable |
| { |
| CompositeMethodModel compositeMethod = methods.get( method ); |
| |
| if( compositeMethod == null ) |
| { |
| if( method.getDeclaringClass().equals( Object.class ) ) |
| { |
| return mixins.invokeObject( proxy, args, method ); |
| } |
| |
| if( !method.getDeclaringClass().isInterface() ) |
| { |
| Iterable<Class<?>> types = mixinsModel.mixinTypes(); |
| for( Class<?> aClass : types ) |
| { |
| try |
| { |
| Method realMethod = aClass.getMethod( method.getName(), method.getParameterTypes() ); |
| compositeMethod = methods.get( realMethod ); |
| break; |
| } |
| catch( NoSuchMethodException e ) |
| { |
| } |
| catch( SecurityException e ) |
| { |
| } |
| } |
| } |
| // return mixins.invokeObject( proxy, args, method ); |
| throw new MissingMethodException( "Method '" + method + "' is not implemented" ); |
| } |
| else |
| { |
| return compositeMethod.invoke( proxy, args, mixins, moduleInstance ); |
| } |
| } |
| |
| public void addMethod( CompositeMethodModel methodModel ) |
| { |
| methods.put( methodModel.method(), methodModel ); |
| } |
| |
| public boolean isImplemented( Method method ) |
| { |
| return methods.containsKey( method ); |
| } |
| |
| public Iterable<Method> methods() |
| { |
| return methods.keySet(); |
| } |
| |
| @Override |
| public <ThrowableType extends Throwable> boolean accept( HierarchicalVisitor<? super Object, ? super Object, ThrowableType> modelVisitor ) |
| throws ThrowableType |
| { |
| if( modelVisitor.visitEnter( this ) ) |
| { |
| for( CompositeMethodModel compositeMethodModel : methods.values() ) |
| { |
| if( !compositeMethodModel.accept( modelVisitor ) ) |
| { |
| break; |
| } |
| } |
| } |
| return modelVisitor.visitLeave( this ); |
| } |
| |
| @Override |
| public String toString() |
| { |
| return methods().toString(); |
| } |
| } |