fix: custom shell generate failed

2026-09-22 23:11:52 +08:00 · 2025-09-19 22:09:47 +08:00
parent f5112bd86d
commit ba545a2800
1 changed files with 6 additions and 3 deletions
@@ -31,13 +31,16 @@ public class CustomShellGenerator extends ByteBuddyShellGenerator<CustomConfig>
        if (StringUtils.isBlank(shellToolConfig.getShellClassName())) {
            shellToolConfig.setShellClassName(className);
        }
-        ClassFileLocator classFileLocator = ClassFileLocator.Simple.of(className, classBytes);
+        ClassFileLocator compoundLocator = new ClassFileLocator.Compound(
+                ClassFileLocator.Simple.of(className, classBytes),
+                ClassFileLocator.ForClassLoader.of(this.getClass().getClassLoader())
+        );
        TypeDescription typeDescription = new TypePool.Default(
-                new TypePool.CacheProvider.Simple(), classFileLocator,
+                new TypePool.CacheProvider.Simple(), compoundLocator,
                TypePool.Default.ReaderMode.FAST, TypePool.Default.ofSystemLoader()
        ).describe(className).resolve();
        shellToolConfig.setShellTypeDescription(typeDescription);
        return new ByteBuddy()
-                .redefine(typeDescription, classFileLocator);
+                .redefine(typeDescription, compoundLocator);
    }
 }