blob: ef63f167b187093de9fe73f8970b64e1692c0438 [file] [log] [blame]
/*
* Copyright (C) 2018 The Android Open Source Project
*
* 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.
*/
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodType;
import java.lang.invoke.VarHandle;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
public class Main {
private static final int ITERATIONS = 100;
private static final VarHandle widgetIdVarHandle;
private static int initialHotnessCounter;
public static native int getHotnessCounter(Class<?> cls, String methodName);
public static class Widget {
public Widget(int id) {
this.id = id;
}
int getId() {
return id;
}
int id;
}
static {
try {
widgetIdVarHandle = MethodHandles.lookup().findVarHandle(Widget.class, "id", int.class);
} catch (Exception e) {
throw new Error(e);
}
}
private static void assertEquals(int i1, int i2) {
if (i1 == i2) {
return;
}
throw new AssertionError("assertEquals i1: " + i1 + ", i2: " + i2);
}
private static void assertEquals(Object o, Object p) {
if (o == p) {
return;
}
if (o != null && p != null && o.equals(p)) {
return;
}
throw new AssertionError("assertEquals: o1: " + o + ", o2: " + p);
}
private static void fail() {
System.out.println("fail");
Thread.dumpStack();
}
private static void fail(String message) {
System.out.println("fail: " + message);
Thread.dumpStack();
}
private static void testMethodHandleCounters() throws Throwable {
for (int i = 0; i < ITERATIONS; ++i) {
// Regular MethodHandle invocations
MethodHandle mh =
MethodHandles.lookup()
.findConstructor(
Widget.class, MethodType.methodType(void.class, int.class));
Widget w = (Widget) mh.invoke(3);
w = (Widget) mh.invokeExact(3);
assertEquals(initialHotnessCounter, getHotnessCounter(MethodHandle.class, "invoke"));
assertEquals(initialHotnessCounter, getHotnessCounter(MethodHandle.class, "invokeExact"));
// Reflective MethodHandle invocations
String[] methodNames = {"invoke", "invokeExact"};
for (String methodName : methodNames) {
Method invokeMethod = MethodHandle.class.getMethod(methodName, Object[].class);
MethodHandle instance =
MethodHandles.lookup()
.findVirtual(
Widget.class, "getId", MethodType.methodType(int.class));
try {
invokeMethod.invoke(instance, new Object[] {new Object[] {}});
fail();
} catch (InvocationTargetException ite) {
assertEquals(ite.getCause().getClass(), UnsupportedOperationException.class);
}
}
assertEquals(initialHotnessCounter,
getHotnessCounter(MethodHandle.class, "invoke"));
assertEquals(initialHotnessCounter,
getHotnessCounter(MethodHandle.class, "invokeExact"));
}
System.out.println("MethodHandle OK");
}
private static void testVarHandleCounters() throws Throwable {
Widget w = new Widget(0);
for (int i = 0; i < ITERATIONS; ++i) {
// Regular accessor invocations
widgetIdVarHandle.set(w, i);
assertEquals(i, widgetIdVarHandle.get(w));
assertEquals(initialHotnessCounter, getHotnessCounter(VarHandle.class, "set"));
assertEquals(initialHotnessCounter, getHotnessCounter(VarHandle.class, "get"));
// Reflective accessor invocations
for (String accessorName : new String[] {"get", "set"}) {
Method setMethod = VarHandle.class.getMethod(accessorName, Object[].class);
try {
setMethod.invoke(widgetIdVarHandle, new Object[] {new Object[0]});
fail();
} catch (InvocationTargetException ite) {
assertEquals(ite.getCause().getClass(), UnsupportedOperationException.class);
}
}
assertEquals(initialHotnessCounter, getHotnessCounter(VarHandle.class, "set"));
assertEquals(initialHotnessCounter, getHotnessCounter(VarHandle.class, "get"));
}
System.out.println("VarHandle OK");
}
public static void main(String[] args) throws Throwable {
System.loadLibrary(args[0]);
initialHotnessCounter = getHotnessCounter(VarHandle.class, "set");
testMethodHandleCounters();
testVarHandleCounters();
}
}