attila@90: /*
attila@90:  * Copyright (c) 2010, 2013, Oracle and/or its affiliates. All rights reserved.
attila@90:  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
attila@90:  *
attila@90:  * This code is free software; you can redistribute it and/or modify it
attila@90:  * under the terms of the GNU General Public License version 2 only, as
attila@90:  * published by the Free Software Foundation.  Oracle designates this
attila@90:  * particular file as subject to the "Classpath" exception as provided
attila@90:  * by Oracle in the LICENSE file that accompanied this code.
attila@90:  *
attila@90:  * This code is distributed in the hope that it will be useful, but WITHOUT
attila@90:  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
attila@90:  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
attila@90:  * version 2 for more details (a copy is included in the LICENSE file that
attila@90:  * accompanied this code).
attila@90:  *
attila@90:  * You should have received a copy of the GNU General Public License version
attila@90:  * 2 along with this work; if not, write to the Free Software Foundation,
attila@90:  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
attila@90:  *
attila@90:  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
attila@90:  * or visit www.oracle.com if you need additional information or have any
attila@90:  * questions.
attila@90:  */
attila@90: 
attila@90: /*
attila@90:  * This file is available under and governed by the GNU General Public
attila@90:  * License version 2 only, as published by the Free Software Foundation.
attila@90:  * However, the following notice accompanied the original version of this
attila@90:  * file, and Oracle licenses the original version of this file under the BSD
attila@90:  * license:
attila@90:  */
attila@90: /*
attila@90:    Copyright 2009-2013 Attila Szegedi
attila@90: 
attila@90:    Licensed under both the Apache License, Version 2.0 (the "Apache License")
attila@90:    and the BSD License (the "BSD License"), with licensee being free to
attila@90:    choose either of the two at their discretion.
attila@90: 
attila@90:    You may not use this file except in compliance with either the Apache
attila@90:    License or the BSD License.
attila@90: 
attila@90:    If you choose to use this file in compliance with the Apache License, the
attila@90:    following notice applies to you:
attila@90: 
attila@90:        You may obtain a copy of the Apache License at
attila@90: 
attila@90:            http://www.apache.org/licenses/LICENSE-2.0
attila@90: 
attila@90:        Unless required by applicable law or agreed to in writing, software
attila@90:        distributed under the License is distributed on an "AS IS" BASIS,
attila@90:        WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
attila@90:        implied. See the License for the specific language governing
attila@90:        permissions and limitations under the License.
attila@90: 
attila@90:    If you choose to use this file in compliance with the BSD License, the
attila@90:    following notice applies to you:
attila@90: 
attila@90:        Redistribution and use in source and binary forms, with or without
attila@90:        modification, are permitted provided that the following conditions are
attila@90:        met:
attila@90:        * Redistributions of source code must retain the above copyright
attila@90:          notice, this list of conditions and the following disclaimer.
attila@90:        * Redistributions in binary form must reproduce the above copyright
attila@90:          notice, this list of conditions and the following disclaimer in the
attila@90:          documentation and/or other materials provided with the distribution.
attila@90:        * Neither the name of the copyright holder nor the names of
attila@90:          contributors may be used to endorse or promote products derived from
attila@90:          this software without specific prior written permission.
attila@90: 
attila@90:        THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
attila@90:        IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
attila@90:        TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
attila@90:        PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL COPYRIGHT HOLDER
attila@90:        BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
attila@90:        CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
attila@90:        SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
attila@90:        BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
attila@90:        WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
attila@90:        OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
attila@90:        ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
attila@90: */
attila@90: 
attila@90: package jdk.internal.dynalink.beans;
attila@90: 
attila@90: import java.lang.invoke.MethodHandle;
attila@90: import java.lang.invoke.MethodHandles;
attila@90: import java.lang.invoke.MethodType;
attila@90: import java.util.ArrayList;
attila@90: import java.util.Iterator;
attila@90: import java.util.List;
attila@90: import java.util.Map;
attila@90: import java.util.concurrent.ConcurrentHashMap;
attila@90: import jdk.internal.dynalink.linker.LinkerServices;
attila@90: import jdk.internal.dynalink.support.Lookup;
attila@90: 
attila@90: /**
attila@90:  * Represents a subset of overloaded methods for a certain method name on a certain class. It can be either a fixarg or
attila@90:  * a vararg subset depending on the subclass. The method is for a fixed number of arguments though (as it is generated
attila@90:  * for a concrete call site). As such, all methods in the subset can be invoked with the specified number of arguments
attila@90:  * (exactly matching for fixargs, or having less than or equal fixed arguments, for varargs).
attila@90:  *
attila@90:  * @author Attila Szegedi
attila@90:  */
attila@90: class OverloadedMethod {
attila@90:     private final Map<ClassString, MethodHandle> argTypesToMethods = new ConcurrentHashMap<>();
attila@90:     private final OverloadedDynamicMethod parent;
attila@90:     private final MethodType callSiteType;
attila@90:     private final MethodHandle invoker;
attila@90:     private final LinkerServices linkerServices;
attila@90:     private final ArrayList<MethodHandle> fixArgMethods;
attila@90:     private final ArrayList<MethodHandle> varArgMethods;
attila@90: 
attila@90:     OverloadedMethod(List<MethodHandle> methodHandles, OverloadedDynamicMethod parent, MethodType callSiteType,
attila@90:             LinkerServices linkerServices) {
attila@90:         this.parent = parent;
attila@90:         this.callSiteType = callSiteType;
attila@90:         this.linkerServices = linkerServices;
attila@90: 
attila@90:         fixArgMethods = new ArrayList<>(methodHandles.size());
attila@90:         varArgMethods = new ArrayList<>(methodHandles.size());
attila@90:         final int argNum = callSiteType.parameterCount();
attila@90:         for(MethodHandle mh: methodHandles) {
attila@90:             if(mh.isVarargsCollector()) {
attila@90:                 final MethodHandle asFixed = mh.asFixedArity();
attila@90:                 if(argNum == asFixed.type().parameterCount()) {
attila@90:                     fixArgMethods.add(asFixed);
attila@90:                 }
attila@90:                 varArgMethods.add(mh);
attila@90:             } else {
attila@90:                 fixArgMethods.add(mh);
attila@90:             }
attila@90:         }
attila@90:         fixArgMethods.trimToSize();
attila@90:         varArgMethods.trimToSize();
attila@90: 
attila@90:         final MethodHandle bound = SELECT_METHOD.bindTo(this);
attila@90:         final MethodHandle collecting = SimpleDynamicMethod.collectArguments(bound, argNum).asType(
attila@90:                 callSiteType.changeReturnType(MethodHandle.class));
attila@90:         invoker = MethodHandles.foldArguments(MethodHandles.exactInvoker(callSiteType), collecting);
attila@90:     }
attila@90: 
attila@90:     MethodHandle getInvoker() {
attila@90:         return invoker;
attila@90:     }
attila@90: 
attila@90:     private static final MethodHandle SELECT_METHOD = Lookup.findOwnSpecial(MethodHandles.lookup(), "selectMethod",
attila@90:             MethodHandle.class, Object[].class);
attila@90: 
attila@90:     @SuppressWarnings("unused")
attila@90:     private MethodHandle selectMethod(Object[] args) throws NoSuchMethodException {
attila@90:         final Class<?>[] argTypes = new Class[args.length];
attila@90:         for(int i = 0; i < argTypes.length; ++i) {
attila@90:             final Object arg = args[i];
attila@166:             argTypes[i] = arg == null ? ClassString.NULL_CLASS : arg.getClass();
attila@90:         }
attila@90:         final ClassString classString = new ClassString(argTypes);
attila@90:         MethodHandle method = argTypesToMethods.get(classString);
attila@90:         if(method == null) {
attila@90:             List<MethodHandle> methods = classString.getMaximallySpecifics(fixArgMethods, linkerServices, false);
attila@90:             if(methods.isEmpty()) {
attila@90:                 methods = classString.getMaximallySpecifics(varArgMethods, linkerServices, true);
attila@90:             }
attila@90:             switch(methods.size()) {
attila@90:                 case 0: {
attila@90:                     method = getNoSuchMethodThrower(argTypes);
attila@90:                     break;
attila@90:                 }
attila@90:                 case 1: {
attila@90:                     method = new SimpleDynamicMethod(methods.get(0)).getInvocation(callSiteType, linkerServices);
attila@90:                     break;
attila@90:                 }
attila@90:                 default: {
attila@90:                     // This is unfortunate - invocation time ambiguity. We can still save the day if
attila@90:                     method = getAmbiguousMethodThrower(argTypes, methods);
attila@90:                     break;
attila@90:                 }
attila@90:             }
attila@90:             // Avoid keeping references to unrelated classes; this ruins the performance a bit, but avoids class loader
attila@90:             // memory leaks.
attila@90:             if(classString.isVisibleFrom(parent.getClassLoader())) {
attila@90:                 argTypesToMethods.put(classString, method);
attila@90:             }
attila@90:         }
attila@90:         return method;
attila@90:     }
attila@90: 
attila@90:     private MethodHandle getNoSuchMethodThrower(Class<?>[] argTypes) {
attila@90:         return adaptThrower(MethodHandles.insertArguments(THROW_NO_SUCH_METHOD, 0, this, argTypes));
attila@90:     }
attila@90: 
attila@90:     private static final MethodHandle THROW_NO_SUCH_METHOD = Lookup.findOwnSpecial(MethodHandles.lookup(),
attila@90:             "throwNoSuchMethod", void.class, Class[].class);
attila@90: 
attila@90:     @SuppressWarnings("unused")
attila@90:     private void throwNoSuchMethod(Class<?>[] argTypes) throws NoSuchMethodException {
attila@90:         if(varArgMethods.isEmpty()) {
attila@90:             throw new NoSuchMethodException("None of the fixed arity signatures " + getSignatureList(fixArgMethods) +
attila@90:                     " of method " + parent.getName() + " match the argument types " + argTypesString(argTypes));
attila@90:         }
attila@90:         throw new NoSuchMethodException("None of the fixed arity signatures " + getSignatureList(fixArgMethods) +
attila@90:                 " or the variable arity signatures " + getSignatureList(varArgMethods) + " of the method " +
attila@90:                 parent.getName() + " match the argument types " + argTypesString(argTypes));
attila@90:     }
attila@90: 
attila@90:     private MethodHandle getAmbiguousMethodThrower(Class<?>[] argTypes, List<MethodHandle> methods) {
attila@90:         return adaptThrower(MethodHandles.insertArguments(THROW_AMBIGUOUS_METHOD, 0, this, argTypes, methods));
attila@90:     }
attila@90: 
attila@90:     private MethodHandle adaptThrower(MethodHandle rawThrower) {
attila@90:         return MethodHandles.dropArguments(rawThrower, 0, callSiteType.parameterList()).asType(callSiteType);
attila@90:     }
attila@90: 
attila@90:     private static final MethodHandle THROW_AMBIGUOUS_METHOD = Lookup.findOwnSpecial(MethodHandles.lookup(),
attila@90:             "throwAmbiguousMethod", void.class, Class[].class, List.class);
attila@90: 
attila@90:     @SuppressWarnings("unused")
attila@90:     private void throwAmbiguousMethod(Class<?>[] argTypes, List<MethodHandle> methods) throws NoSuchMethodException {
attila@90:         final String arity = methods.get(0).isVarargsCollector() ? "variable" : "fixed";
attila@90:         throw new NoSuchMethodException("Can't unambiguously select between " + arity + " arity signatures " +
attila@90:                 getSignatureList(methods) + " of the method " + parent.getName() + " for argument types " +
attila@90:                 argTypesString(argTypes));
attila@90:     }
attila@90: 
attila@90:     private static String argTypesString(Class<?>[] classes) {
attila@90:         final StringBuilder b = new StringBuilder().append('[');
attila@90:         appendTypes(b, classes, false);
attila@90:         return b.append(']').toString();
attila@90:     }
attila@90: 
attila@90:     private static String getSignatureList(List<MethodHandle> methods) {
attila@90:         final StringBuilder b = new StringBuilder().append('[');
attila@90:         final Iterator<MethodHandle> it = methods.iterator();
attila@90:         if(it.hasNext()) {
attila@90:             appendSig(b, it.next());
attila@90:             while(it.hasNext()) {
attila@90:                 appendSig(b.append(", "), it.next());
attila@90:             }
attila@90:         }
attila@90:         return b.append(']').toString();
attila@90:     }
attila@90: 
attila@90:     private static void appendSig(StringBuilder b, MethodHandle m) {
attila@90:         b.append('(');
attila@90:         appendTypes(b, m.type().parameterArray(), m.isVarargsCollector());
attila@90:         b.append(')');
attila@90:     }
attila@90: 
attila@90:     private static void appendTypes(StringBuilder b, Class<?>[] classes, boolean varArg) {
attila@90:         final int l = classes.length;
attila@90:         if(!varArg) {
attila@90:             if(l > 1) {
attila@90:                 b.append(classes[1].getCanonicalName());
attila@90:                 for(int i = 2; i < l; ++i) {
attila@90:                     b.append(", ").append(classes[i].getCanonicalName());
attila@90:                 }
attila@90:             }
attila@90:         } else {
attila@90:             for(int i = 1; i < l - 1; ++i) {
attila@90:                 b.append(classes[i].getCanonicalName()).append(", ");
attila@90:             }
attila@90:             b.append(classes[l - 1].getComponentType().getCanonicalName()).append("...");
attila@90:         }
attila@90:     }
attila@101: }