lxy/redkale
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SerializedLambda优化

Browse Source
This commit is contained in:
redkale
2023-06-29 11:31:45 +08:00
parent 7e7e352924
commit 773a260438
2 changed files with 24 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
src/main/java/org/redkale/util/SerializedLambda.java
View File

@@ -3,10 +3,9 @@
*/
package org.redkale.util;
import java.io.*;
import java.io.Serializable;
import java.lang.invoke.MethodHandleInfo;
import java.util.Objects;
import java.util.concurrent.ConcurrentHashMap;
/**
* 完全复制java.lang.invoke.SerializedLambda类源码，必须保持字段信息一样
@@ -249,29 +248,4 @@ public class SerializedLambda implements Serializable {
"numCaptured", capturedArgs.length);
}
private static final ConcurrentHashMap<Class, String> lambdaFieldNameCache = new ConcurrentHashMap();
public static String readLambdaFieldName(Serializable func) {
if (!func.getClass().isSynthetic()) { //必须是Lambda表达式的合成类
throw new RedkaleException("Not a synthetic lambda class");
}
return lambdaFieldNameCache.computeIfAbsent(func.getClass(), clazz -> {
try {
ByteArrayOutputStream out = new ByteArrayOutputStream();
ObjectOutputStream oos = new ObjectOutputStream(out);
oos.writeObject(func);
ObjectInputStream in = new ObjectInputStream(new ByteArrayInputStream(out.toByteArray())) {
@Override
protected Class<?> resolveClass(ObjectStreamClass desc) throws IOException, ClassNotFoundException {
Class<?> clazz = super.resolveClass(desc);
return clazz == java.lang.invoke.SerializedLambda.class ? SerializedLambda.class : clazz;
}
};
return Utility.readFieldName(((SerializedLambda) in.readObject()).getImplMethodName());
} catch (Exception e) {
throw new RedkaleException(e);
}
});
}
}

25
src/main/java/org/redkale/util/Utility.java
View File

@@ -282,7 +282,7 @@ public final class Utility {
return lambdaFieldNameCache.computeIfAbsent(func.getClass(), clazz -> {
try {
if (nativeImageEnv) {
return org.redkale.util.SerializedLambda.readLambdaFieldName(func);
return readLambdaFieldNameFromBytes(func);
} else {
MethodHandles.Lookup lookup = MethodHandles.privateLookupIn(func.getClass(), MethodHandles.lookup());
MethodHandle mh = lookup.findVirtual(func.getClass(), "writeReplace", MethodType.methodType(Object.class));
@@ -290,11 +290,32 @@ public final class Utility {
return readFieldName(methodName);
}
} catch (Throwable e) {
throw new RedkaleException(e);
if (!nativeImageEnv) {
return readLambdaFieldNameFromBytes(func);
}
throw e instanceof RedkaleException ? (RedkaleException) e : new RedkaleException(e);
}
});
}
private static String readLambdaFieldNameFromBytes(Serializable func) {
try {
ByteArrayOutputStream out = new ByteArrayOutputStream();
ObjectOutputStream oos = new ObjectOutputStream(out);
oos.writeObject(func);
ObjectInputStream in = new ObjectInputStream(new ByteArrayInputStream(out.toByteArray())) {
@Override
protected Class<?> resolveClass(ObjectStreamClass desc) throws IOException, ClassNotFoundException {
Class<?> clazz = super.resolveClass(desc);
return clazz == java.lang.invoke.SerializedLambda.class ? org.redkale.util.SerializedLambda.class : clazz;
}
};
return Utility.readFieldName(((org.redkale.util.SerializedLambda) in.readObject()).getImplMethodName());
} catch (Throwable e) {
throw new RedkaleException(e);
}
}
static String readFieldName(String methodName) {
String name;
if (methodName.startsWith("is")) {