1 /*
2 * Copyright (c) 2024, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23
24 import jdk.incubator.code.CodeReflection;
25 import java.util.function.BiConsumer;
26 import java.util.function.Consumer;
27 import java.util.function.Function;
28 import java.util.function.IntFunction;
29 import java.util.function.Supplier;
30
31 /*
32 * @test
33 * @summary Smoke test for code reflection with method reference expressions.
34 * @modules jdk.incubator.code
35 * @build MethodReferenceTest
36 * @build CodeReflectionTester
37 * @run main CodeReflectionTester MethodReferenceTest
38 */
39
40 public class MethodReferenceTest {
41
42 static void m_s(String s) {}
43
44 void m(String s) {}
45
46 @CodeReflection
47 @IR("""
48 func @"test1" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
49 %1 : java.type:"java.util.function.Consumer<java.lang.String>" = lambda (%2 : java.type:"java.lang.String")java.type:"void" -> {
50 %3 : Var<java.type:"java.lang.String"> = var %2 @"x$0";
51 %4 : java.type:"java.lang.String" = var.load %3;
52 invoke %4 @java.ref:"MethodReferenceTest::m_s(java.lang.String):void";
53 return;
54 };
55 %5 : Var<java.type:"java.util.function.Consumer<java.lang.String>"> = var %1 @"c";
56 return;
57 };
58 """)
59 void test1() {
60 Consumer<String> c = MethodReferenceTest::m_s;
61 }
62
63 @CodeReflection
64 @IR("""
65 func @"test2" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
66 %1 : java.type:"java.util.function.BiConsumer<MethodReferenceTest, java.lang.String>" = lambda (%2 : java.type:"MethodReferenceTest", %3 : java.type:"java.lang.String")java.type:"void" -> {
67 %4 : Var<java.type:"MethodReferenceTest"> = var %2 @"rec$";
68 %5 : Var<java.type:"java.lang.String"> = var %3 @"x$0";
69 %6 : java.type:"MethodReferenceTest" = var.load %4;
70 %7 : java.type:"java.lang.String" = var.load %5;
71 invoke %6 %7 @java.ref:"MethodReferenceTest::m(java.lang.String):void";
72 return;
73 };
74 %8 : Var<java.type:"java.util.function.BiConsumer<MethodReferenceTest, java.lang.String>"> = var %1 @"bc";
75 return;
76 };
77 """)
78 void test2() {
79 BiConsumer<MethodReferenceTest, String> bc = MethodReferenceTest::m;
80 }
81
82 @CodeReflection
83 @IR("""
84 func @"test3" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
85 %1 : java.type:"java.util.function.Consumer<java.lang.String>" = lambda (%2 : java.type:"java.lang.String")java.type:"void" -> {
86 %3 : Var<java.type:"java.lang.String"> = var %2 @"x$0";
87 %4 : java.type:"java.lang.String" = var.load %3;
88 invoke %0 %4 @java.ref:"MethodReferenceTest::m(java.lang.String):void";
89 return;
90 };
91 %5 : Var<java.type:"java.util.function.Consumer<java.lang.String>"> = var %1 @"c";
92 return;
93 };
94 """)
95 void test3() {
96 Consumer<String> c = this::m;
97 }
98
99 class A<T> {
100 T m(T t) { return t; }
101 }
102
103 <T> A<T> a(T t) { return null; }
104
105 @CodeReflection
106 @IR("""
107 func @"test4" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
108 %1 : java.type:"java.lang.String" = constant @"s";
109 %2 : java.type:"MethodReferenceTest::A<java.lang.String>" = invoke %0 %1 @java.ref:"MethodReferenceTest::a(java.lang.Object):MethodReferenceTest::A";
110 %3 : Var<java.type:"MethodReferenceTest::A<java.lang.String>"> = var %2 @"rec$";
111 %4 : java.type:"java.util.function.Function<java.lang.String, java.lang.String>" = lambda (%5 : java.type:"java.lang.String")java.type:"java.lang.String" -> {
112 %6 : Var<java.type:"java.lang.String"> = var %5 @"x$0";
113 %7 : java.type:"MethodReferenceTest::A<java.lang.String>" = var.load %3;
114 %8 : java.type:"java.lang.String" = var.load %6;
115 %9 : java.type:"java.lang.String" = invoke %7 %8 @java.ref:"MethodReferenceTest::A::m(java.lang.Object):java.lang.Object";
116 return %9;
117 };
118 %10 : Var<java.type:"java.util.function.Function<java.lang.String, java.lang.String>"> = var %4 @"f";
119 return;
120 };
121 """)
122 void test4() {
123 Function<String, String> f = a("s")::m;
124 }
125
126 @CodeReflection
127 @IR("""
128 func @"test5" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
129 %1 : java.type:"java.io.PrintStream" = field.load @java.ref:"java.lang.System::out:java.io.PrintStream";
130 %2 : Var<java.type:"java.io.PrintStream"> = var %1 @"rec$";
131 %3 : java.type:"java.util.function.Consumer<java.lang.String>" = lambda (%4 : java.type:"java.lang.String")java.type:"void" -> {
132 %5 : Var<java.type:"java.lang.String"> = var %4 @"x$0";
133 %6 : java.type:"java.io.PrintStream" = var.load %2;
134 %7 : java.type:"java.lang.String" = var.load %5;
135 invoke %6 %7 @java.ref:"java.io.PrintStream::println(java.lang.String):void";
136 return;
137 };
138 %8 : Var<java.type:"java.util.function.Consumer<java.lang.String>"> = var %3 @"c3";
139 return;
140 };
141 """)
142 void test5() {
143 Consumer<String> c3 = System.out::println;
144 }
145
146 static class X {
147 X(int i) {}
148 }
149
150 @CodeReflection
151 @IR("""
152 func @"test6" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
153 %1 : java.type:"java.util.function.Function<java.lang.Integer, MethodReferenceTest$X>" = lambda (%2 : java.type:"java.lang.Integer")java.type:"MethodReferenceTest$X" -> {
154 %3 : Var<java.type:"java.lang.Integer"> = var %2 @"x$0";
155 %4 : java.type:"java.lang.Integer" = var.load %3;
156 %5 : java.type:"int" = invoke %4 @java.ref:"java.lang.Integer::intValue():int";
157 %6 : java.type:"MethodReferenceTest$X" = new %5 @java.ref:"MethodReferenceTest$X::(int)";
158 return %6;
159 };
160 %7 : Var<java.type:"java.util.function.Function<java.lang.Integer, MethodReferenceTest$X>"> = var %1 @"xNew";
161 return;
162 };
163 """)
164 void test6() {
165 Function<Integer, X> xNew = X::new;
166 }
167
168 @CodeReflection
169 @IR("""
170 func @"test7" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
171 %1 : java.type:"java.util.function.Supplier<MethodReferenceTest::A<java.lang.String>>" = lambda ()java.type:"MethodReferenceTest::A<java.lang.String>" -> {
172 %2 : java.type:"MethodReferenceTest::A<java.lang.String>" = new %0 @java.ref:"MethodReferenceTest::A::(MethodReferenceTest)";
173 return %2;
174 };
175 %3 : Var<java.type:"java.util.function.Supplier<MethodReferenceTest::A<java.lang.String>>"> = var %1 @"aNew";
176 return;
177 };
178 """)
179 void test7() {
180 Supplier<A<String>> aNew = A::new;
181 }
182
183 @CodeReflection
184 @IR("""
185 func @"test8" (%0 : java.type:"MethodReferenceTest")java.type:"void" -> {
186 %1 : java.type:"java.util.function.IntFunction<MethodReferenceTest::A<java.lang.String>[]>" = lambda (%2 : java.type:"int")java.type:"MethodReferenceTest::A<java.lang.String>[]" -> {
187 %3 : Var<java.type:"int"> = var %2 @"x$0";
188 %4 : java.type:"int" = var.load %3;
189 %5 : java.type:"MethodReferenceTest::A[]" = new %4 @java.ref:"MethodReferenceTest::A[]::(int)";
190 return %5;
191 };
192 %6 : Var<java.type:"java.util.function.IntFunction<MethodReferenceTest::A<java.lang.String>[]>"> = var %1 @"aNewArray";
193 return;
194 };
195 """)
196 void test8() {
197 IntFunction<A<String>[]> aNewArray = A[]::new;
198 }
199
200 }