lanxi/window-axis-innovators-box
2
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(core): 重构类加载器并添加插件支持

Browse Source 
-重构 BoxClassLoader 以支持插件加载
- 添加核心插件加载逻辑
- 实现类转换器和黑名单功能
- 优化工具类别和工具项的注册
- 修复日志输出和异常处理
This commit is contained in:
tzdwindows 7
2025-02-11 13:21:45 +08:00
parent fcc4115638
commit 4d08fbeab0
9 changed files with 287 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
build.gradle
View File

@@ -80,6 +80,12 @@ sourceSets {
}
}
tasks.withType(JavaExec).configureEach {
jvmArgs = [
'-Djava.system.class.loader=com.axis.innovators.box.plugins.BoxClassLoader'
]
}
// 单独打包文档
task packageOpenSourceDocs(type: Jar) {
archiveClassifier = 'docs'

11
src/main/java/com/axis/innovators/box/Log4j2OutputStream.java
View File

@@ -11,7 +11,7 @@ import java.io.PrintStream;
* @author tzdwindows 7
*/
public class Log4j2OutputStream extends OutputStream {
private static final Logger logger = LogManager.getLogger(Log4j2OutputStream.class);
private static final Logger logger = LogManager.getLogger();
@Override
public void write(int b) {
@@ -20,14 +20,15 @@ public class Log4j2OutputStream extends OutputStream {
@Override
public void write(byte[] b, int off, int len) {
logger.info(new String(b, off, len));
String message = new String(b, off, len).trim();
logger.info(message);
}
/**
* 重定向 System.out 和 System.err 到 Log4j2
*/
public static void redirectSystemStreams() {
System.setOut(new PrintStream(new Log4j2OutputStream()));
System.setErr(new PrintStream(new Log4j2OutputStream()));
System.setOut(new PrintStream(new Log4j2OutputStream(), true));
System.setErr(new PrintStream(new Log4j2OutputStream(), true));
}
}
}

21
src/main/java/com/axis/innovators/box/Main.java
View File

@@ -36,7 +36,7 @@ public class Main {
private Thread thread;
private final String[] args;
private boolean isWindow = false;
private RegistrationTool registrationTool;
private final RegistrationTool registrationTool = new RegistrationTool(this);
public Main(String[] args){
this.args = args;
@@ -118,9 +118,10 @@ public class Main {
main.thread = new Thread(() -> {
try {
// 主任务1：加载插件
logger.info("Loaded plugins Started");
main.progressBarManager.updateMainProgress(++main.completedTasks);
PluginLoader.loadPlugins();
logger.info("Loaded plugins");
logger.info("Loaded plugins End");
main.progressBarManager.close();
@@ -182,13 +183,15 @@ public class Main {
// 主任务2：加载工具栏
progressBarManager.updateMainProgress(++completedTasks);
for (int i = 0; i < registrationTool.getToolCategories()
.size(); i++) {
ex.addToolCategory(registrationTool.getToolCategories()
.get(i));
progressBarManager.updateSubProgress(
"add tools",i,
registrationTool.getToolCategories().size());
if (!registrationTool.getToolCategories().isEmpty()) {
for (int i = 0; i < registrationTool.getToolCategories()
.size(); i++) {
ex.addToolCategory(registrationTool.getToolCategories()
.get(i));
progressBarManager.updateSubProgress(
"add tools", i,
registrationTool.getToolCategories().size());
}
}
ex.initUI();

95
src/main/java/com/axis/innovators/box/gui/MainWindow.java
View File

@@ -210,6 +210,13 @@ public class MainWindow extends JFrame {
verticalScrollBar.setUI(new CustomScrollBarUI());
verticalScrollBar.setPreferredSize(new Dimension(10, 100));
if (category.getIcon() == null){
tabbedPane.addTab(
category.getName(),
category.getIconImage(),
scrollPane
);
}
tabbedPane.addTab(
category.getName(),
LoadIcon.loadIcon(category.getIcon(), 24),
@@ -312,8 +319,13 @@ public class MainWindow extends JFrame {
BorderFactory.createEmptyBorder(20, 20, 20, 20)
));
card.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));
JLabel iconLabel;
if (tool.getIcon() == null){
iconLabel = new JLabel(tool.getImageIcon());
}else {
iconLabel = new JLabel(LoadIcon.loadIcon(tool.getIcon(), 64));
}
JLabel iconLabel = new JLabel(LoadIcon.loadIcon(tool.icon(), 64));
iconLabel.setHorizontalAlignment(SwingConstants.CENTER);
// 文字面板
@@ -321,12 +333,12 @@ public class MainWindow extends JFrame {
textPanel.setLayout(new BoxLayout(textPanel, BoxLayout.Y_AXIS));
textPanel.setOpaque(false);
JLabel titleLabel = new JLabel(tool.title());
JLabel titleLabel = new JLabel(tool.getTitle());
titleLabel.setFont(new Font("微软雅黑", Font.BOLD, 18));
titleLabel.setForeground(new Color(44, 62, 80));
titleLabel.setAlignmentX(Component.CENTER_ALIGNMENT);
JTextArea descArea = new JTextArea(tool.description());
JTextArea descArea = new JTextArea(tool.getDescription());
descArea.setFont(new Font("微软雅黑", Font.PLAIN, 14));
descArea.setForeground(new Color(127, 140, 153));
descArea.setLineWrap(true);
@@ -374,7 +386,7 @@ public class MainWindow extends JFrame {
private String createToolTipHTML(ToolItem tool) {
return "<html><body style='width: 300px; padding: 10px;'>" +
"<h3 style='color: #2c3e50; margin: 0 0 8px 0;'>" + tool.getName() + "</h3>" +
"<p style='color: #7f8c99; margin: 0;'>" + tool.description() + "</p>" +
"<p style='color: #7f8c99; margin: 0;'>" + tool.getDescription() + "</p>" +
"</body></html>";
}
@@ -382,6 +394,7 @@ public class MainWindow extends JFrame {
public static class ToolCategory {
private final String name;
private final String icon;
private final ImageIcon iconImage;
private final String description;
private final List<ToolItem> tools = new ArrayList<>();
@@ -395,6 +408,14 @@ public class MainWindow extends JFrame {
this.name = name;
this.icon = icon;
this.description = description;
this.iconImage = null;
}
public ToolCategory(String name, ImageIcon icon, String description) {
this.name = name;
this.iconImage = icon;
this.description = description;
this.icon = null;
}
/**
@@ -420,19 +441,71 @@ public class MainWindow extends JFrame {
public List<ToolItem> getTools() {
return tools;
}
public ImageIcon getIconImage() {
return iconImage;
}
}
// 工具项数据类
/**
* 工具注册类
* @param title 工具的标题（显示名称）
* @param icon 工具的图标（resources的路径）
* @param description 工具的描述
* @param id 工具的id（请不要重复注册相同id的工具）
* @param action 工具的点击事件
*/
public record ToolItem(String title, String icon, String description, int id, Action action) {
public static class ToolItem {
private final ImageIcon imageIcon;
private final String title;
private final String icon;
private final String description;
private final int id;
private final Action action;
public ToolItem(String title, String icon, String description, int id, Action action){
this.title = title;
this.icon = icon;
this.description = description;
this.id = id;
this.action = action;
this.imageIcon = null;
}
public ToolItem(String title, ImageIcon icon, String description, int id, Action action) {
this.title = title;
this.imageIcon = icon;
this.description = description;
this.id = id;
this.action = action;
this.icon = null;
}
public String getTitle() {
return title;
}
public ImageIcon icon() {
return imageIcon;
}
public String getIcon() {
return icon;
}
public Action getAction() {
return action;
}
public ImageIcon getImageIcon() {
return imageIcon;
}
public int getId() {
return id;
}
public String getDescription() {
return description;
}
public String getName() {
return title;
@@ -495,7 +568,7 @@ public class MainWindow extends JFrame {
@Override
public void mouseReleased(MouseEvent e) {
startReleaseAnimation(() -> tool.action().actionPerformed(
startReleaseAnimation(() -> tool.getAction().actionPerformed(
new ActionEvent(card, ActionEvent.ACTION_PERFORMED, "")
));
}

116
src/main/java/com/axis/innovators/box/plugins/BoxClassLoader.java
View File

@@ -1,21 +1,30 @@
package com.axis.innovators.box.plugins;
import org.objectweb.asm.ClassReader;
import org.objectweb.asm.ClassWriter;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;
/**
* 自定义加载器
* @author tzdwindows 7
*/
public class BoxClassLoader extends URLClassLoader {
public BoxClassLoader(ClassLoader parent) {
super(new URL[0], parent);
private static final List<IClassTransformer> CLASS_TRANSFORMS = new ArrayList<>();
private static final List<String> CLASS_BLACKLIST = new ArrayList<>();
private static final List<String> CLASS_LOADING_LIST = new ArrayList<>();
private static final List<Class<?>> CLASS_LOADING_LIST_OBJECT = new ArrayList<>();
static {
// 添加黑名单，避免加载核心类
CLASS_BLACKLIST.add("java.");
CLASS_BLACKLIST.add("javax.");
CLASS_BLACKLIST.add("sun.");
}
public BoxClassLoader(URL[] sources) {
super(sources);
public BoxClassLoader(ClassLoader parent) {
super(new URL[]{}, parent);
}
public BoxClassLoader(URL[] urls, ClassLoader parent) {
@@ -23,18 +32,75 @@ public class BoxClassLoader extends URLClassLoader {
}
@Override
protected Class<?> findClass(String name) throws ClassNotFoundException {
byte[] classBytes;
try {
classBytes = getClassBytes(name);
} catch (IOException e) {
throw new RuntimeException(e);
}
byte[] transformedClass = transformClass(name, null, classBytes);
return defineClass(name, transformedClass, 0, transformedClass.length);
public void addURL(URL url) {
super.addURL(url);
}
public byte[] getClassBytes(String className) throws IOException, ClassNotFoundException {
//@Override
//protected Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException {
// synchronized (getClassLoadingLock(name)) {
// // 1. 检查类是否已加载
// Class<?> c = findLoadedClass(name);
// if (c != null) {
// return c;
// }
//
// // 2. 检查黑名单
// if (isBlacklisted(name)) {
// return super.loadClass(name, resolve);
// }
//
// // 3. 尝试自定义加载
// try {
// c = findClass(name);
// } catch (ClassNotFoundException e) {
// // 4. 自定义加载失败则委托给父加载器
// c = super.loadClass(name, resolve);
// }
//
// if (resolve) {
// resolveClass(c);
// }
// return c;
// }
//}
@Override
protected Class<?> findClass(String name) throws ClassNotFoundException {
// 检查是否在黑名单中
if (isBlacklisted(name)) {
throw new ClassNotFoundException("Class is blacklisted: " + name);
}
// 检查是否正在加载
if (CLASS_LOADING_LIST.contains(name)) {
throw new ClassNotFoundException("Class is already loading: " + name);
}
// 读取类字节码
byte[] clazzByte;
try {
clazzByte = getClassBytes(name);
} catch (IOException e) {
throw new ClassNotFoundException("Failed to load class bytes: " + name, e);
}
// 应用类转换器
for (IClassTransformer transformer : CLASS_TRANSFORMS) {
byte[] transformed = transformer.transform(name, transformer.getClass().getName(), clazzByte);
if (transformed != null) {
clazzByte = transformed;
}
}
// 定义类
CLASS_LOADING_LIST.add(name);
Class<?> clazz = defineClass(name, clazzByte, 0, clazzByte.length);
CLASS_LOADING_LIST_OBJECT.add(clazz);
return clazz;
}
private byte[] getClassBytes(String className) throws IOException, ClassNotFoundException {
String path = className.replace('.', '/') + ".class";
URL classUrl = getResource(path);
if (classUrl == null) {
@@ -45,7 +111,15 @@ public class BoxClassLoader extends URLClassLoader {
}
}
public byte[] transformClass(String className, String transformedName, byte[] basicClass) {
return PluginLoader.transformClass(className, transformedName, basicClass);
private boolean isBlacklisted(String className) {
return CLASS_BLACKLIST.stream().anyMatch(className::startsWith);
}
}
public static void addClassTransformer(IClassTransformer transformer) {
CLASS_TRANSFORMS.add(transformer);
}
public static List<Class<?>> getClassList() {
return CLASS_LOADING_LIST_OBJECT;
}
}

84
src/main/java/com/axis/innovators/box/plugins/PluginLoader.java
View File

@@ -6,6 +6,8 @@ import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import java.io.*;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.net.*;
import java.util.*;
import java.util.jar.*;
@@ -15,20 +17,26 @@ import java.util.jar.*;
* @author tzdwindows 7
*/
public class PluginLoader {
private static final Logger logger = LogManager.getLogger(PluginLoader.class);
private static Logger logger;
public static final String PLUGIN_PATH = FolderCreator.getPluginFolder();
private static final List<PluginDescriptor> loadedPlugins = new ArrayList<>();
private static final List<IClassTransformer> transformers = new ArrayList<>();
public static void loadPlugins() throws IOException {
File pluginDir = new File(PLUGIN_PATH);
File[] jars = pluginDir.listFiles((dir, name) -> name.toLowerCase().endsWith(".jar"));
logger = LogManager.getLogger(PluginLoader.class);
File pluginDir = null;
if (PLUGIN_PATH != null) {
pluginDir = new File(PLUGIN_PATH);
}
File[] jars = null;
if (pluginDir != null) {
jars = pluginDir.listFiles((dir, name) -> name.toLowerCase().endsWith(".jar"));
}
if (jars == null) {
return;
}
for (int i = 0; i < jars.length; i++) {
processJarFile(jars[i]);
processJarFile(jars[i],false);
Main.getMain().progressBarManager.updateSubProgress(
"Loading Plugin " + i,
i,
@@ -36,24 +44,47 @@ public class PluginLoader {
}
}
private static void processJarFile(File jarFile) throws IOException {
private static void processJarFile(File jarFile, boolean isCorePlugin) throws IOException {
try (JarFile jar = new JarFile(jarFile)) {
JarEntry pluginFile = jar.getJarEntry("plug-in.box");
if (pluginFile != null) {
processWithManifest(jar, pluginFile, jarFile);
} else {
processWithAnnotations(jar, jarFile);
}
// Check for CorePlugin in MANIFEST.MF
Attributes attributes = jar.getManifest().getMainAttributes();
String corePluginClass = attributes.getValue("CorePlugin");
if (corePluginClass != null) {
processCorePlugin(jarFile, corePluginClass);
if (isCorePlugin) {
Attributes attributes = jar.getManifest().getMainAttributes();
String corePluginClass = attributes.getValue("CorePlugin");
if (corePluginClass != null) {
processCorePlugin(jarFile, corePluginClass);
}
} else {
JarEntry pluginFile = jar.getJarEntry("plug-in.box");
if (pluginFile != null) {
processWithManifest(jar, pluginFile, jarFile);
} else {
processWithAnnotations(jar, jarFile);
}
}
}
}
/**
* 加载核心插件
* @throws IOException 插件加载失败
*/
public static void loadCorePlugin() throws IOException {
File pluginDir = null;
if (PLUGIN_PATH != null) {
pluginDir = new File(PLUGIN_PATH);
}
File[] jars = null;
if (pluginDir != null) {
jars = pluginDir.listFiles((dir, name) -> name.toLowerCase().endsWith(".jar"));
}
if (jars == null) {
return;
}
for (File jar : jars) {
processJarFile(jar, true);
}
}
private static void processCorePlugin(File jarFile, String corePluginClass) {
try (URLClassLoader classLoader = new URLClassLoader(
new URL[]{jarFile.toURI().toURL()},
@@ -62,22 +93,23 @@ public class PluginLoader {
Class<?> coreClass = classLoader.loadClass(corePluginClass);
if (LoadingCorePlugin.class.isAssignableFrom(coreClass)) {
LoadingCorePlugin corePlugin = (LoadingCorePlugin) coreClass.getDeclaredConstructor().newInstance();
registerTransformers(corePlugin);
registerTransformers(corePlugin, classLoader);
}
} catch (Exception e) {
logger.error("Failed to load core plugin: {}", corePluginClass, e);
}
}
private static void registerTransformers(LoadingCorePlugin corePlugin) {
private static void registerTransformers(LoadingCorePlugin corePlugin, ClassLoader classLoader) {
String[] transformerClasses = corePlugin.getASMTransformerClass();
if (transformerClasses != null) {
for (String transformerClass : transformerClasses) {
try {
Class<?> transformerClazz = Class.forName(transformerClass);
Class<?> transformerClazz = classLoader.loadClass(transformerClass);
if (IClassTransformer.class.isAssignableFrom(transformerClazz)) {
IClassTransformer transformer = (IClassTransformer) transformerClazz.getDeclaredConstructor().newInstance();
transformers.add(transformer);
BoxClassLoader.addClassTransformer(transformer);
}
} catch (Exception e) {
logger.error("Failed to register transformer: {}", transformerClass, e);
@@ -113,11 +145,12 @@ public class PluginLoader {
private static void processWithAnnotations(JarFile jar, File jarFile) {
URLClassLoader classLoader = createClassLoader(jarFile);
Enumeration<JarEntry> entries = jar.entries();
while (entries.hasMoreElements()) {
JarEntry entry = entries.nextElement();
if (entry.getName().endsWith(".class")) {
processClassEntry(entry, classLoader);
if (classLoader != null) {
processClassEntry(entry, classLoader);
}
}
}
}
@@ -183,11 +216,4 @@ public class PluginLoader {
public static List<PluginDescriptor> getLoadedPlugins() {
return Collections.unmodifiableList(loadedPlugins);
}
public static byte[] transformClass(String name, String transformedName, byte[] basicClass) {
for (IClassTransformer transformer : transformers) {
basicClass = transformer.transform(name, transformedName, basicClass);
}
return basicClass;
}
}

2
src/main/java/com/axis/innovators/box/register/RegistrationTool.java
View File

@@ -25,7 +25,7 @@ public class RegistrationTool {
* 注册ToolCategory
* @param toolCategory ToolCategory
*/
void addToolCategory(MainWindow.ToolCategory toolCategory){
public void addToolCategory(MainWindow.ToolCategory toolCategory){
if (!main.isWindow()) {
toolCategories.add(toolCategory);
} else {

11
src/main/java/com/axis/innovators/box/tools/FolderCreator.java
View File

@@ -10,7 +10,7 @@ import java.io.File;
* @author tzdwindows 7
*/
public class FolderCreator {
private static final Logger logger = LogManager.getLogger(FolderCreator.class);
private static Logger logger;
public static final String LIBRARY_NAME = "library";
public static final String MODEL_PATH = "model";
public static final String PLUGIN_PATH = "plug-in";
@@ -27,6 +27,9 @@ public class FolderCreator {
public static String getModelFolder() {
String folder = createFolder(MODEL_PATH);
if (folder == null) {
if (logger == null) {
logger = LogManager.getLogger();
}
logger.error("Model folder creation failure, please use administrator privileges to execute this procedure");
return null;
}
@@ -36,6 +39,9 @@ public class FolderCreator {
public static String getLibraryFolder() {
String folder = createFolder(LIBRARY_NAME);
if (folder == null) {
if (logger == null) {
logger = LogManager.getLogger();
}
logger.error("Library folder creation failed, please use administrator privileges to execute this procedure");
return null;
}
@@ -52,6 +58,9 @@ public class FolderCreator {
File folder = new File(jarDir, folderName);
if (!folder.exists()) {
if (!folder.mkdir()) {
if (logger == null) {
logger = LogManager.getLogger();
}
logger.error("Folder creation failure");
return null;
}

28
src/main/java/org/tzd/lm/LM.java
View File

@@ -10,17 +10,28 @@ import org.apache.logging.log4j.Logger;
* @author tzdwindows 7
*/
public class LM {
public static boolean CUDA = false;
public static boolean CUDA = true;
public final static String DEEP_SEEK = FolderCreator.getModelFolder() + "\\DeepSeek-R1-Distill-Qwen-1.5B-Q8_0.gguf";
private static final Logger logger = LogManager.getLogger(LM.class);
static {
if (!CUDA) {
loadLibrary(CUDA);
}
private static void loadLibrary(boolean cuda){
if (!cuda) {
logger.warn("The cpu will be used for inference");
LibraryLoad.loadLibrary("cpu/ggml-base");
LibraryLoad.loadLibrary("cpu/ggml-cpu");
LibraryLoad.loadLibrary("cpu/ggml");
LibraryLoad.loadLibrary("cpu/llama");
try {
LibraryLoad.loadLibrary("cpu/ggml-base");
LibraryLoad.loadLibrary("cpu/ggml-cpu");
LibraryLoad.loadLibrary("cpu/ggml");
LibraryLoad.loadLibrary("cpu/llama");
} catch (UnsatisfiedLinkError e) {
logger.error("Unable to load library: " + e.getMessage(), e);
logger.error("Missing dependency: " + e.getMessage());
} catch (Exception e) {
logger.error("Unable to load cpu, please try updating driver", e);
}
} else {
try {
LibraryLoad.loadLibrary("cuda/ggml-base");
@@ -33,10 +44,7 @@ public class LM {
} catch (Exception e) {
CUDA = false;
logger.warn("The cuda library could not be loaded, the cpu will be used for inference");
LibraryLoad.loadLibrary("cpu/ggml-base");
LibraryLoad.loadLibrary("cpu/ggml-cpu");
LibraryLoad.loadLibrary("cpu/ggml");
LibraryLoad.loadLibrary("cpu/llama");
loadLibrary(false);
}
}
LibraryLoad.loadLibrary("LM");