springboot 2 下 java 代碼 動(dòng)態(tài)編譯 動(dòng)態(tài)加載 實(shí)現(xiàn)

先放出代碼其骄,再說有幾個(gè)坑的地方:

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();
  }
}

調(diào)用的時(shí)候 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. 啟動(dòng) springboot 的時(shí)候炮温,加上 -Xbootclasspath/a:/usr/local/java/lib/tools.jar
  2. 由于 springboot 特殊的機(jī)制,它的啟動(dòng) launcher 是可以從 fat jar 里面直接讀取資源的,所以要充分利用這一點(diǎn),編譯的時(shí)候姥饰,利用 springboot 的loader 來查找資源,大膽的假設(shè)這個(gè) loader 的類型為 URLClassLoader
  3. 盡量不要直接用springboot 啟動(dòng)的 loader孝治, loader 里面的資源讀過一次之后媳否,后面讀不出來,這樣導(dǎo)致的后果是: 在 springboot 的 main 函數(shù)里面荆秦,直接動(dòng)態(tài)加載會(huì)有問題,需要新建一個(gè)loader力图,繼承springboot 的loader
  4. 注意適用這種方式后步绸,系統(tǒng)會(huì)有多個(gè) loader 存在,系統(tǒng)的的loader 是無法loader 這里動(dòng)態(tài)加載的class吃媒, 由此可能引發(fā)一系列的異常瓤介,需要 case by case分析
  5. 同一個(gè)classloader,對(duì)于相同的 classname赘那, 不能 defineclass 調(diào)用多次刑桑,否則會(huì)出異常
  6. 參考:
    http://atamur.blogspot.com/2009/10/using-built-in-javacompiler-with-custom.html 實(shí)現(xiàn)自定義的 filemanager
    https://github.com/michaelliao/compiler 最開始找到的實(shí)現(xiàn),不支持 springboot募舟,通過把 springboot 解壓后祠斧,也勉強(qiáng)可以用,但是在多線程的情況下拱礁,使用 mybatisplus 的lambdaQuery 會(huì)有問題
最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
  • 序言:七十年代末琢锋,一起剝皮案震驚了整個(gè)濱河市,隨后出現(xiàn)的幾起案子呢灶,更是在濱河造成了極大的恐慌吴超,老刑警劉巖,帶你破解...
    沈念sama閱讀 218,858評(píng)論 6 508
  • 序言:濱河連續(xù)發(fā)生了三起死亡事件鸯乃,死亡現(xiàn)場(chǎng)離奇詭異鲸阻,居然都是意外死亡,警方通過查閱死者的電腦和手機(jī)缨睡,發(fā)現(xiàn)死者居然都...
    沈念sama閱讀 93,372評(píng)論 3 395
  • 文/潘曉璐 我一進(jìn)店門鸟悴,熙熙樓的掌柜王于貴愁眉苦臉地迎上來,“玉大人奖年,你說我怎么就攤上這事遣臼。” “怎么了拾并?”我有些...
    開封第一講書人閱讀 165,282評(píng)論 0 356
  • 文/不壞的土叔 我叫張陵揍堰,是天一觀的道長鹏浅。 經(jīng)常有香客問我,道長屏歹,這世上最難降的妖魔是什么隐砸? 我笑而不...
    開封第一講書人閱讀 58,842評(píng)論 1 295
  • 正文 為了忘掉前任,我火速辦了婚禮蝙眶,結(jié)果婚禮上季希,老公的妹妹穿的比我還像新娘。我一直安慰自己幽纷,他們只是感情好式塌,可當(dāng)我...
    茶點(diǎn)故事閱讀 67,857評(píng)論 6 392
  • 文/花漫 我一把揭開白布。 她就那樣靜靜地躺著友浸,像睡著了一般峰尝。 火紅的嫁衣襯著肌膚如雪。 梳的紋絲不亂的頭發(fā)上收恢,一...
    開封第一講書人閱讀 51,679評(píng)論 1 305
  • 那天武学,我揣著相機(jī)與錄音,去河邊找鬼伦意。 笑死火窒,一個(gè)胖子當(dāng)著我的面吹牛,可吹牛的內(nèi)容都是我干的驮肉。 我是一名探鬼主播熏矿,決...
    沈念sama閱讀 40,406評(píng)論 3 418
  • 文/蒼蘭香墨 我猛地睜開眼,長吁一口氣:“原來是場(chǎng)噩夢(mèng)啊……” “哼离钝!你這毒婦竟也來了曲掰?” 一聲冷哼從身側(cè)響起,我...
    開封第一講書人閱讀 39,311評(píng)論 0 276
  • 序言:老撾萬榮一對(duì)情侶失蹤奈辰,失蹤者是張志新(化名)和其女友劉穎栏妖,沒想到半個(gè)月后,有當(dāng)?shù)厝嗽跇淞掷锇l(fā)現(xiàn)了一具尸體奖恰,經(jīng)...
    沈念sama閱讀 45,767評(píng)論 1 315
  • 正文 獨(dú)居荒郊野嶺守林人離奇死亡吊趾,尸身上長有42處帶血的膿包…… 初始之章·張勛 以下內(nèi)容為張勛視角 年9月15日...
    茶點(diǎn)故事閱讀 37,945評(píng)論 3 336
  • 正文 我和宋清朗相戀三年,在試婚紗的時(shí)候發(fā)現(xiàn)自己被綠了瑟啃。 大學(xué)時(shí)的朋友給我發(fā)了我未婚夫和他白月光在一起吃飯的照片论泛。...
    茶點(diǎn)故事閱讀 40,090評(píng)論 1 350
  • 序言:一個(gè)原本活蹦亂跳的男人離奇死亡,死狀恐怖蛹屿,靈堂內(nèi)的尸體忽然破棺而出屁奏,到底是詐尸還是另有隱情,我是刑警寧澤错负,帶...
    沈念sama閱讀 35,785評(píng)論 5 346
  • 正文 年R本政府宣布坟瓢,位于F島的核電站勇边,受9級(jí)特大地震影響,放射性物質(zhì)發(fā)生泄漏折联。R本人自食惡果不足惜粒褒,卻給世界環(huán)境...
    茶點(diǎn)故事閱讀 41,420評(píng)論 3 331
  • 文/蒙蒙 一、第九天 我趴在偏房一處隱蔽的房頂上張望诚镰。 院中可真熱鬧奕坟,春花似錦、人聲如沸清笨。這莊子的主人今日做“春日...
    開封第一講書人閱讀 31,988評(píng)論 0 22
  • 文/蒼蘭香墨 我抬頭看了看天上的太陽抠艾。三九已至苛萎,卻和暖如春,著一層夾襖步出監(jiān)牢的瞬間跌帐,已是汗流浹背。 一陣腳步聲響...
    開封第一講書人閱讀 33,101評(píng)論 1 271
  • 我被黑心中介騙來泰國打工绊率, 沒想到剛下飛機(jī)就差點(diǎn)兒被人妖公主榨干…… 1. 我叫王不留谨敛,地道東北人。 一個(gè)月前我還...
    沈念sama閱讀 48,298評(píng)論 3 372
  • 正文 我出身青樓滤否,卻偏偏與公主長得像脸狸,于是被迫代替她去往敵國和親。 傳聞我的和親對(duì)象是個(gè)殘疾皇子藐俺,可洞房花燭夜當(dāng)晚...
    茶點(diǎn)故事閱讀 45,033評(píng)論 2 355