ICode9

精准搜索请尝试: 精确搜索
首页 > 编程语言> 文章详细

手写一个动态编译Java源码并加载Class

2022-03-21 10:01:57  阅读:150  来源: 互联网

标签:null Java name public 源码 new classes Class append


import javax.tools.SimpleJavaFileObject;
import java.net.URI;

/**
 * 将代码保存在字符串构建器中的 Java 源代码。
 *
 * @description:
 * @author: mabh
 * @create: 2022/3/19 22:33
 **/
public class StringBuilderJavaSource extends SimpleJavaFileObject {
    private StringBuilder code;

    public StringBuilderJavaSource(String name) {
        super(URI.create("string:///" + name.replace(".", "/") + Kind.SOURCE.extension), Kind.SOURCE);
        code = new StringBuilder();
    }

    @Override
    public CharSequence getCharContent(boolean ignoreEncodingErrors) {
        return code;
    }


    public void append(String str) {
        code.append(str);
        code.append("\n");
    }



}

import java.util.Map;

public class MapClassLoader extends ClassLoader {

    private final Map<String, byte[]> classes;

    public MapClassLoader(Map<String, byte[]> classes) {
        this.classes = classes;
    }

    @Override
    protected Class<?> findClass(String name) throws ClassNotFoundException {
        byte[] classBytes = classes.get(name);
        if (classBytes == null) {
            throw new ClassNotFoundException(name);
        }
        Class<?> cl = defineClass(name, classBytes, 0, classBytes.length);
        if (cl == null) {
            throw new ClassNotFoundException(name);
        }
        return cl;
    }
}
import java.util.Map;

public class MapClassLoader extends ClassLoader {

    private final Map<String, byte[]> classes;

    public MapClassLoader(Map<String, byte[]> classes) {
        this.classes = classes;
    }


    @Override
    protected Class<?> findClass(String name) throws ClassNotFoundException {
        byte[] classBytes = classes.get(name);
        if (classBytes == null) {
            throw new ClassNotFoundException(name);
        }
        Class<?> cl = defineClass(name, classBytes, 0, classBytes.length);
        if (cl == null) {
            throw new ClassNotFoundException(name);
        }
        return cl;
    }
}

需要一个测试类,我们新建一个接口

/**
 * @description:
 * @author: mabh
 * @create: 2022/3/19 23:15
 **/
public interface IABC {
    void main2();
}

重头戏

import javax.tools.*;
import java.io.IOException;
import java.util.*;

/**
 * @description:
 * @author: mabh
 * @create: 2022/3/19 21:49
 **/
public class Test222{
    public static void main(String[] args) throws Exception {
        JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();

        List<ByteArrayJavaClass> classFileObjects = new ArrayList<>();

        DiagnosticCollector<JavaFileObject> diagnostic = new DiagnosticCollector<>();
        JavaFileManager fileManager = compiler.getStandardFileManager(diagnostic, null, null);
        fileManager = new ForwardingJavaFileManager<JavaFileManager>(fileManager) {
            @Override
            public JavaFileObject getJavaFileForOutput(Location location, String className, JavaFileObject.Kind kind, FileObject sibling) throws IOException {
                ByteArrayJavaClass arrayJavaClass = new ByteArrayJavaClass(className);
                classFileObjects.add(arrayJavaClass);
                return arrayJavaClass;
            }
        };

        StringBuilderJavaSource javaSource = new StringBuilderJavaSource("ABC");
        javaSource.append("package com.epoch.planning;");
        javaSource.append(" public class ABC implements IABC {");
        javaSource.append("    public void main2() {");
        javaSource.append(" System.out.print(\"我是使用java动态编译的类\");");
        javaSource.append("    }");
        javaSource.append(" }");

        JavaCompiler.CompilationTask compilerTask = compiler.getTask(null, fileManager, diagnostic, null, null, Arrays.asList(javaSource));
        Boolean result = compilerTask.call();

        for (Diagnostic<? extends JavaFileObject> diagnosticDiagnostic : diagnostic.getDiagnostics()) {
            System.out.println(diagnosticDiagnostic.getKind() + ": " + diagnosticDiagnostic.getMessage(null));
        }
        fileManager.close();

        if (!result) {
            List<Diagnostic<? extends JavaFileObject>> diagnostics = diagnostic.getDiagnostics();
            for (Diagnostic<? extends JavaFileObject> diagnostic1 : diagnostics) {
                System.out.println(diagnostic1.toString());
                break;
            }
            System.out.println("compile failed!");
        }

        Map<String, byte[]> byteCodeMap = new HashMap<>();
        for (ByteArrayJavaClass classFileObject : classFileObjects) {
            byteCodeMap.put(classFileObject.getName().substring(1), classFileObject.getBytes());
        }
        MapClassLoader loader = new MapClassLoader(byteCodeMap);
        IABC abc = (IABC) loader.findClass("com.epoch.planning.ABC").newInstance();
        abc.main2();


    }
}

失败了会自动打印日志

在这里插入图片描述

修改正确即可正确输出

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-XRWqfs2i-1647827615131)(/Users/mbh/Library/Application Support/typora-user-images/image-20220321095304671.png)]

标签:null,Java,name,public,源码,new,classes,Class,append
来源: https://blog.csdn.net/mbh12333/article/details/123628062

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有