springboot 2 下 java 代碼 動態編譯 動態加載 實現

先放出代碼，再說有幾個坑的地方：

package com.yrxd.telemarketing.service;


import org.apache.log4j.Logger;
import org.springframework.boot.loader.LaunchedURLClassLoader;

import javax.lang.model.element.Modifier;
import javax.lang.model.element.NestingKind;
import javax.tools.*;
import java.io.*;
import java.net.*;
import java.nio.CharBuffer;
import java.util.*;
import java.util.jar.JarEntry;

/**
 * Created by xidongzhou1 on 2020/1/17.
 */
public class CusCompiler {
  static Logger logger = Logger.getLogger(CusCompiler.class);
  static class CustomJavaFileObject implements JavaFileObject {
    private String binaryName;
    private URI uri;
    private String name;

    public String binaryName() {
      return binaryName;
    }

    public CustomJavaFileObject(String binaryName, URI uri) {
      this.uri = uri;
      this.binaryName = binaryName;
      name = uri.getPath() == null ? uri.getSchemeSpecificPart() : uri.getPath();
    }

    @Override
    public Kind getKind() {
      return Kind.CLASS;
    }

    @Override
    public boolean isNameCompatible(String simpleName, Kind kind) {
      String baseName = simpleName + kind.extension;
      return kind.equals(getKind()) && (baseName.equals(getName()) || getName().endsWith("/" + baseName));
    }

    @Override
    public NestingKind getNestingKind() {
      throw new UnsupportedOperationException();
    }

    @Override
    public Modifier getAccessLevel() {
      throw new UnsupportedOperationException();
    }

    @Override
    public URI toUri() {
      return uri;
    }

    @Override
    public String getName() {
      return name;
    }

    @Override
    public InputStream openInputStream() throws IOException {
      return uri.toURL().openStream();
    }

    @Override
    public OutputStream openOutputStream() throws IOException {
      throw new UnsupportedOperationException();
    }

    @Override
    public Reader openReader(boolean ignoreEncodingErrors) throws IOException {
      throw new UnsupportedOperationException();
    }

    @Override
    public CharSequence getCharContent(boolean ignoreEncodingErrors) throws IOException {
      throw new UnsupportedOperationException();
    }

    @Override
    public Writer openWriter() throws IOException {
      throw new UnsupportedOperationException();
    }

    @Override
    public long getLastModified() {
      return 0;
    }

    @Override
    public boolean delete() {
      throw new UnsupportedOperationException();
    }
  }

  static class MemoryInputJavaFileObject extends SimpleJavaFileObject {
    final String code;

    MemoryInputJavaFileObject(String name, String code) {
      super(URI.create(name.replaceAll("\\.", "/") + Kind.SOURCE.extension), Kind.SOURCE);
      this.code = code;
    }

    @Override
    public CharBuffer getCharContent(boolean ignoreEncodingErrors) {
      return CharBuffer.wrap(code);
    }
  }

  static class MemoryOutputJavaFileObject extends SimpleJavaFileObject {
    final String name;
    Map<String, byte[]> class_out;

    MemoryOutputJavaFileObject(String name, Map<String, byte[]> out) {
      super(URI.create(name.replaceAll("\\.", "/") + Kind.SOURCE.extension), Kind.CLASS);
      this.name = name;
      this.class_out = out;
    }

    @Override
    public OutputStream openOutputStream() {
      return new FilterOutputStream(new ByteArrayOutputStream()) {
        @Override
        public void close() throws IOException {
          out.close();
          ByteArrayOutputStream bos = (ByteArrayOutputStream) out;
          class_out.put(name, bos.toByteArray());
        }
      };
    }
  }

  static class SpringBootJarFileManager implements JavaFileManager {
    private URLClassLoader classLoader;
    private StandardJavaFileManager standardJavaFileManager;
    final Map<String, byte[]> classBytes = new HashMap<>();

    SpringBootJarFileManager(StandardJavaFileManager standardJavaFileManager, URLClassLoader systemLoader) {
      this.classLoader = new URLClassLoader(systemLoader.getURLs(), systemLoader);
      this.standardJavaFileManager = standardJavaFileManager;
    }

    @Override
    public ClassLoader getClassLoader(Location location) {
      return classLoader;
    }

    private List<JavaFileObject> find(String packageName) {
      List<JavaFileObject> result = new ArrayList<>();
      String javaPackageName = packageName.replaceAll("\\.", "/");
      try {
        Enumeration<URL> urls = classLoader.findResources(javaPackageName);
        while (urls.hasMoreElements()) {
          URL ll = urls.nextElement();
          String ext_form = ll.toExternalForm();
          String jar = ext_form.substring(0, ext_form.lastIndexOf("!"));
          String pkg = ext_form.substring(ext_form.lastIndexOf("!") + 1);

          JarURLConnection conn = (JarURLConnection) ll.openConnection();
          conn.connect();
          Enumeration<JarEntry> jar_items = conn.getJarFile().entries();
          while (jar_items.hasMoreElements()) {
            JarEntry item = jar_items.nextElement();
            if (item.isDirectory() || (!item.getName().endsWith(".class"))) {
              continue;
            }
            if (item.getName().lastIndexOf("/") != (pkg.length() - 1)) {
              continue;
            }
            String name = item.getName();
            URI uri = URI.create(jar + "!/" + name);
            String binaryName = name.replaceAll("/", ".");
            binaryName = binaryName.substring(0, binaryName.indexOf(JavaFileObject.Kind.CLASS.extension));
            result.add(new CustomJavaFileObject(binaryName, uri));
          }
        }
      } catch (Exception e) {
        e.printStackTrace();
      }
      return result;
    }

    @Override
    public Iterable<JavaFileObject> list(Location location, String packageName, Set<JavaFileObject.Kind> kinds, boolean recurse) throws IOException {
      Iterable<JavaFileObject> ret = null;
      if (location == StandardLocation.PLATFORM_CLASS_PATH) {
        ret = standardJavaFileManager.list(location, packageName, kinds, recurse);
      } else if (location == StandardLocation.CLASS_PATH && kinds.contains(JavaFileObject.Kind.CLASS)) {
        ret = find(packageName);
        if (ret == null || (!ret.iterator().hasNext())) {
          ret = standardJavaFileManager.list(location, packageName, kinds, recurse);
        }
      } else {
        ret = Collections.emptyList();
      }
      return ret;
    }

    @Override
    public String inferBinaryName(Location location, JavaFileObject file) {
      String ret = "";
      if (file instanceof CustomJavaFileObject) {
        ret = ((CustomJavaFileObject)file).binaryName;
      } else {
        ret = standardJavaFileManager.inferBinaryName(location, file);
      }
      return ret;
    }

    @Override
    public boolean isSameFile(FileObject a, FileObject b) {
      throw new UnsupportedOperationException();
    }

    @Override
    public boolean handleOption(String current, Iterator<String> remaining) {
      return standardJavaFileManager.handleOption(current, remaining);
    }

    @Override
    public boolean hasLocation(Location location) {
      return location == StandardLocation.CLASS_PATH || location == StandardLocation.PLATFORM_CLASS_PATH;
    }

    @Override
    public JavaFileObject getJavaFileForInput(Location location, String className, JavaFileObject.Kind kind) throws IOException {
      throw new UnsupportedOperationException();
    }

    @Override
    public FileObject getFileForInput(Location location, String packageName, String relativeName) throws IOException {
      throw new UnsupportedOperationException();
    }

    @Override
    public FileObject getFileForOutput(Location location, String packageName, String relativeName, FileObject sibling) throws IOException {
      throw new UnsupportedOperationException();
    }

    @Override
    public void flush() throws IOException {
    }

    @Override
    public void close() throws IOException {
      classBytes.clear();
    }

    @Override
    public int isSupportedOption(String option) {
      return -1;
    }

    public Map<String, byte[]> getClassBytes() {
      return new HashMap<String, byte[]>(this.classBytes);
    }

    @Override
    public JavaFileObject getJavaFileForOutput(Location location, String className, JavaFileObject.Kind kind,
                                               FileObject sibling) throws IOException {
      if (kind == JavaFileObject.Kind.CLASS) {
        return new MemoryOutputJavaFileObject(className, classBytes);
      } else {
        return standardJavaFileManager.getJavaFileForOutput(location, className, kind, sibling);
      }
    }
  }

  private static class MemoryClassLoader extends LaunchedURLClassLoader {
    Map<String, byte[]> classBytes = new HashMap<>();

    public MemoryClassLoader(Map<String, byte[]> classBytes, ClassLoader classLoader) {
      super(new URL[0], classLoader);
      this.classBytes.putAll(classBytes);
    }

    @Override
    protected Class<?> findClass(String name) throws ClassNotFoundException {
      System.out.println("findClass: " + name);
      byte[] buf = classBytes.get(name);
      if (buf == null) {
        return super.findClass(name);
      }
      classBytes.remove(name);
      return defineClass(name, buf, 0, buf.length);
    }
  }


  public Class<?> loadClass(String name, Map<String, byte[]> classBytes) throws Exception {
    ClassLoader loader = new ClassLoader() {
      @Override
      public Class<?> loadClass(String name) throws ClassNotFoundException {
        Class<?> r = null;
        if (classBytes.containsKey(name)) {
          byte[] buf = classBytes.get(name);
          r =  defineClass(name, buf, 0, buf.length);
        } else {
          r = systemClassLoader.loadClass(name);
        }
        return r;
      }
    };
    return loader.loadClass(name);
  }

  private URLClassLoader systemClassLoader;
  public CusCompiler(URLClassLoader loader) {
    systemClassLoader = loader;
  }

  public Map<String, byte[]> compile(String className, String code) throws Exception {
    JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
    StandardJavaFileManager stdManager = compiler.getStandardFileManager(null, null, null);
    SpringBootJarFileManager springBootJarFileManager = new SpringBootJarFileManager(stdManager, systemClassLoader);
    JavaFileObject javaFileObject = new MemoryInputJavaFileObject(className, code);
    List<String> options = new ArrayList<>();
    //options.addAll(Arrays.asList("-classpath", System.getProperty("java.class.path"), "-bootclasspath", System.getProperty("sun.boot.class.path"), "-extdirs", System.getProperty("java.ext.dirs")));
    JavaCompiler.CompilationTask task = compiler.getTask(null, springBootJarFileManager, null, null, null, Arrays.asList(javaFileObject));
    Boolean compileRet= task.call();
    if (compileRet == null || (!compileRet.booleanValue())) {
      throw new RuntimeException("java filter compile error");
    }
    for (String key : springBootJarFileManager.getClassBytes().keySet()) {
      logger.info("class: " + key + " len: " + Integer.valueOf(springBootJarFileManager.getClassBytes().get(key).length).toString());
    }
    return springBootJarFileManager.getClassBytes();
  }
}

調用的時候 sample 如下：

         class_name = "com.yrxd.telemarketing.script." + class_name;
          //CusCompiler compiler = new CusCompiler((URLClassLoader)getClass().getClassLoader());
          CusCompiler compiler = new CusCompiler((URLClassLoader)Thread.currentThread().getContextClassLoader());
          Map<String, byte[]> results = compiler.compile(class_name, script);
          Class<?> clazz = compiler.loadClass(class_name, results);

踩坑的地方如下：

  1. 啟動 springboot 的時候，加上 -Xbootclasspath/a:/usr/local/java/lib/tools.jar
  2. 由于 springboot 特殊的機制，它的啟動 launcher 是可以從 fat jar 里面直接讀取資源的，所以要充分利用這一點，編譯的時候，利用 springboot 的loader 來查找資源，大膽的假設這個 loader 的類型為 URLClassLoader
  3. 盡量不要直接用springboot 啟動的 loader， loader 里面的資源讀過一次之后，后面讀不出來，這樣導致的后果是： 在 springboot 的 main 函數里面，直接動態加載會有問題，需要新建一個loader，繼承springboot 的loader
  4. 注意適用這種方式后，系統會有多個 loader 存在，系統的的loader 是無法loader 這里動態加載的class， 由此可能引發一系列的異常，需要 case by case分析
  5. 同一個classloader，對于相同的 classname， 不能 defineclass 調用多次，否則會出異常
  6. 參考：
    http://atamur.blogspot.com/2009/10/using-built-in-javacompiler-with-custom.html 實現自定義的 filemanager
    https://github.com/michaelliao/compiler 最開始找到的實現，不支持 springboot，通過把 springboot 解壓后，也勉強可以用，但是在多線程的情況下，使用 mybatisplus 的lambdaQuery 會有問題
最后編輯于 ：
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明：文章內容（如有圖片或視頻亦包括在內）由作者上傳并發布，文章內容僅代表作者本人觀點，簡書系信息發布平臺，僅提供信息存儲服務。
  • 人面猴
    序言：七十年代末，一起剝皮案震驚了整個濱河市，隨后出現的幾起案子，更是在濱河造成了極大的恐慌，老刑警劉巖，帶你破解...
    沈念sama閱讀 227,967評論 6贊 531
  • 死咒
    序言：濱河連續發生了三起死亡事件，死亡現場離奇詭異，居然都是意外死亡，警方通過查閱死者的電腦和手機，發現死者居然都...
    沈念sama閱讀 98,273評論 3贊 415
  • 救了他兩次的神仙讓他今天三更去死
    文/潘曉璐 我一進店門，熙熙樓的掌柜王于貴愁眉苦臉地迎上來，“玉大人，你說我怎么就攤上這事?！?“怎么了？”我有些...
    開封第一講書人閱讀 175,870評論 0贊 373
  • 道士緝兇錄：失蹤的賣姜人
    文/不壞的土叔 我叫張陵，是天一觀的道長。 經常有香客問我，道長，這世上最難降的妖魔是什么？ 我笑而不...
    開封第一講書人閱讀 62,742評論 1贊 309
  • ?港島之戀（遺憾婚禮）
    正文 為了忘掉前任，我火速辦了婚禮，結果婚禮上，老公的妹妹穿的比我還像新娘。我一直安慰自己，他們只是感情好，可當我...
    茶點故事閱讀 71,527評論 6贊 407
  • 惡毒庶女頂嫁案：這布局不是一般人想出來的
    文/花漫 我一把揭開白布。 她就那樣靜靜地躺著，像睡著了一般。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發上，一...
    開封第一講書人閱讀 55,010評論 1贊 322
  • 城市分裂傳說
    那天，我揣著相機與錄音，去河邊找鬼。 笑死，一個胖子當著我的面吹牛，可吹牛的內容都是我干的。 我是一名探鬼主播，決...
    沈念sama閱讀 43,108評論 3贊 440
  • 雙鴛鴦連環套：你想象不到人心有多黑
    文/蒼蘭香墨 我猛地睜開眼，長吁一口氣：“原來是場噩夢啊……” “哼！你這毒婦竟也來了？” 一聲冷哼從身側響起，我...
    開封第一講書人閱讀 42,250評論 0贊 288
  • 萬榮殺人案實錄
    序言：老撾萬榮一對情侶失蹤，失蹤者是張志新（化名）和其女友劉穎，沒想到半個月后，有當地人在樹林里發現了一具尸體，經...
    沈念sama閱讀 48,769評論 1贊 333
  • ?護林員之死
    正文 獨居荒郊野嶺守林人離奇死亡，尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內容為張勛視角 年9月15日...
    茶點故事閱讀 40,656評論 3贊 354
  • ?白月光啟示錄
    正文 我和宋清朗相戀三年，在試婚紗的時候發現自己被綠了。 大學時的朋友給我發了我未婚夫和他白月光在一起吃飯的照片。...
    茶點故事閱讀 42,853評論 1贊 369
  • 活死人
    序言：一個原本活蹦亂跳的男人離奇死亡，死狀恐怖，靈堂內的尸體忽然破棺而出，到底是詐尸還是另有隱情，我是刑警寧澤，帶...
    沈念sama閱讀 38,371評論 5贊 358
  • ?日本核電站爆炸內幕
    正文 年R本政府宣布，位于F島的核電站，受9級特大地震影響，放射性物質發生泄漏。R本人自食惡果不足惜，卻給世界環境...
    茶點故事閱讀 44,103評論 3贊 347
  • 男人毒藥：我在死后第九天來索命
    文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望。 院中可真熱鬧，春花似錦、人聲如沸。這莊子的主人今日做“春日...
    開封第一講書人閱讀 34,472評論 0贊 26
  • 一樁弒父案，背后竟有這般陰謀
    文/蒼蘭香墨 我抬頭看了看天上的太陽。三九已至，卻和暖如春，著一層夾襖步出監牢的瞬間，已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 35,717評論 1贊 281
  • 情欲美人皮
    我被黑心中介騙來泰國打工， 沒想到剛下飛機就差點兒被人妖公主榨干…… 1. 我叫王不留，地道東北人。 一個月前我還...
    沈念sama閱讀 51,487評論 3贊 390
  • 代替公主和親
    正文 我出身青樓，卻偏偏與公主長得像，于是被迫代替她去往敵國和親。 傳聞我的和親對象是個殘疾皇子，可洞房花燭夜當晚...
    茶點故事閱讀 47,815評論 2贊 372