| /* | |
| * Copyright (c) 2009, 2016, Oracle and/or its affiliates. All rights reserved. | |
| * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. | |
| * | |
| * This code is free software; you can redistribute it and/or modify it | |
| * under the terms of the GNU General Public License version 2 only, as | |
| * published by the Free Software Foundation. | |
| * | |
| * This code is distributed in the hope that it will be useful, but WITHOUT | |
| * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or | |
| * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License | |
| * version 2 for more details (a copy is included in the LICENSE file that | |
| * accompanied this code). | |
| * | |
| * You should have received a copy of the GNU General Public License version | |
| * 2 along with this work; if not, write to the Free Software Foundation, | |
| * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | |
| * | |
| * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA | |
| * or visit www.oracle.com if you need additional information or have any | |
| * questions. | |
| */ | |
| /* @test | |
| * @summary example code used in javadoc for java.lang.invoke API | |
| * @compile JavaDocExamplesTest.java | |
| * @run testng/othervm test.java.lang.invoke.JavaDocExamplesTest | |
| */ | |
| package test.java.lang.invoke; | |
| import java.io.StringWriter; | |
| import java.lang.invoke.*; | |
| import static java.lang.invoke.MethodHandles.*; | |
| import static java.lang.invoke.MethodType.*; | |
| import java.util.*; | |
| import org.testng.*; | |
| import org.testng.annotations.*; | |
| /** | |
| * @author jrose | |
| */ | |
| public class JavaDocExamplesTest { | |
| /** Wrapper for running the TestNG tests in this module. | |
| * Put TestNG on the classpath! | |
| */ | |
| public static void main(String... ignore) throws Throwable { | |
| new JavaDocExamplesTest().run(); | |
| } | |
| public void run() throws Throwable { | |
| testMisc(); | |
| testFindStatic(); | |
| testFindConstructor(); | |
| testFindVirtual(); | |
| testFindSpecial(); | |
| testPermuteArguments(); | |
| testZero(); | |
| testDropArguments(); | |
| testDropArgumentsToMatch(); | |
| testFilterArguments(); | |
| testFoldArguments(); | |
| testFoldArguments2(); | |
| testMethodHandlesSummary(); | |
| testAsSpreader(); | |
| testAsCollector(); | |
| testAsVarargsCollector(); | |
| testAsFixedArity(); | |
| testAsTypeCornerCases(); | |
| testMutableCallSite(); | |
| } | |
| // How much output? | |
| static final Class<?> THIS_CLASS = JavaDocExamplesTest.class; | |
| static int verbosity = Integer.getInteger(THIS_CLASS.getSimpleName()+".verbosity", 0); | |
| {} | |
| private static final Lookup LOOKUP = lookup(); | |
| // static final private MethodHandle CONCAT_1 = LOOKUP.findVirtual(String.class, | |
| // "concat", methodType(String.class, String.class)); | |
| // static final private MethodHandle HASHCODE_1 = LOOKUP.findVirtual(Object.class, | |
| // "hashCode", methodType(int.class)); | |
| // form required if ReflectiveOperationException is intercepted: | |
| private static final MethodHandle CONCAT_2, HASHCODE_2, ADD_2, SUB_2; | |
| static { | |
| try { | |
| Class<?> THIS_CLASS = LOOKUP.lookupClass(); | |
| CONCAT_2 = LOOKUP.findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| HASHCODE_2 = LOOKUP.findVirtual(Object.class, | |
| "hashCode", methodType(int.class)); | |
| ADD_2 = LOOKUP.findStatic(THIS_CLASS, "add", methodType(int.class, int.class, int.class)); | |
| SUB_2 = LOOKUP.findStatic(THIS_CLASS, "sub", methodType(int.class, int.class, int.class)); | |
| } catch (ReflectiveOperationException ex) { | |
| throw new RuntimeException(ex); | |
| } | |
| } | |
| static int add(int x, int y) { return x + y; } | |
| static int sub(int x, int y) { return x - y; } | |
| {} | |
| @Test public void testMisc() throws Throwable { | |
| // Extra tests, not from javadoc: | |
| {} | |
| MethodHandle CONCAT_3 = LOOKUP.findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| MethodHandle HASHCODE_3 = LOOKUP.findVirtual(Object.class, | |
| "hashCode", methodType(int.class)); | |
| //assertEquals("xy", (String) CONCAT_1.invokeExact("x", "y")); | |
| assertEquals("xy", (String) CONCAT_2.invokeExact("x", "y")); | |
| assertEquals("xy", (String) CONCAT_3.invokeExact("x", "y")); | |
| //assertEquals("xy".hashCode(), (int) HASHCODE_1.invokeExact((Object)"xy")); | |
| assertEquals("xy".hashCode(), (int) HASHCODE_2.invokeExact((Object)"xy")); | |
| assertEquals("xy".hashCode(), (int) HASHCODE_3.invokeExact((Object)"xy")); | |
| {} | |
| } | |
| @Test public void testFindStatic() throws Throwable { | |
| {} | |
| MethodHandle MH_asList = publicLookup().findStatic(Arrays.class, | |
| "asList", methodType(List.class, Object[].class)); | |
| assertEquals("[x, y]", MH_asList.invoke("x", "y").toString()); | |
| {} | |
| } | |
| @Test public void testFindVirtual() throws Throwable { | |
| {} | |
| MethodHandle MH_concat = publicLookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| MethodHandle MH_hashCode = publicLookup().findVirtual(Object.class, | |
| "hashCode", methodType(int.class)); | |
| MethodHandle MH_hashCode_String = publicLookup().findVirtual(String.class, | |
| "hashCode", methodType(int.class)); | |
| assertEquals("xy", (String) MH_concat.invokeExact("x", "y")); | |
| assertEquals("xy".hashCode(), (int) MH_hashCode.invokeExact((Object)"xy")); | |
| assertEquals("xy".hashCode(), (int) MH_hashCode_String.invokeExact("xy")); | |
| // interface method: | |
| MethodHandle MH_subSequence = publicLookup().findVirtual(CharSequence.class, | |
| "subSequence", methodType(CharSequence.class, int.class, int.class)); | |
| assertEquals("def", MH_subSequence.invoke("abcdefghi", 3, 6).toString()); | |
| // constructor "internal method" must be accessed differently: | |
| MethodType MT_newString = methodType(void.class); //()V for new String() | |
| try { assertEquals("impossible", lookup() | |
| .findVirtual(String.class, "<init>", MT_newString)); | |
| } catch (NoSuchMethodException ex) { } // OK | |
| MethodHandle MH_newString = publicLookup() | |
| .findConstructor(String.class, MT_newString); | |
| assertEquals("", (String) MH_newString.invokeExact()); | |
| {} | |
| } | |
| @Test public void testFindConstructor() throws Throwable { | |
| {} | |
| MethodHandle MH_newArrayList = publicLookup().findConstructor( | |
| ArrayList.class, methodType(void.class, Collection.class)); | |
| Collection orig = Arrays.asList("x", "y"); | |
| Collection copy = (ArrayList) MH_newArrayList.invokeExact(orig); | |
| assert(orig != copy); | |
| assertEquals(orig, copy); | |
| // a variable-arity constructor: | |
| MethodHandle MH_newProcessBuilder = publicLookup().findConstructor( | |
| ProcessBuilder.class, methodType(void.class, String[].class)); | |
| ProcessBuilder pb = (ProcessBuilder) | |
| MH_newProcessBuilder.invoke("x", "y", "z"); | |
| assertEquals("[x, y, z]", pb.command().toString()); | |
| {} | |
| } | |
| // for testFindSpecial | |
| {} | |
| static class Listie extends ArrayList { | |
| public String toString() { return "[wee Listie]"; } | |
| static Lookup lookup() { return MethodHandles.lookup(); } | |
| } | |
| {} | |
| @Test public void testFindSpecial() throws Throwable { | |
| {} | |
| // no access to constructor via invokeSpecial: | |
| MethodHandle MH_newListie = Listie.lookup() | |
| .findConstructor(Listie.class, methodType(void.class)); | |
| Listie l = (Listie) MH_newListie.invokeExact(); | |
| try { assertEquals("impossible", Listie.lookup().findSpecial( | |
| Listie.class, "<init>", methodType(void.class), Listie.class)); | |
| } catch (NoSuchMethodException ex) { } // OK | |
| // access to super and self methods via invokeSpecial: | |
| MethodHandle MH_super = Listie.lookup().findSpecial( | |
| ArrayList.class, "toString" , methodType(String.class), Listie.class); | |
| MethodHandle MH_this = Listie.lookup().findSpecial( | |
| Listie.class, "toString" , methodType(String.class), Listie.class); | |
| MethodHandle MH_duper = Listie.lookup().findSpecial( | |
| Object.class, "toString" , methodType(String.class), Listie.class); | |
| assertEquals("[]", (String) MH_super.invokeExact(l)); | |
| assertEquals(""+l, (String) MH_this.invokeExact(l)); | |
| assertEquals("[]", (String) MH_duper.invokeExact(l)); // ArrayList method | |
| try { assertEquals("inaccessible", Listie.lookup().findSpecial( | |
| String.class, "toString", methodType(String.class), Listie.class)); | |
| } catch (IllegalAccessException ex) { } // OK | |
| Listie subl = new Listie() { public String toString() { return "[subclass]"; } }; | |
| assertEquals(""+l, (String) MH_this.invokeExact(subl)); // Listie method | |
| {} | |
| } | |
| @Test public void testPermuteArguments() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodType intfn1 = methodType(int.class, int.class); | |
| MethodType intfn2 = methodType(int.class, int.class, int.class); | |
| MethodHandle sub = SUB_2;// ... {int x, int y => x-y} ...; | |
| assert(sub.type().equals(intfn2)); | |
| MethodHandle sub1 = permuteArguments(sub, intfn2, 0, 1); | |
| MethodHandle rsub = permuteArguments(sub, intfn2, 1, 0); | |
| assert((int)rsub.invokeExact(1, 100) == 99); | |
| MethodHandle add = ADD_2;// ... {int x, int y => x+y} ...; | |
| assert(add.type().equals(intfn2)); | |
| MethodHandle twice = permuteArguments(add, intfn1, 0, 0); | |
| assert(twice.type().equals(intfn1)); | |
| assert((int)twice.invokeExact(21) == 42); | |
| }} | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle cat = lookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| assertEquals("xy", (String) cat.invokeExact("x", "y")); | |
| MethodHandle d0 = dropArguments(cat, 0, String.class); | |
| assertEquals("yz", (String) d0.invokeExact("x", "y", "z")); | |
| MethodHandle d1 = dropArguments(cat, 1, String.class); | |
| assertEquals("xz", (String) d1.invokeExact("x", "y", "z")); | |
| MethodHandle d2 = dropArguments(cat, 2, String.class); | |
| assertEquals("xy", (String) d2.invokeExact("x", "y", "z")); | |
| MethodHandle d12 = dropArguments(cat, 1, int.class, boolean.class); | |
| assertEquals("xz", (String) d12.invokeExact("x", 12, true, "z")); | |
| }} | |
| } | |
| @Test public void testZero() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| Class<?> type = Double.class; | |
| MethodHandle mh1 = MethodHandles.explicitCastArguments(MethodHandles.constant(Object.class, null), methodType(type)); | |
| assertEquals("()Double", mh1.type().toString()); | |
| MethodHandle mh2 = MethodHandles.empty(methodType(type)); | |
| assertEquals("()Double", mh2.type().toString()); | |
| }} | |
| } | |
| @Test public void testDropArguments() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle cat = lookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| assertEquals("xy", (String) cat.invokeExact("x", "y")); | |
| MethodType bigType = cat.type().insertParameterTypes(0, int.class, String.class); | |
| MethodHandle d0 = dropArguments(cat, 0, bigType.parameterList().subList(0,2)); | |
| assertEquals(bigType, d0.type()); | |
| assertEquals("yz", (String) d0.invokeExact(123, "x", "y", "z")); | |
| }} | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle cat = lookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| assertEquals("xy", (String) cat.invokeExact("x", "y")); | |
| MethodHandle d0 = dropArguments(cat, 0, String.class); | |
| assertEquals("yz", (String) d0.invokeExact("x", "y", "z")); | |
| MethodHandle d1 = dropArguments(cat, 1, String.class); | |
| assertEquals("xz", (String) d1.invokeExact("x", "y", "z")); | |
| MethodHandle d2 = dropArguments(cat, 2, String.class); | |
| assertEquals("xy", (String) d2.invokeExact("x", "y", "z")); | |
| MethodHandle d12 = dropArguments(cat, 1, int.class, boolean.class); | |
| assertEquals("xz", (String) d12.invokeExact("x", 12, true, "z")); | |
| }} | |
| } | |
| @Test public void testDropArgumentsToMatch() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle h0= constant(boolean.class, true); | |
| MethodHandle h1 = lookup().findVirtual(String.class, "concat", methodType(String.class, String.class)); | |
| MethodType bigType = h1.type().insertParameterTypes(1, String.class, int.class); | |
| MethodHandle h2 = dropArguments(h1, 0, bigType.parameterList()); | |
| if (h1.type().parameterCount() < h2.type().parameterCount()) { | |
| h1 = dropArgumentsToMatch(h1, 0, h2.type().parameterList(), 0); // lengthen h1 | |
| } | |
| else { | |
| h2 = dropArgumentsToMatch(h2, 0, h1.type().parameterList(), 0); // lengthen h2 | |
| } | |
| MethodHandle h3 = guardWithTest(h0, h1, h2); | |
| assertEquals("xy", h3.invoke("x", "y", 1, "a", "b", "c")); | |
| }} | |
| } | |
| @Test public void testFilterArguments() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle cat = lookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| MethodHandle upcase = lookup().findVirtual(String.class, | |
| "toUpperCase", methodType(String.class)); | |
| assertEquals("xy", (String) cat.invokeExact("x", "y")); | |
| MethodHandle f0 = filterArguments(cat, 0, upcase); | |
| assertEquals("Xy", (String) f0.invokeExact("x", "y")); // Xy | |
| MethodHandle f1 = filterArguments(cat, 1, upcase); | |
| assertEquals("xY", (String) f1.invokeExact("x", "y")); // xY | |
| MethodHandle f2 = filterArguments(cat, 0, upcase, upcase); | |
| assertEquals("XY", (String) f2.invokeExact("x", "y")); // XY | |
| }} | |
| } | |
| @Test public void testCollectArguments() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle deepToString = publicLookup() | |
| .findStatic(Arrays.class, "deepToString", methodType(String.class, Object[].class)); | |
| MethodHandle ts1 = deepToString.asCollector(String[].class, 1); | |
| assertEquals("[strange]", (String) ts1.invokeExact("strange")); | |
| MethodHandle ts2 = deepToString.asCollector(String[].class, 2); | |
| assertEquals("[up, down]", (String) ts2.invokeExact("up", "down")); | |
| MethodHandle ts3 = deepToString.asCollector(String[].class, 3); | |
| MethodHandle ts3_ts2 = collectArguments(ts3, 1, ts2); | |
| assertEquals("[top, [up, down], strange]", | |
| (String) ts3_ts2.invokeExact("top", "up", "down", "strange")); | |
| MethodHandle ts3_ts2_ts1 = collectArguments(ts3_ts2, 3, ts1); | |
| assertEquals("[top, [up, down], [strange]]", | |
| (String) ts3_ts2_ts1.invokeExact("top", "up", "down", "strange")); | |
| MethodHandle ts3_ts2_ts3 = collectArguments(ts3_ts2, 1, ts3); | |
| assertEquals("[top, [[up, down, strange], charm], bottom]", | |
| (String) ts3_ts2_ts3.invokeExact("top", "up", "down", "strange", "charm", "bottom")); | |
| }} | |
| } | |
| @Test public void testFoldArguments() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle trace = publicLookup().findVirtual(java.io.PrintStream.class, | |
| "println", methodType(void.class, String.class)) | |
| .bindTo(System.out); | |
| MethodHandle cat = lookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| assertEquals("boojum", (String) cat.invokeExact("boo", "jum")); | |
| MethodHandle catTrace = foldArguments(cat, trace); | |
| // also prints "boo": | |
| assertEquals("boojum", (String) catTrace.invokeExact("boo", "jum")); | |
| }} | |
| } | |
| static void assertEquals(Object exp, Object act) { | |
| if (verbosity > 0) | |
| System.out.println("result: "+act); | |
| Assert.assertEquals(exp, act); | |
| } | |
| static void assertTrue(boolean b) { | |
| if (verbosity > 0) { | |
| System.out.println("result: " + b); | |
| } | |
| Assert.assertTrue(b); | |
| } | |
| @Test public void testMethodHandlesSummary() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| Object x, y; String s; int i; | |
| MethodType mt; MethodHandle mh; | |
| MethodHandles.Lookup lookup = MethodHandles.lookup(); | |
| // mt is (char,char)String | |
| mt = MethodType.methodType(String.class, char.class, char.class); | |
| mh = lookup.findVirtual(String.class, "replace", mt); | |
| s = (String) mh.invokeExact("daddy",'d','n'); | |
| // invokeExact(Ljava/lang/String;CC)Ljava/lang/String; | |
| assertEquals(s, "nanny"); | |
| // weakly typed invocation (using MHs.invoke) | |
| s = (String) mh.invokeWithArguments("sappy", 'p', 'v'); | |
| assertEquals(s, "savvy"); | |
| // mt is (Object[])List | |
| mt = MethodType.methodType(java.util.List.class, Object[].class); | |
| mh = lookup.findStatic(java.util.Arrays.class, "asList", mt); | |
| assert(mh.isVarargsCollector()); | |
| x = mh.invoke("one", "two"); | |
| // invoke(Ljava/lang/String;Ljava/lang/String;)Ljava/lang/Object; | |
| assertEquals(x, java.util.Arrays.asList("one","two")); | |
| // mt is (Object,Object,Object)Object | |
| mt = MethodType.genericMethodType(3); | |
| mh = mh.asType(mt); | |
| x = mh.invokeExact((Object)1, (Object)2, (Object)3); | |
| // invokeExact(Ljava/lang/Object;Ljava/lang/Object;Ljava/lang/Object;)Ljava/lang/Object; | |
| assertEquals(x, java.util.Arrays.asList(1,2,3)); | |
| // mt is ()int | |
| mt = MethodType.methodType(int.class); | |
| mh = lookup.findVirtual(java.util.List.class, "size", mt); | |
| i = (int) mh.invokeExact(java.util.Arrays.asList(1,2,3)); | |
| // invokeExact(Ljava/util/List;)I | |
| assert(i == 3); | |
| mt = MethodType.methodType(void.class, String.class); | |
| mh = lookup.findVirtual(java.io.PrintStream.class, "println", mt); | |
| mh.invokeExact(System.out, "Hello, world."); | |
| // invokeExact(Ljava/io/PrintStream;Ljava/lang/String;)V | |
| {} | |
| }} | |
| } | |
| @Test public void testAsSpreader() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle equals = publicLookup() | |
| .findVirtual(String.class, "equals", methodType(boolean.class, Object.class)); | |
| assert( (boolean) equals.invokeExact("me", (Object)"me")); | |
| assert(!(boolean) equals.invokeExact("me", (Object)"thee")); | |
| // spread both arguments from a 2-array: | |
| MethodHandle eq2 = equals.asSpreader(Object[].class, 2); | |
| assert( (boolean) eq2.invokeExact(new Object[]{ "me", "me" })); | |
| assert(!(boolean) eq2.invokeExact(new Object[]{ "me", "thee" })); | |
| // try to spread from anything but a 2-array: | |
| for (int n = 0; n <= 10; n++) { | |
| Object[] badArityArgs = (n == 2 ? new Object[0] : new Object[n]); | |
| try { assert((boolean) eq2.invokeExact(badArityArgs) && false); } | |
| catch (IllegalArgumentException ex) { } // OK | |
| } | |
| // spread both arguments from a String array: | |
| MethodHandle eq2s = equals.asSpreader(String[].class, 2); | |
| assert( (boolean) eq2s.invokeExact(new String[]{ "me", "me" })); | |
| assert(!(boolean) eq2s.invokeExact(new String[]{ "me", "thee" })); | |
| // spread second arguments from a 1-array: | |
| MethodHandle eq1 = equals.asSpreader(Object[].class, 1); | |
| assert( (boolean) eq1.invokeExact("me", new Object[]{ "me" })); | |
| assert(!(boolean) eq1.invokeExact("me", new Object[]{ "thee" })); | |
| // spread no arguments from a 0-array or null: | |
| MethodHandle eq0 = equals.asSpreader(Object[].class, 0); | |
| assert( (boolean) eq0.invokeExact("me", (Object)"me", new Object[0])); | |
| assert(!(boolean) eq0.invokeExact("me", (Object)"thee", (Object[])null)); | |
| // asSpreader and asCollector are approximate inverses: | |
| for (int n = 0; n <= 2; n++) { | |
| for (Class<?> a : new Class<?>[]{Object[].class, String[].class, CharSequence[].class}) { | |
| MethodHandle equals2 = equals.asSpreader(a, n).asCollector(a, n); | |
| assert( (boolean) equals2.invokeWithArguments("me", "me")); | |
| assert(!(boolean) equals2.invokeWithArguments("me", "thee")); | |
| } | |
| } | |
| MethodHandle caToString = publicLookup() | |
| .findStatic(Arrays.class, "toString", methodType(String.class, char[].class)); | |
| assertEquals("[A, B, C]", (String) caToString.invokeExact("ABC".toCharArray())); | |
| MethodHandle caString3 = caToString.asCollector(char[].class, 3); | |
| assertEquals("[A, B, C]", (String) caString3.invokeExact('A', 'B', 'C')); | |
| MethodHandle caToString2 = caString3.asSpreader(char[].class, 2); | |
| assertEquals("[A, B, C]", (String) caToString2.invokeExact('A', "BC".toCharArray())); | |
| }} | |
| } | |
| @Test public void testAsCollector() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle deepToString = publicLookup() | |
| .findStatic(Arrays.class, "deepToString", methodType(String.class, Object[].class)); | |
| assertEquals("[won]", (String) deepToString.invokeExact(new Object[]{"won"})); | |
| MethodHandle ts1 = deepToString.asCollector(Object[].class, 1); | |
| assertEquals(methodType(String.class, Object.class), ts1.type()); | |
| //assertEquals("[won]", (String) ts1.invokeExact( new Object[]{"won"})); //FAIL | |
| assertEquals("[[won]]", (String) ts1.invokeExact((Object) new Object[]{"won"})); | |
| // arrayType can be a subtype of Object[] | |
| MethodHandle ts2 = deepToString.asCollector(String[].class, 2); | |
| assertEquals(methodType(String.class, String.class, String.class), ts2.type()); | |
| assertEquals("[two, too]", (String) ts2.invokeExact("two", "too")); | |
| MethodHandle ts0 = deepToString.asCollector(Object[].class, 0); | |
| assertEquals("[]", (String) ts0.invokeExact()); | |
| // collectors can be nested, Lisp-style | |
| MethodHandle ts22 = deepToString.asCollector(Object[].class, 3).asCollector(String[].class, 2); | |
| assertEquals("[A, B, [C, D]]", ((String) ts22.invokeExact((Object)'A', (Object)"B", "C", "D"))); | |
| // arrayType can be any primitive array type | |
| MethodHandle bytesToString = publicLookup() | |
| .findStatic(Arrays.class, "toString", methodType(String.class, byte[].class)) | |
| .asCollector(byte[].class, 3); | |
| assertEquals("[1, 2, 3]", (String) bytesToString.invokeExact((byte)1, (byte)2, (byte)3)); | |
| MethodHandle longsToString = publicLookup() | |
| .findStatic(Arrays.class, "toString", methodType(String.class, long[].class)) | |
| .asCollector(long[].class, 1); | |
| assertEquals("[123]", (String) longsToString.invokeExact((long)123)); | |
| }} | |
| } | |
| @SuppressWarnings("rawtypes") | |
| @Test public void testAsVarargsCollector() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle deepToString = publicLookup() | |
| .findStatic(Arrays.class, "deepToString", methodType(String.class, Object[].class)); | |
| MethodHandle ts1 = deepToString.asVarargsCollector(Object[].class); | |
| assertEquals("[won]", (String) ts1.invokeExact( new Object[]{"won"})); | |
| assertEquals("[won]", (String) ts1.invoke( new Object[]{"won"})); | |
| assertEquals("[won]", (String) ts1.invoke( "won" )); | |
| assertEquals("[[won]]", (String) ts1.invoke((Object) new Object[]{"won"})); | |
| // findStatic of Arrays.asList(...) produces a variable arity method handle: | |
| MethodHandle asList = publicLookup() | |
| .findStatic(Arrays.class, "asList", methodType(List.class, Object[].class)); | |
| assertEquals(methodType(List.class, Object[].class), asList.type()); | |
| assert(asList.isVarargsCollector()); | |
| assertEquals("[]", asList.invoke().toString()); | |
| assertEquals("[1]", asList.invoke(1).toString()); | |
| assertEquals("[two, too]", asList.invoke("two", "too").toString()); | |
| String[] argv = { "three", "thee", "tee" }; | |
| assertEquals("[three, thee, tee]", asList.invoke(argv).toString()); | |
| assertEquals("[three, thee, tee]", asList.invoke((Object[])argv).toString()); | |
| List ls = (List) asList.invoke((Object)argv); | |
| assertEquals(1, ls.size()); | |
| assertEquals("[three, thee, tee]", Arrays.toString((Object[])ls.get(0))); | |
| }} | |
| } | |
| @Test public void testAsFixedArity() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle asListVar = publicLookup() | |
| .findStatic(Arrays.class, "asList", methodType(List.class, Object[].class)) | |
| .asVarargsCollector(Object[].class); | |
| MethodHandle asListFix = asListVar.asFixedArity(); | |
| assertEquals("[1]", asListVar.invoke(1).toString()); | |
| Exception caught = null; | |
| try { asListFix.invoke((Object)1); } | |
| catch (Exception ex) { caught = ex; } | |
| assert(caught instanceof ClassCastException); | |
| assertEquals("[two, too]", asListVar.invoke("two", "too").toString()); | |
| try { asListFix.invoke("two", "too"); } | |
| catch (Exception ex) { caught = ex; } | |
| assert(caught instanceof WrongMethodTypeException); | |
| Object[] argv = { "three", "thee", "tee" }; | |
| assertEquals("[three, thee, tee]", asListVar.invoke(argv).toString()); | |
| assertEquals("[three, thee, tee]", asListFix.invoke(argv).toString()); | |
| assertEquals(1, ((List) asListVar.invoke((Object)argv)).size()); | |
| assertEquals("[three, thee, tee]", asListFix.invoke((Object)argv).toString()); | |
| }} | |
| } | |
| @Test public void testAsTypeCornerCases() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle i2s = publicLookup() | |
| .findVirtual(Integer.class, "toString", methodType(String.class)); | |
| i2s = i2s.asType(i2s.type().unwrap()); | |
| MethodHandle l2s = publicLookup() | |
| .findVirtual(Long.class, "toString", methodType(String.class)); | |
| l2s = l2s.asType(l2s.type().unwrap()); | |
| Exception caught = null; | |
| try { i2s.asType(methodType(String.class, String.class)); } | |
| catch (Exception ex) { caught = ex; } | |
| assert(caught instanceof WrongMethodTypeException); | |
| i2s.asType(methodType(String.class, byte.class)); | |
| i2s.asType(methodType(String.class, Byte.class)); | |
| i2s.asType(methodType(String.class, Character.class)); | |
| i2s.asType(methodType(String.class, Integer.class)); | |
| l2s.asType(methodType(String.class, byte.class)); | |
| l2s.asType(methodType(String.class, Byte.class)); | |
| l2s.asType(methodType(String.class, Character.class)); | |
| l2s.asType(methodType(String.class, Integer.class)); | |
| l2s.asType(methodType(String.class, Long.class)); | |
| caught = null; | |
| try { i2s.asType(methodType(String.class, Long.class)); } | |
| catch (Exception ex) { caught = ex; } | |
| assert(caught instanceof WrongMethodTypeException); | |
| MethodHandle i2sGen = i2s.asType(methodType(String.class, Object.class)); | |
| MethodHandle l2sGen = l2s.asType(methodType(String.class, Object.class)); | |
| i2sGen.invoke(42); // int -> Integer -> Object -> Integer -> int | |
| i2sGen.invoke((byte)4); // byte -> Byte -> Object -> Byte -> byte -> int | |
| l2sGen.invoke(42); // int -> Integer -> Object -> Integer -> int | |
| l2sGen.invoke((byte)4); // byte -> Byte -> Object -> Byte -> byte -> int | |
| l2sGen.invoke(0x420000000L); | |
| caught = null; | |
| try { i2sGen.invoke(0x420000000L); } // long -> Long -> Object -> Integer CCE | |
| catch (Exception ex) { caught = ex; } | |
| assert(caught instanceof ClassCastException); | |
| caught = null; | |
| try { i2sGen.invoke("asdf"); } // String -> Object -> Integer CCE | |
| catch (Exception ex) { caught = ex; } | |
| assert(caught instanceof ClassCastException); | |
| {} | |
| }} | |
| } | |
| @Test public void testMutableCallSite() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MutableCallSite name = new MutableCallSite(MethodType.methodType(String.class)); | |
| MethodHandle MH_name = name.dynamicInvoker(); | |
| MethodType MT_str1 = MethodType.methodType(String.class); | |
| MethodHandle MH_upcase = MethodHandles.lookup() | |
| .findVirtual(String.class, "toUpperCase", MT_str1); | |
| MethodHandle worker1 = MethodHandles.filterReturnValue(MH_name, MH_upcase); | |
| name.setTarget(MethodHandles.constant(String.class, "Rocky")); | |
| assertEquals("ROCKY", (String) worker1.invokeExact()); | |
| name.setTarget(MethodHandles.constant(String.class, "Fred")); | |
| assertEquals("FRED", (String) worker1.invokeExact()); | |
| // (mutation can be continued indefinitely) | |
| /* | |
| * </pre></blockquote> | |
| * <p> | |
| * The same call site may be used in several places at once. | |
| * <blockquote><pre> | |
| */ | |
| MethodType MT_str2 = MethodType.methodType(String.class, String.class); | |
| MethodHandle MH_cat = lookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| MethodHandle MH_dear = MethodHandles.insertArguments(MH_cat, 1, ", dear?"); | |
| MethodHandle worker2 = MethodHandles.filterReturnValue(MH_name, MH_dear); | |
| assertEquals("Fred, dear?", (String) worker2.invokeExact()); | |
| name.setTarget(MethodHandles.constant(String.class, "Wilma")); | |
| assertEquals("WILMA", (String) worker1.invokeExact()); | |
| assertEquals("Wilma, dear?", (String) worker2.invokeExact()); | |
| {} | |
| }} | |
| } | |
| @Test public void testSwitchPoint() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle MH_strcat = MethodHandles.lookup() | |
| .findVirtual(String.class, "concat", MethodType.methodType(String.class, String.class)); | |
| SwitchPoint spt = new SwitchPoint(); | |
| assert(!spt.hasBeenInvalidated()); | |
| // the following steps may be repeated to re-use the same switch point: | |
| MethodHandle worker1 = MH_strcat; | |
| MethodHandle worker2 = MethodHandles.permuteArguments(MH_strcat, MH_strcat.type(), 1, 0); | |
| MethodHandle worker = spt.guardWithTest(worker1, worker2); | |
| assertEquals("method", (String) worker.invokeExact("met", "hod")); | |
| SwitchPoint.invalidateAll(new SwitchPoint[]{ spt }); | |
| assert(spt.hasBeenInvalidated()); | |
| assertEquals("hodmet", (String) worker.invokeExact("met", "hod")); | |
| {} | |
| }} | |
| } | |
| @Test public void testFoldArguments2() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| // argument-based dispatch for methods of the form boolean x.___(y: String) | |
| Lookup lookup = publicLookup(); | |
| // first, a tracing hack: | |
| MethodHandle println = lookup.findVirtual(java.io.PrintStream.class, "println", methodType(void.class, String.class)); | |
| MethodHandle arrayToString = lookup.findStatic(Arrays.class, "toString", methodType(String.class, Object[].class)); | |
| MethodHandle concat = lookup.findVirtual(String.class, "concat", methodType(String.class, String.class)); | |
| MethodHandle arrayToString_DIS = filterReturnValue(arrayToString, concat.bindTo("DIS:")); | |
| MethodHandle arrayToString_INV = filterReturnValue(arrayToString, concat.bindTo("INV:")); | |
| MethodHandle printArgs_DIS = filterReturnValue(arrayToString_DIS, println.bindTo(System.out)).asVarargsCollector(Object[].class); | |
| MethodHandle printArgs_INV = filterReturnValue(arrayToString_INV, println.bindTo(System.out)).asVarargsCollector(Object[].class); | |
| // metaobject protocol: | |
| MethodType mtype = methodType(boolean.class, String.class); | |
| MethodHandle findVirtual = lookup.findVirtual(Lookup.class, | |
| "findVirtual", methodType(MethodHandle.class, Class.class, String.class, MethodType.class)); | |
| MethodHandle getClass = lookup.findVirtual(Object.class, | |
| "getClass", methodType(Class.class)); | |
| MethodHandle dispatch = findVirtual; | |
| dispatch = filterArguments(dispatch, 1, getClass); | |
| dispatch = insertArguments(dispatch, 3, mtype); | |
| dispatch = dispatch.bindTo(lookup); | |
| assertEquals(methodType(MethodHandle.class, Object.class, String.class), dispatch.type()); | |
| MethodHandle invoker = invoker(mtype.insertParameterTypes(0, Object.class)); | |
| // wrap tracing around the dispatch and invoke steps: | |
| dispatch = foldArguments(dispatch, printArgs_DIS.asType(dispatch.type().changeReturnType(void.class))); | |
| invoker = foldArguments(invoker, printArgs_INV.asType(invoker.type().changeReturnType(void.class))); | |
| invoker = dropArguments(invoker, 2, String.class); // ignore selector | |
| // compose the dispatcher and the invoker: | |
| MethodHandle invokeDispatched = foldArguments(invoker, dispatch); | |
| Object x = "football", y = new java.util.Scanner("bar"); | |
| assert( (boolean) invokeDispatched.invokeExact(x, "startsWith", "foo")); | |
| assert(!(boolean) invokeDispatched.invokeExact(x, "startsWith", "#")); | |
| assert( (boolean) invokeDispatched.invokeExact(x, "endsWith", "all")); | |
| assert(!(boolean) invokeDispatched.invokeExact(x, "endsWith", "foo")); | |
| assert( (boolean) invokeDispatched.invokeExact(y, "hasNext", "[abc]+[rst]")); | |
| assert(!(boolean) invokeDispatched.invokeExact(y, "hasNext", "[123]+[789]")); | |
| }} | |
| } | |
| static int one(int k) { return 1; } | |
| static int inc(int i, int acc, int k) { return i + 1; } | |
| static int mult(int i, int acc, int k) { return i * acc; } | |
| static boolean pred(int i, int acc, int k) { return i < k; } | |
| static int fin(int i, int acc, int k) { return acc; } | |
| @Test public void testLoop() throws Throwable { | |
| MethodHandle MH_inc, MH_one, MH_mult, MH_pred, MH_fin; | |
| Class<?> I = int.class; | |
| MH_inc = LOOKUP.findStatic(THIS_CLASS, "inc", methodType(I, I, I, I)); | |
| MH_one = LOOKUP.findStatic(THIS_CLASS, "one", methodType(I, I)); | |
| MH_mult = LOOKUP.findStatic(THIS_CLASS, "mult", methodType(I, I, I, I)); | |
| MH_pred = LOOKUP.findStatic(THIS_CLASS, "pred", methodType(boolean.class, I, I, I)); | |
| MH_fin = LOOKUP.findStatic(THIS_CLASS, "fin", methodType(I, I, I, I)); | |
| {{ | |
| {} /// JAVADOC | |
| // iterative implementation of the factorial function as a loop handle | |
| // null initializer for counter, should initialize to 0 | |
| MethodHandle[] counterClause = new MethodHandle[]{null, MH_inc}; | |
| MethodHandle[] accumulatorClause = new MethodHandle[]{MH_one, MH_mult, MH_pred, MH_fin}; | |
| MethodHandle loop = MethodHandles.loop(counterClause, accumulatorClause); | |
| assertEquals(120, loop.invoke(5)); | |
| {} | |
| }} | |
| } | |
| static int inc(int i) { return i + 1; } // drop acc, k | |
| static int mult(int i, int acc) { return i * acc; } //drop k | |
| static boolean cmp(int i, int k) { return i < k; } | |
| @Test public void testSimplerLoop() throws Throwable { | |
| MethodHandle MH_inc, MH_mult, MH_cmp; | |
| Class<?> I = int.class; | |
| MH_inc = LOOKUP.findStatic(THIS_CLASS, "inc", methodType(I, I)); | |
| MH_mult = LOOKUP.findStatic(THIS_CLASS, "mult", methodType(I, I, I)); | |
| MH_cmp = LOOKUP.findStatic(THIS_CLASS, "cmp", methodType(boolean.class, I, I)); | |
| {{ | |
| {} /// JAVADOC | |
| // simplified implementation of the factorial function as a loop handle | |
| // null initializer for counter, should initialize to 0 | |
| MethodHandle MH_one = MethodHandles.constant(int.class, 1); | |
| MethodHandle MH_pred = MethodHandles.dropArguments(MH_cmp, 1, int.class); // drop acc | |
| MethodHandle MH_fin = MethodHandles.dropArguments(MethodHandles.identity(int.class), 0, int.class); // drop i | |
| MethodHandle[] counterClause = new MethodHandle[]{null, MH_inc}; | |
| MethodHandle[] accumulatorClause = new MethodHandle[]{MH_one, MH_mult, MH_pred, MH_fin}; | |
| MethodHandle loop = MethodHandles.loop(counterClause, accumulatorClause); | |
| assertEquals(720, loop.invoke(6)); | |
| {} | |
| }} | |
| } | |
| // for testFacLoop | |
| {} | |
| static class FacLoop { | |
| final int k; | |
| FacLoop(int k) { this.k = k; } | |
| int inc(int i) { return i + 1; } | |
| int mult(int i, int acc) { return i * acc; } | |
| boolean pred(int i) { return i < k; } | |
| int fin(int i, int acc) { return acc; } | |
| } | |
| {} | |
| // assume MH_inc, MH_mult, and MH_pred are handles to the above methods | |
| @Test public void testFacLoop() throws Throwable { | |
| MethodHandle MH_FacLoop, MH_inc, MH_mult, MH_pred, MH_fin; | |
| Class<?> I = int.class; | |
| MH_FacLoop = LOOKUP.findConstructor(FacLoop.class, methodType(void.class, I)); | |
| MH_inc = LOOKUP.findVirtual(FacLoop.class, "inc", methodType(I, I)); | |
| MH_mult = LOOKUP.findVirtual(FacLoop.class, "mult", methodType(I, I, I)); | |
| MH_pred = LOOKUP.findVirtual(FacLoop.class, "pred", methodType(boolean.class, I)); | |
| MH_fin = LOOKUP.findVirtual(FacLoop.class, "fin", methodType(I, I, I)); | |
| {{ | |
| {} /// JAVADOC | |
| // instance-based implementation of the factorial function as a loop handle | |
| // null initializer for counter, should initialize to 0 | |
| MethodHandle MH_one = MethodHandles.constant(int.class, 1); | |
| MethodHandle[] instanceClause = new MethodHandle[]{MH_FacLoop}; | |
| MethodHandle[] counterClause = new MethodHandle[]{null, MH_inc}; | |
| MethodHandle[] accumulatorClause = new MethodHandle[]{MH_one, MH_mult, MH_pred, MH_fin}; | |
| MethodHandle loop = MethodHandles.loop(instanceClause, counterClause, accumulatorClause); | |
| assertEquals(5040, loop.invoke(7)); | |
| {} | |
| }} | |
| } | |
| static List<String> initZip(Iterator<String> a, Iterator<String> b) { return new ArrayList<>(); } | |
| static boolean zipPred(List<String> zip, Iterator<String> a, Iterator<String> b) { return a.hasNext() && b.hasNext(); } | |
| static List<String> zipStep(List<String> zip, Iterator<String> a, Iterator<String> b) { | |
| zip.add(a.next()); | |
| zip.add(b.next()); | |
| return zip; | |
| } | |
| @Test public void testWhileLoop() throws Throwable { | |
| MethodHandle MH_initZip, MH_zipPred, MH_zipStep; | |
| Class<?> IT = Iterator.class; | |
| Class<?> L = List.class; | |
| MH_initZip = LOOKUP.findStatic(THIS_CLASS, "initZip", methodType(L, IT, IT)); | |
| MH_zipPred = LOOKUP.findStatic(THIS_CLASS, "zipPred", methodType(boolean.class, L, IT, IT)); | |
| MH_zipStep = LOOKUP.findStatic(THIS_CLASS, "zipStep", methodType(L, L, IT, IT)); | |
| {{ | |
| {} /// JAVADOC | |
| // implement the zip function for lists as a loop handle | |
| MethodHandle loop = MethodHandles.whileLoop(MH_initZip, MH_zipPred, MH_zipStep); | |
| List<String> a = Arrays.asList("a", "b", "c", "d"); | |
| List<String> b = Arrays.asList("e", "f", "g", "h"); | |
| List<String> zipped = Arrays.asList("a", "e", "b", "f", "c", "g", "d", "h"); | |
| assertEquals(zipped, (List<String>) loop.invoke(a.iterator(), b.iterator())); | |
| {} | |
| }} | |
| } | |
| static int zero(int limit) { return 0; } | |
| static int step(int i, int limit) { return i + 1; } | |
| static boolean pred(int i, int limit) { return i < limit; } | |
| @Test public void testDoWhileLoop() throws Throwable { | |
| MethodHandle MH_zero, MH_step, MH_pred; | |
| Class<?> I = int.class; | |
| MH_zero = LOOKUP.findStatic(THIS_CLASS, "zero", methodType(I, I)); | |
| MH_step = LOOKUP.findStatic(THIS_CLASS, "step", methodType(I, I, I)); | |
| MH_pred = LOOKUP.findStatic(THIS_CLASS, "pred", methodType(boolean.class, I, I)); | |
| {{ | |
| {} /// JAVADOC | |
| // int i = 0; while (i < limit) { ++i; } return i; => limit | |
| MethodHandle loop = MethodHandles.doWhileLoop(MH_zero, MH_step, MH_pred); | |
| assertEquals(23, loop.invoke(23)); | |
| {} | |
| }} | |
| } | |
| static String step(String v, int counter, String start_) { return "na " + v; } //#0 | |
| static String step(String v, int counter ) { return "na " + v; } //#1 | |
| static String step(String v, int counter, int iterations_, String pre, String start_) { return pre + " " + v; } //#2 | |
| static String step3(String v, int counter, String pre) { return pre + " " + v; } //#3 | |
| @Test public void testCountedLoop() throws Throwable { | |
| MethodHandle MH_step; | |
| Class<?> S = String.class, I = int.class; | |
| // Theme: | |
| MH_step = LOOKUP.findStatic(THIS_CLASS, "step", methodType(S, S, I, S)); | |
| {{ | |
| {} /// JAVADOC | |
| // String s = "Lambdaman!"; for (int i = 0; i < 13; ++i) { s = "na " + s; } return s; | |
| // => a variation on a well known theme | |
| MethodHandle fit13 = MethodHandles.constant(int.class, 13); | |
| MethodHandle start = MethodHandles.identity(String.class); | |
| MethodHandle loop = MethodHandles.countedLoop(fit13, start, MH_step); // (v, i, _) -> "na " + v | |
| assertEquals("na na na na na na na na na na na na na Lambdaman!", loop.invoke("Lambdaman!")); | |
| {} | |
| }} | |
| // Variation #1: | |
| MH_step = LOOKUP.findStatic(THIS_CLASS, "step", methodType(S, S, I)); | |
| {{ | |
| {} /// JAVADOC | |
| // String s = "Lambdaman!"; for (int i = 0; i < 13; ++i) { s = "na " + s; } return s; | |
| // => a variation on a well known theme | |
| MethodHandle count = MethodHandles.dropArguments(MethodHandles.identity(int.class), 1, String.class); | |
| MethodHandle start = MethodHandles.dropArguments(MethodHandles.identity(String.class), 0, int.class); | |
| MethodHandle loop = MethodHandles.countedLoop(count, start, MH_step); // (v, i) -> "na " + v | |
| assertEquals("na na na na na na na na na na na na na Lambdaman!", loop.invoke(13, "Lambdaman!")); | |
| {} | |
| assertEquals("na na Lambdaman!", loop.invoke(2, "Lambdaman!")); | |
| assertEquals("Lambdaman!", loop.invoke(0, "Lambdaman!")); | |
| assertEquals("Lambdaman!", loop.invoke(-1, "Lambdaman!")); | |
| assertEquals("Lambdaman!", loop.invoke(Integer.MIN_VALUE, "Lambdaman!")); | |
| }} | |
| // Variation #2: | |
| MH_step = LOOKUP.findStatic(THIS_CLASS, "step", methodType(S, S, I, I, S, S)); | |
| {{ | |
| {} /// JAVADOC | |
| // String s = "Lambdaman!", t = "na"; for (int i = 0; i < 13; ++i) { s = t + " " + s; } return s; | |
| // => a variation on a well known theme | |
| MethodHandle count = MethodHandles.identity(int.class); | |
| MethodHandle start = MethodHandles.dropArguments(MethodHandles.identity(String.class), 0, int.class, String.class); | |
| MethodHandle loop = MethodHandles.countedLoop(count, start, MH_step); // (v, i, _, pre, _) -> pre + " " + v | |
| assertEquals("na na na na na na na na na na na na na Lambdaman!", loop.invoke(13, "na", "Lambdaman!")); | |
| {} | |
| }} | |
| // Variation #3: | |
| MH_step = LOOKUP.findStatic(THIS_CLASS, "step3", methodType(S, S, I, S)); | |
| {{ | |
| {} /// JAVADOC | |
| // String s = "Lambdaman!", t = "na"; for (int i = 0; i < 13; ++i) { s = t + " " + s; } return s; | |
| // => a variation on a well known theme | |
| MethodType loopType = methodType(String.class, String.class, int.class, String.class); | |
| MethodHandle count = MethodHandles.dropArgumentsToMatch(MethodHandles.identity(int.class), 0, loopType.parameterList(), 1); | |
| MethodHandle start = MethodHandles.dropArgumentsToMatch(MethodHandles.identity(String.class), 0, loopType.parameterList(), 2); | |
| MethodHandle body = MethodHandles.dropArgumentsToMatch(MH_step, 2, loopType.parameterList(), 0); | |
| MethodHandle loop = MethodHandles.countedLoop(count, start, body); // (v, i, pre, _, _) -> pre + " " + v | |
| assertEquals("na na na na na na na na na na na na na Lambdaman!", loop.invoke("na", 13, "Lambdaman!")); | |
| {} | |
| }} | |
| } | |
| static List<String> reverseStep(List<String> r, String e) { | |
| r.add(0, e); | |
| return r; | |
| } | |
| static List<String> newArrayList() { return new ArrayList<>(); } | |
| @Test public void testIteratedLoop() throws Throwable { | |
| MethodHandle MH_newArrayList, MH_reverseStep; | |
| Class<?> L = List.class, S = String.class; | |
| MH_newArrayList = LOOKUP.findStatic(THIS_CLASS, "newArrayList", methodType(L)); | |
| MH_reverseStep = LOOKUP.findStatic(THIS_CLASS, "reverseStep", methodType(L, L, S)); | |
| {{ | |
| {} /// JAVADOC | |
| // reverse a list | |
| MethodHandle loop = MethodHandles.iteratedLoop(null, MH_newArrayList, MH_reverseStep); | |
| List<String> list = Arrays.asList("a", "b", "c", "d", "e"); | |
| List<String> reversedList = Arrays.asList("e", "d", "c", "b", "a"); | |
| assertEquals(reversedList, (List<String>) loop.invoke(list)); | |
| {} | |
| }} | |
| } | |
| @Test public void testFoldArguments3() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle trace = publicLookup().findVirtual(java.io.PrintStream.class, | |
| "println", methodType(void.class, String.class)) | |
| .bindTo(System.out); | |
| MethodHandle cat = lookup().findVirtual(String.class, | |
| "concat", methodType(String.class, String.class)); | |
| assertEquals("boojum", (String) cat.invokeExact("boo", "jum")); | |
| MethodHandle catTrace = foldArguments(cat, 1, trace); | |
| // also prints "jum": | |
| assertEquals("boojum", (String) catTrace.invokeExact("boo", "jum")); | |
| {} | |
| }} | |
| } | |
| @Test public void testAsCollector2() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| StringWriter swr = new StringWriter(); | |
| MethodHandle swWrite = LOOKUP.findVirtual(StringWriter.class, "write", methodType(void.class, char[].class, int.class, int.class)).bindTo(swr); | |
| MethodHandle swWrite4 = swWrite.asCollector(0, char[].class, 4); | |
| swWrite4.invoke('A', 'B', 'C', 'D', 1, 2); | |
| assertEquals("BC", swr.toString()); | |
| swWrite4.invoke('P', 'Q', 'R', 'S', 0, 4); | |
| assertEquals("BCPQRS", swr.toString()); | |
| swWrite4.invoke('W', 'X', 'Y', 'Z', 3, 1); | |
| assertEquals("BCPQRSZ", swr.toString()); | |
| {} | |
| }} | |
| } | |
| @Test public void testAsSpreader2() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| MethodHandle compare = LOOKUP.findStatic(Objects.class, "compare", methodType(int.class, Object.class, Object.class, Comparator.class)); | |
| MethodHandle compare2FromArray = compare.asSpreader(0, Object[].class, 2); | |
| Object[] ints = new Object[]{3, 9, 7, 7}; | |
| Comparator<Integer> cmp = (a, b) -> a - b; | |
| assertTrue((int) compare2FromArray.invoke(Arrays.copyOfRange(ints, 0, 2), cmp) < 0); | |
| assertTrue((int) compare2FromArray.invoke(Arrays.copyOfRange(ints, 1, 3), cmp) > 0); | |
| assertTrue((int) compare2FromArray.invoke(Arrays.copyOfRange(ints, 2, 4), cmp) == 0); | |
| {} | |
| }} | |
| } | |
| /* ---- TEMPLATE ---- | |
| @Test public void testFoo() throws Throwable { | |
| {{ | |
| {} /// JAVADOC | |
| {} | |
| }} | |
| } | |
| */ | |
| } |