ZhouXY108/hutool
1
0
Fork 0
mirror of https://gitee.com/chinabugotech/hutool.git synced 2025-04-19 03:01:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix code

Browse Source
This commit is contained in:
Looly 2023-01-17 11:26:16 +08:00
parent 4c8a06e749
commit 2f38e2e138
3 changed files with 69 additions and 63 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

31
hutool-core/src/main/java/cn/hutool/core/lang/func/LambdaFactory.java
View File

@ -51,8 +51,9 @@ public class LambdaFactory {
* @param methodName 方法名称
* @param paramTypes 方法参数数组
* @return 接受Lambda的函数式接口对象
* @param <F> Function类型
*/
public static <F> F build(Class<F> functionInterfaceType, Class methodClass, String methodName, Class... paramTypes) {
public static <F> F build(final Class<F> functionInterfaceType, final Class<?> methodClass, final String methodName, final Class<?>... paramTypes) {
return build(functionInterfaceType, MethodUtil.getMethod(methodClass, methodName, paramTypes));
}
@ -62,30 +63,31 @@ public class LambdaFactory {
* @param functionInterfaceType 接受Lambda的函数式接口类型
* @param method 方法对象
* @return 接受Lambda的函数式接口对象
* @param <F> Function类型
*/
public static <F> F build(Class<F> functionInterfaceType, Method method) {
public static <F> F build(final Class<F> functionInterfaceType, final Method method) {
Assert.notNull(functionInterfaceType);
Assert.notNull(method);
MutableEntry<Class<?>, Method> cacheKey = new MutableEntry<>(functionInterfaceType, method);
final MutableEntry<Class<?>, Method> cacheKey = new MutableEntry<>(functionInterfaceType, method);
//noinspection unchecked
return (F) CACHE.computeIfAbsent(cacheKey, key -> {
List<Method> abstractMethods = Arrays.stream(functionInterfaceType.getMethods())
final List<Method> abstractMethods = Arrays.stream(functionInterfaceType.getMethods())
.filter(m -> Modifier.isAbstract(m.getModifiers()))
.collect(Collectors.toList());
Assert.equals(abstractMethods.size(), 1, "不支持非函数式接口");
if (!method.isAccessible()) {
method.setAccessible(true);
}
Method invokeMethod = abstractMethods.get(0);
MethodHandles.Lookup caller = LookupFactory.lookup(method.getDeclaringClass());
String invokeName = invokeMethod.getName();
MethodType invokedType = methodType(functionInterfaceType);
MethodType samMethodType = methodType(invokeMethod.getReturnType(), invokeMethod.getParameterTypes());
MethodHandle implMethod = Opt.ofTry(() -> caller.unreflect(method)).get();
MethodType insMethodType = methodType(method.getReturnType(), method.getDeclaringClass(), method.getParameterTypes());
boolean isSerializable = Serializable.class.isAssignableFrom(functionInterfaceType);
final Method invokeMethod = abstractMethods.get(0);
final MethodHandles.Lookup caller = LookupFactory.lookup(method.getDeclaringClass());
final String invokeName = invokeMethod.getName();
final MethodType invokedType = methodType(functionInterfaceType);
final MethodType samMethodType = methodType(invokeMethod.getReturnType(), invokeMethod.getParameterTypes());
final MethodHandle implMethod = Opt.ofTry(() -> caller.unreflect(method)).get();
final MethodType insMethodType = methodType(method.getReturnType(), method.getDeclaringClass(), method.getParameterTypes());
final boolean isSerializable = Serializable.class.isAssignableFrom(functionInterfaceType);
try {
CallSite callSite = isSerializable ?
final CallSite callSite = isSerializable ?
LambdaMetafactory.altMetafactory(
caller,
invokeName,
@ -103,8 +105,9 @@ public class LambdaFactory {
implMethod,
insMethodType
);
//noinspection unchecked
return (F) callSite.getTarget().invoke();
} catch (Throwable e) {
} catch (final Throwable e) {
throw new UtilException(e);
}
});

75
hutool-core/src/test/java/cn/hutool/core/lang/func/LambdaFactoryTest.java
View File

@ -1,7 +1,6 @@
package cn.hutool.core.lang.func;
import cn.hutool.core.collection.ListUtil;
import cn.hutool.core.exceptions.UtilException;
import cn.hutool.core.reflect.MethodHandleUtil;
import lombok.Data;
import lombok.Getter;
@ -12,7 +11,11 @@ import org.junit.Test;
import org.junit.runner.RunWith;
import org.junit.runners.Parameterized;
import java.lang.invoke.*;
import java.lang.invoke.LambdaConversionException;
import java.lang.invoke.MethodHandle;
import java.lang.invoke.MethodHandleProxies;
import java.lang.invoke.MethodHandles;
import java.lang.invoke.MethodType;
import java.lang.reflect.Method;
import java.util.Arrays;
import java.util.Collection;
@ -31,25 +34,25 @@ public class LambdaFactoryTest {
public void testMethodNotMatch() {
try {
LambdaFactory.build(Function.class, Something.class, "setId", Long.class);
} catch (Exception e) {
} catch (final Exception e) {
Assert.assertTrue(e.getCause() instanceof LambdaConversionException);
}
}
@Test
public void buildLambdaTest() {
Something something = new Something();
final Something something = new Something();
something.setId(1L);
something.setName("name");
Function<Something, Long> get11 = LambdaFactory.build(Function.class, Something.class, "getId");
Function<Something, Long> get12 = LambdaFactory.build(Function.class, Something.class, "getId");
final Function<Something, Long> get11 = LambdaFactory.build(Function.class, Something.class, "getId");
final Function<Something, Long> get12 = LambdaFactory.build(Function.class, Something.class, "getId");
Assert.assertEquals(get11, get12);
Assert.assertEquals(something.getId(), get11.apply(something));
String name = "sname";
BiConsumer<Something, String> set = LambdaFactory.build(BiConsumer.class, Something.class, "setName", String.class);
final String name = "sname";
final BiConsumer<Something, String> set = LambdaFactory.build(BiConsumer.class, Something.class, "setName", String.class);
set.accept(something, name);
Assert.assertEquals(something.getName(), name);
@ -136,19 +139,19 @@ public class LambdaFactoryTest {
@Test
@SneakyThrows
public void lambdaGetPerformanceTest() {
Something something = new Something();
final Something something = new Something();
something.setId(1L);
something.setName("name");
Method getByReflect = Something.class.getMethod("getId");
MethodHandle getByMh = MethodHandleUtil.findMethod(Something.class, "getId", MethodType.methodType(Long.class));
Function getByProxy = MethodHandleProxies.asInterfaceInstance(Function.class, MethodHandles.lookup().unreflect(getByReflect));
Function getByLambda = LambdaFactory.build(Function.class, getByReflect);
Task lambdaTask = new Task("lambda", () -> getByLambda.apply(something));
Task mhTask = new Task("mh", () -> getByMh.invoke(something));
Task proxyTask = new Task("proxy", () -> getByProxy.apply(something));
Task reflectTask = new Task("reflect", () -> getByReflect.invoke(something));
Task hardCodeTask = new Task("hardCode", () -> something.getId());
Task[] tasks = {hardCodeTask, lambdaTask, mhTask, proxyTask, reflectTask};
final Method getByReflect = Something.class.getMethod("getId");
final MethodHandle getByMh = MethodHandleUtil.findMethod(Something.class, "getId", MethodType.methodType(Long.class));
final Function getByProxy = MethodHandleProxies.asInterfaceInstance(Function.class, MethodHandles.lookup().unreflect(getByReflect));
final Function getByLambda = LambdaFactory.build(Function.class, getByReflect);
final Task lambdaTask = new Task("lambda", () -> getByLambda.apply(something));
final Task mhTask = new Task("mh", () -> getByMh.invoke(something));
final Task proxyTask = new Task("proxy", () -> getByProxy.apply(something));
final Task reflectTask = new Task("reflect", () -> getByReflect.invoke(something));
final Task hardCodeTask = new Task("hardCode", () -> something.getId());
final Task[] tasks = {hardCodeTask, lambdaTask, mhTask, proxyTask, reflectTask};
loop(count, tasks);
}
@ -211,43 +214,43 @@ public class LambdaFactoryTest {
@Test
@SneakyThrows
public void lambdaSetPerformanceTest() {
Something something = new Something();
final Something something = new Something();
something.setId(1L);
something.setName("name");
Method setByReflect = Something.class.getMethod("setName", String.class);
MethodHandle setByMh = MethodHandleUtil.findMethod(Something.class, "setName", MethodType.methodType(Void.TYPE, String.class));
BiConsumer setByProxy = MethodHandleProxies.asInterfaceInstance(BiConsumer.class, setByMh);
BiConsumer setByLambda = LambdaFactory.build(BiConsumer.class, setByReflect);
String name = "name1";
Task lambdaTask = new Task("lambda", () -> {
final Method setByReflect = Something.class.getMethod("setName", String.class);
final MethodHandle setByMh = MethodHandleUtil.findMethod(Something.class, "setName", MethodType.methodType(Void.TYPE, String.class));
final BiConsumer setByProxy = MethodHandleProxies.asInterfaceInstance(BiConsumer.class, setByMh);
final BiConsumer setByLambda = LambdaFactory.build(BiConsumer.class, setByReflect);
final String name = "name1";
final Task lambdaTask = new Task("lambda", () -> {
setByLambda.accept(something, name);
return null;
});
Task proxyTask = new Task("proxy", () -> {
final Task proxyTask = new Task("proxy", () -> {
setByProxy.accept(something, name);
return null;
});
Task mhTask = new Task("mh", () -> {
final Task mhTask = new Task("mh", () -> {
setByMh.invoke(something, name);
return null;
});
Task reflectTask = new Task("reflect", () -> {
final Task reflectTask = new Task("reflect", () -> {
setByReflect.invoke(something, name);
return null;
});
Task hardCodeTask = new Task("hardCode", () -> {
final Task hardCodeTask = new Task("hardCode", () -> {
something.setName(name);
return null;
});
Task[] tasks = {hardCodeTask, lambdaTask, proxyTask, mhTask, reflectTask};
final Task[] tasks = {hardCodeTask, lambdaTask, proxyTask, mhTask, reflectTask};
loop(count, tasks);
}
@SneakyThrows
private void loop(int count, Task... tasks) {
private void loop(final int count, final Task... tasks) {
Arrays.stream(tasks)
.peek(task -> {
LambdaFactoryTest.SupplierThrowable runnable = task.getRunnable();
final LambdaFactoryTest.SupplierThrowable runnable = task.getRunnable();
long cost = System.nanoTime();
for (int i = 0; i < count; i++) {
runnable.get();
@ -271,13 +274,13 @@ public class LambdaFactoryTest {
@Setter
private Integer count;
public Task(String name, LambdaFactoryTest.SupplierThrowable<?> runnable) {
public Task(final String name, final LambdaFactoryTest.SupplierThrowable<?> runnable) {
this.name = name;
this.runnable = runnable;
}
public String format() {
TimeUnit timeUnit = TimeUnit.NANOSECONDS;
final TimeUnit timeUnit = TimeUnit.NANOSECONDS;
return String.format("%-10s 运行%d次耗时 %d %s", name, count, timeUnit.convert(cost, TimeUnit.NANOSECONDS), timeUnit.name());
}
}
@ -291,7 +294,7 @@ public class LambdaFactoryTest {
default T get() {
try {
return get0();
} catch (Throwable e) {
} catch (final Throwable e) {
throw new RuntimeException(e);
}
}

26
hutool-core/src/test/java/cn/hutool/core/lang/func/LambdaUtilTest.java
View File

@ -144,7 +144,7 @@ public class LambdaUtilTest {
Assert.assertEquals(MyTeacher.class, LambdaUtil.getRealClass(lambda));
}, () -> {
// 数组测试
final SerConsumer<String[]> lambda = (String[] stringList) -> {
final SerConsumer<String[]> lambda = (final String[] stringList) -> {
};
Assert.assertEquals(String[].class, LambdaUtil.getRealClass(lambda));
}).forEach(Runnable::run);
@ -152,11 +152,11 @@ public class LambdaUtilTest {
@Test
public void getterTest() {
Bean bean = new Bean();
final Bean bean = new Bean();
bean.setId(2L);
Function<Bean, Long> getId = LambdaUtil.buildGetter(MethodUtil.getMethod(Bean.class, "getId"));
Function<Bean, Long> getId2 = LambdaUtil.buildGetter(Bean.class, Bean.Fields.id);
final Function<Bean, Long> getId = LambdaUtil.buildGetter(MethodUtil.getMethod(Bean.class, "getId"));
final Function<Bean, Long> getId2 = LambdaUtil.buildGetter(Bean.class, Bean.Fields.id);
Assert.assertEquals(getId, getId2);
Assert.assertEquals(bean.getId(), getId.apply(bean));
@ -164,13 +164,13 @@ public class LambdaUtilTest {
@Test
public void setterTest() {
Bean bean = new Bean();
final Bean bean = new Bean();
bean.setId(2L);
bean.setFlag(false);
BiConsumer<Bean, Long> setId = LambdaUtil.buildSetter(MethodUtil.getMethod(Bean.class, "setId", Long.class));
BiConsumer<Bean, Long> setId2 = LambdaUtil.buildSetter(Bean.class, Bean.Fields.id);
BiConsumer<Bean, Boolean> setFlag = LambdaUtil.buildSetter(Bean.class, Bean.Fields.flag);
final BiConsumer<Bean, Long> setId = LambdaUtil.buildSetter(MethodUtil.getMethod(Bean.class, "setId", Long.class));
final BiConsumer<Bean, Long> setId2 = LambdaUtil.buildSetter(Bean.class, Bean.Fields.id);
final BiConsumer<Bean, Boolean> setFlag = LambdaUtil.buildSetter(Bean.class, Bean.Fields.flag);
Assert.assertEquals(setId, setId2);
setId.accept(bean, 3L);
@ -181,12 +181,12 @@ public class LambdaUtilTest {
@Test
public void lambdaTest() {
Bean bean = new Bean();
final Bean bean = new Bean();
bean.setId(1L);
bean.setPid(0L);
bean.setFlag(true);
BiFunction<Bean, String, Tuple> uniqueKeyFunction = LambdaUtil.build(BiFunction.class, Bean.class, "uniqueKey", String.class);
Function4<Tuple, Bean, String, Integer, Double> paramsFunction = LambdaUtil.build(Function4.class, Bean.class, "params", String.class, Integer.class, Double.class);
final BiFunction<Bean, String, Tuple> uniqueKeyFunction = LambdaUtil.build(BiFunction.class, Bean.class, "uniqueKey", String.class);
final Function4<Tuple, Bean, String, Integer, Double> paramsFunction = LambdaUtil.build(Function4.class, Bean.class, "params", String.class, Integer.class, Double.class);
Assert.assertEquals(bean.uniqueKey("test"), uniqueKeyFunction.apply(bean, "test"));
Assert.assertEquals(bean.params("test", 1, 0.5), paramsFunction.apply(bean, "test", 1, 0.5));
}
@ -203,11 +203,11 @@ public class LambdaUtilTest {
Long pid;
boolean flag;
private Tuple uniqueKey(String name) {
private Tuple uniqueKey(final String name) {
return new Tuple(id, pid, flag, name);
}
public Tuple params(String name, Integer length, Double score) {
public Tuple params(final String name, final Integer length, final Double score) {
return new Tuple(name, length, score);
}