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 java.lang.runtime.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 * @build MethodReferenceTest
35 * @build CodeReflectionTester
36 * @run main CodeReflectionTester MethodReferenceTest
37 */
38
39 public class MethodReferenceTest {
40
41 static void m_s(String s) {}
42
43 void m(String s) {}
44
45 @CodeReflection
46 @IR("""
47 func @"test1" (%0 : MethodReferenceTest)void -> {
48 %1 : java.util.function.Consumer<java.lang.String> = lambda (%2 : java.lang.String)void -> {
49 %3 : Var<java.lang.String> = var %2 @"x$0";
50 %4 : java.lang.String = var.load %3;
51 invoke %4 @"MethodReferenceTest::m_s(java.lang.String)void";
52 return;
53 };
54 %5 : Var<java.util.function.Consumer<java.lang.String>> = var %1 @"c";
55 return;
56 };
57 """)
58 void test1() {
59 Consumer<String> c = MethodReferenceTest::m_s;
60 }
61
62 @CodeReflection
63 @IR("""
64 func @"test2" (%0 : MethodReferenceTest)void -> {
65 %1 : java.util.function.BiConsumer<MethodReferenceTest, java.lang.String> = lambda (%2 : MethodReferenceTest, %3 : java.lang.String)void -> {
66 %4 : Var<MethodReferenceTest> = var %2 @"rec$";
67 %5 : Var<java.lang.String> = var %3 @"x$0";
68 %6 : MethodReferenceTest = var.load %4;
69 %7 : java.lang.String = var.load %5;
70 invoke %6 %7 @"MethodReferenceTest::m(java.lang.String)void";
71 return;
72 };
73 %8 : Var<java.util.function.BiConsumer<MethodReferenceTest, java.lang.String>> = var %1 @"bc";
74 return;
75 };
76 """)
77 void test2() {
78 BiConsumer<MethodReferenceTest, String> bc = MethodReferenceTest::m;
79 }
80
81 @CodeReflection
82 @IR("""
83 func @"test3" (%0 : MethodReferenceTest)void -> {
84 %1 : java.util.function.Consumer<java.lang.String> = lambda (%2 : java.lang.String)void -> {
85 %3 : Var<java.lang.String> = var %2 @"x$0";
86 %4 : java.lang.String = var.load %3;
87 invoke %0 %4 @"MethodReferenceTest::m(java.lang.String)void";
88 return;
89 };
90 %5 : Var<java.util.function.Consumer<java.lang.String>> = var %1 @"c";
91 return;
92 };
93 """)
94 void test3() {
95 Consumer<String> c = this::m;
96 }
97
98 class A<T> {
99 T m(T t) { return t; }
100 }
101
102 <T> A<T> a(T t) { return null; }
103
104 @CodeReflection
105 @IR("""
106 func @"test4" (%0 : MethodReferenceTest)void -> {
107 %1 : java.lang.String = constant @"s";
108 %2 : MethodReferenceTest$A<java.lang.String> = invoke %0 %1 @"MethodReferenceTest::a(java.lang.Object)MethodReferenceTest$A";
109 %3 : Var<MethodReferenceTest$A<java.lang.String>> = var %2 @"rec$";
110 %4 : java.util.function.Function<java.lang.String, java.lang.String> = lambda (%5 : java.lang.String)java.lang.String -> {
111 %6 : Var<java.lang.String> = var %5 @"x$0";
112 %7 : MethodReferenceTest$A<java.lang.String> = var.load %3;
113 %8 : java.lang.String = var.load %6;
114 %9 : java.lang.String = invoke %7 %8 @"MethodReferenceTest$A::m(java.lang.Object)java.lang.Object";
115 return %9;
116 };
117 %10 : Var<java.util.function.Function<java.lang.String, java.lang.String>> = var %4 @"f";
118 return;
119 };
120 """)
121 void test4() {
122 Function<String, String> f = a("s")::m;
123 }
124
125 @CodeReflection
126 @IR("""
127 func @"test5" (%0 : MethodReferenceTest)void -> {
128 %1 : java.io.PrintStream = field.load @"java.lang.System::out()java.io.PrintStream";
129 %2 : Var<java.io.PrintStream> = var %1 @"rec$";
130 %3 : java.util.function.Consumer<java.lang.String> = lambda (%4 : java.lang.String)void -> {
131 %5 : Var<java.lang.String> = var %4 @"x$0";
132 %6 : java.io.PrintStream = var.load %2;
133 %7 : java.lang.String = var.load %5;
134 invoke %6 %7 @"java.io.PrintStream::println(java.lang.String)void";
135 return;
136 };
137 %8 : Var<java.util.function.Consumer<java.lang.String>> = var %3 @"c3";
138 return;
139 };
140 """)
141 void test5() {
142 Consumer<String> c3 = System.out::println;
143 }
144
145 static class X {
146 X(int i) {}
147 }
148
149 @CodeReflection
150 @IR("""
151 func @"test6" (%0 : MethodReferenceTest)void -> {
152 %1 : java.util.function.Function<java.lang.Integer, MethodReferenceTest$X> = lambda (%2 : java.lang.Integer)MethodReferenceTest$X -> {
153 %3 : Var<java.lang.Integer> = var %2 @"x$0";
154 %4 : java.lang.Integer = var.load %3;
155 %5 : int = invoke %4 @"java.lang.Integer::intValue()int";
156 %6 : MethodReferenceTest$X = new %5 @"func<MethodReferenceTest$X, int>";
157 return %6;
158 };
159 %7 : Var<java.util.function.Function<java.lang.Integer, MethodReferenceTest$X>> = var %1 @"xNew";
160 return;
161 };
162 """)
163 void test6() {
164 Function<Integer, X> xNew = X::new;
165 }
166
167 @CodeReflection
168 @IR("""
169 func @"test7" (%0 : MethodReferenceTest)void -> {
170 %1 : java.util.function.Supplier<MethodReferenceTest$A<java.lang.String>> = lambda ()MethodReferenceTest$A<java.lang.String> -> {
171 %2 : MethodReferenceTest$A<java.lang.String> = new %0 @"func<MethodReferenceTest$A>";
172 return %2;
173 };
174 %3 : Var<java.util.function.Supplier<MethodReferenceTest$A<java.lang.String>>> = var %1 @"aNew";
175 return;
176 };
177 """)
178 void test7() {
179 Supplier<A<String>> aNew = A::new;
180 }
181
182 @CodeReflection
183 @IR("""
184 func @"test8" (%0 : MethodReferenceTest)void -> {
185 %1 : java.util.function.IntFunction<MethodReferenceTest$A<java.lang.String>[]> = lambda (%2 : int)MethodReferenceTest$A<java.lang.String>[] -> {
186 %3 : Var<int> = var %2 @"x$0";
187 %4 : int = var.load %3;
188 %5 : MethodReferenceTest$A[] = new %4 @"func<MethodReferenceTest$A[], int>";
189 return %5;
190 };
191 %6 : Var<java.util.function.IntFunction<MethodReferenceTest$A<java.lang.String>[]>> = var %1 @"aNewArray";
192 return;
193 };
194 """)
195 void test8() {
196 IntFunction<A<String>[]> aNewArray = A[]::new;
197 }
198
199 }