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

Utility

Browse Source
This commit is contained in:
redkale
2024-10-06 23:30:51 +08:00
parent 34fef28ed6
commit c8cd1e23ef
1 changed files with 38 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -65,12 +65,13 @@ public final class Utility {
JAVA_RECORD_CLASS = clz; JAVA_RECORD_CLASS = clz;
} }
private static final MethodHandles.Lookup defaultLookup; private static final MethodHandles.Lookup trustedLookup;
private static final boolean compactStrings;
static { static {
MethodHandles.Lookup defaultLookup0 = null; boolean compactStrings0 = true;
MethodHandles.Lookup trustedLookup0 = null;
try { try {
final Class lookupClass = MethodHandles.Lookup.class;
final ClassLoader loader = Thread.currentThread().getContextClassLoader(); final ClassLoader loader = Thread.currentThread().getContextClassLoader();
Class unsafeClass = loader.loadClass("sun.misc.Unsafe"); Class unsafeClass = loader.loadClass("sun.misc.Unsafe");
Field safeField = unsafeClass.getDeclaredField("theUnsafe"); Field safeField = unsafeClass.getDeclaredField("theUnsafe");
@@ -78,15 +79,24 @@ public final class Utility {
safeField.setAccessible(true); safeField.setAccessible(true);
final Object usafe = safeField.get(null); final Object usafe = safeField.get(null);
Method staticFieldOffsetMethod = usafe.getClass().getMethod("staticFieldOffset", Field.class); Method staticFieldOffsetMethod = usafe.getClass().getMethod("staticFieldOffset", Field.class);
Class lookupClass = MethodHandles.Lookup.class;
Field implField = lookupClass.getDeclaredField("IMPL_LOOKUP");
long implOffset = (Long) staticFieldOffsetMethod.invoke(usafe, implField);
Method getObjectMethod = usafe.getClass().getMethod("getObject", Object.class, long.class); Method getObjectMethod = usafe.getClass().getMethod("getObject", Object.class, long.class);
Field implLookup = lookupClass.getDeclaredField("IMPL_LOOKUP"); trustedLookup0 = (MethodHandles.Lookup) getObjectMethod.invoke(usafe, lookupClass, implOffset);
long fieldOffset = (Long) staticFieldOffsetMethod.invoke(usafe, implLookup); RedkaleClassLoader.putReflectionField(lookupClass.getName(), implField);
defaultLookup0 = (MethodHandles.Lookup) getObjectMethod.invoke(usafe, lookupClass, fieldOffset);
RedkaleClassLoader.putReflectionField(lookupClass.getName(), implLookup); Field compactField = String.class.getDeclaredField("COMPACT_STRINGS");
long compactOffset = (Long) staticFieldOffsetMethod.invoke(usafe, compactField);
Method getBooleanMethod = usafe.getClass().getMethod("getBoolean", Object.class, long.class);
compactStrings0 = (Boolean) getBooleanMethod.invoke(usafe, String.class, compactOffset);
RedkaleClassLoader.putReflectionField(String.class.getName(), compactField);
} catch (Throwable e) { } catch (Throwable e) {
defaultLookup0 = MethodHandles.lookup(); e.printStackTrace();
} }
defaultLookup = defaultLookup0; compactStrings = compactStrings0;
trustedLookup = trustedLookup0;
} }
private static final Executor defaultExecutorConsumer = Runnable::run; private static final Executor defaultExecutorConsumer = Runnable::run;
@@ -171,18 +181,23 @@ public final class Utility {
} catch (Throwable t) { } catch (Throwable t) {
// do nothing // do nothing
} }
if (trustedLookup != null) {
try { try {
// String-LATIN1 // String-LATIN1
MethodHandles.Lookup lookup = defaultLookup; MethodHandles.Lookup lookup = trustedLookup;
VarHandle compactHandle = lookup.findStaticVarHandle(String.class, "COMPACT_STRINGS", boolean.class);
final boolean compact = (Boolean) compactHandle.get(null);
VarHandle coderHandle = lookup.findVarHandle(String.class, "coder", byte.class); VarHandle coderHandle = lookup.findVarHandle(String.class, "coder", byte.class);
VarHandle valueHandle = lookup.findVarHandle(String.class, "value", byte[].class); VarHandle valueHandle = lookup.findVarHandle(String.class, "value", byte[].class);
// LATIN1:0 UTF16:1 // LATIN1:0 UTF16:1
strLatin1Function0 = compact ? (String t) -> (Byte) coderHandle.get(t) == 0 : (String t) -> false; strLatin1Function0 =
compactStrings ? (String t) -> (Byte) coderHandle.get(t) == 0 : (String t) -> false;
strByteFunction0 = (String t) -> (byte[]) valueHandle.get(t); strByteFunction0 = (String t) -> (byte[]) valueHandle.get(t);
} catch (Throwable e) { // 不会发生
// do nothing
e.printStackTrace();
}
}
// signalShutdown // signalShutdown
try {
Class<Consumer<Consumer<String>>> shutdownClazz1 = null; Class<Consumer<Consumer<String>>> shutdownClazz1 = null;
try { try {
shutdownClazz1 = (Class) loader.loadClass("org.redkale.util.SignalShutDown"); shutdownClazz1 = (Class) loader.loadClass("org.redkale.util.SignalShutDown");
@@ -201,9 +216,8 @@ public final class Utility {
RedkaleClassLoader.putReflectionDeclaredConstructors(shutdownClazz1, shutdownClazz1.getName()); RedkaleClassLoader.putReflectionDeclaredConstructors(shutdownClazz1, shutdownClazz1.getName());
signalShutdownConsumer0 = shutdownClazz1.getConstructor().newInstance(); signalShutdownConsumer0 = shutdownClazz1.getConstructor().newInstance();
} }
} catch (Throwable e) { // 不会发生 } catch (Throwable t) {
// do nothing // do nothing
e.printStackTrace();
} }
strByteFunction = strByteFunction0; strByteFunction = strByteFunction0;
strLatin1Function = strLatin1Function0; strLatin1Function = strLatin1Function0;
@@ -243,8 +257,8 @@ public final class Utility {
return CPUS; return CPUS;
} }
public static MethodHandles.Lookup lookup() { public static @Nullable MethodHandles.Lookup trustedLookup() {
return defaultLookup; return trustedLookup;
} }
public static boolean inNativeImage() { public static boolean inNativeImage() {