feat():测试

cloud-rule-engine/src/main/java/com/muyu/compile/OSSFileCompile.java

@@ -2,10 +2,13 @@ package com.muyu.compile;
 
 
 import com.muyu.common.core.domain.Result;
+import org.apache.catalina.startup.EngineConfig;
 
 import javax.tools.*;
 import java.io.File;
 import java.io.IOException;
+import java.nio.file.Path;
+import java.nio.file.Paths;
 import java.util.Arrays;
 import java.util.List;
 import java.util.Map;
@@ -40,22 +43,17 @@ public class OSSFileCompile{
     public static Result<Object> compile(String source,String extLib){
         JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
 
-        // 设置编译选项，这里我们将输出目录设置为源文件的目录
-        String outputDir = new File(source).getParent();
-        String[] options = new String[]{"-d", outputDir};
-
         // 执行编译任务
-        int result = compiler.run(null, null, null, Arrays.toString(options), source);
+        int result = compiler.run(null, null, null, source);
 
         // 检查编译结果
         if (result == 0) {
             System.out.println("编译成功，生成的.class文件位于源代码同目录");
-            Result.success("编译成功，生成的.class文件位于源代码同目录");
+            return Result.success("编译成功，生成的.class文件位于源代码同目录");
         } else {
             System.out.println("编译失败");
-            Result.error("编译失败");
+            return Result.error("编译失败");
         }
-        return Result.warn();
     }
 
 }

cloud-rule-engine/src/main/java/com/muyu/load/OSSFileLoad.java

@@ -6,11 +6,9 @@ import com.aliyun.oss.model.OSSObject;
 import com.aliyun.oss.model.ObjectMetadata;
 import com.muyu.common.core.domain.Result;
 import com.muyu.compile.OSSFileCompile;
+import com.muyu.util.SourceCodeCompiler;
 
-import java.io.BufferedReader;
+import java.io.*;
-import java.io.File;
-import java.io.IOException;
-import java.io.InputStreamReader;
 import java.util.Map;
 
 /**
@@ -51,6 +49,57 @@ public class OSSFileLoad {
         String localPath = "home/"+fileName;
         ObjectMetadata object = ossClient.getObject(new GetObjectRequest(bucketName, filePath + fileName), new File(localPath));
         System.out.println(object);
+
+        ossClient.shutdown();
+
+        try {
+            // 从OSS下载文件
+            InputStream inputStream = ossClient.getObject(bucketName, filePath+fileName).getObjectContent();
+            File downloadFile = new File(localPath);
+            FileOutputStream outputStream = new FileOutputStream(downloadFile);
+
+            byte[] buffer = new byte[1024];
+            int len;
+            while ((len = inputStream.read(buffer)) != -1) {
+                outputStream.write(buffer, 0, len);
+            }
+            outputStream.close();
+            inputStream.close();
+
+            System.out.println("存放路径:" + localPath);
+        } catch (Exception e) {
+            e.printStackTrace();
+        } finally {
+            // 关闭OSSClient
+            if (ossClient != null) {
+                ossClient.shutdown();
+            }
+        }
+
+        //对路径里的.java文件进行编译
+        SourceCodeCompiler.javaCompilerPath("/home");
+        System.out.println("第一步");
+        File outputDir = new File("/home"); // 或者是你指定的其他输出目录
+        System.out.println("第一步");
+        File[] classFiles = outputDir.listFiles(); // 获取输出目录中的所有文件
+        System.out.println("第三步");
+        if (classFiles != null) {
+            for (File classFile : classFiles) {
+                if (classFile.getName().endsWith(".class")) {
+                    // 这里可以处理每个.class文件，例如读取、复制或移动
+                    System.out.println("找到class文件 " + classFile.getName());
+                }
+            }
+            //把.class文件存入oss中
+        }else {
+            System.out.println("没有找到文件");
+        }
+
+        return Result.success(source);
+    }
+
+}
+
 //        OSSObject ossObject = ossClient.getObject(bucketName, fileName);
 //        BufferedReader reader = new BufferedReader(new InputStreamReader(ossObject.getObjectContent()));
 //        while(true){
@@ -61,12 +110,4 @@ public class OSSFileLoad {
 //        System.out.println(source);
 //        System.out.println("读取成功");
 //        reader.close();
-        ossClient.shutdown();
-
 //        Map<String, byte[]> compile = OSSFileCompile.compile(fileName, source, "target/");
-        Result<Object> compile = OSSFileCompile.compile(localPath, null);
-        System.out.println(compile.getData());
-        return Result.success(source);
-    }
-
-}

cloud-rule-engine/src/main/java/com/muyu/util/JavaCodeScan.java

@@ -0,0 +1,169 @@
+package com.muyu.util;
+
+import lombok.extern.log4j.Log4j2;
+import org.springframework.stereotype.Component;
+
+import java.io.*;
+import java.util.Arrays;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @ClassName JavaCodeScan
+ * @Description 描述
+ * @Author YiBo.Liu
+ * @Date 2024/8/23 10:36
+ */
+@Log4j2
+public class JavaCodeScan extends ClassLoader{
+
+
+    public static final String JAVA_SOURCE_SUF = ".java";
+    public static final String JAVA_BINARY_SUF = ".class";
+
+    private static final JavaCodeScan loader = new JavaCodeScan();
+
+    /**
+     *
+     * @param filePath 文件路径
+     * @param suf 扫描后缀
+     * @return
+     */
+    public static File[]  filterFileBySuf(String filePath,String suf){
+        //创建需要扫描的文件夹
+        File file = new File(filePath);
+        //列出所有的筛选suf结尾的文件集
+        //file.listFiles(筛选)
+         File[] childFiles = file.listFiles(new FilenameFilter() {
+            @Override
+            public boolean accept(File dir, String name) {
+                if(name.indexOf(suf) <= -1){
+                    return false;
+                }else {
+                    log.info("扫描到{}文件->{}",suf,name);
+                    return true;
+                }
+
+            }
+        });
+        return childFiles;
+    }
+
+    /**
+     * @param pack     类所在的包名 com.muyu.bean. 后面需要带个点
+     * @param location 包所在路径
+     * @return
+     */
+    public static Map<String, Class<?>> loadClassByLocation(String pack, String location) {
+        //列出所有的class文件
+        List<File> classFiles = JavaCodeScan.getClassFile(location);
+        HashMap<String, Class<?>> map = new HashMap<>();
+        classFiles.forEach(file -> {
+            String name = file.getName().substring(0, file.getName().lastIndexOf(JAVA_BINARY_SUF));
+            Class<?> c = loadClassByNameAndLocation(name, pack + ".", new File(location));
+            map.put(name, c);
+        });
+        return map;
+    }
+
+
+    /**
+     *自定义路径获取类的class对象
+     * @param name
+     * @param pack
+     * @param locationFile
+     * @return
+     */
+    private static Class<?> loadClassByNameAndLocation(String name, String pack, File locationFile) {
+        //将class文件的数据读入到byte数组中
+        byte[] datas = loader.loadClassData(name, locationFile);
+        //通过byte数组加载Class对象
+        Class<?> aClass = loader.defineClass(pack + name, datas, 0, datas.length);
+        log.info("成功加载规则引擎 -- {}", aClass.getName());
+        return aClass;
+    }
+
+    /**
+     * 将文件转换成字节数组
+     *
+     * @param name         文件名称
+     * @param locationFile 文件
+     * @return
+     */
+    public byte[] loadClassData(String name, File locationFile) {
+        byte[] byteArray = null;
+        try {FileInputStream fis = new FileInputStream(locationFile + "/" + name + JAVA_BINARY_SUF);
+            // 创建ByteArrayOutputStream对象
+            ByteArrayOutputStream bos = new ByteArrayOutputStream();
+            // 创建一个缓冲区
+            byte[] buffer = new byte[1024];
+            int length;
+            // 读取文件内容并写入缓冲区
+            while ((length = fis.read(buffer)) != -1) {
+                bos.write(buffer, 0, length);
+            }
+            // 将缓冲区中的内容转换为byte数组
+            byteArray = bos.toByteArray();
+        } catch (IOException e) {
+            log.error("[【{}】文件转换成字节数组错误]：{}", name, e.getMessage(), e);
+            throw new RuntimeException(e);
+        }
+        return byteArray;
+    }
+
+
+    /**
+     * 判断java文件类型
+     * @param file 文件
+     * @return 布尔类型的值
+     */
+    public static Boolean isFileType(File file) {
+        return file.getName().endsWith(JAVA_SOURCE_SUF);
+    }
+
+    /**
+     * 判断是否为class文件类型
+     * @param file
+     * @return
+     */
+    public static Boolean isClassFileType(File file){
+        return file.getName().endsWith(JAVA_BINARY_SUF);
+    }
+
+    /**
+     * 通过文件路径获取该文件夹下的所有java类型文件
+     * @param folderPath 文件夹路径
+     * @return java文件
+     */
+    public static List<File> getJavaFiles(String folderPath) {
+        File folder = new File(folderPath);
+        if (!folder.exists()){
+            folder.mkdirs();
+        }
+        File[] files = folder.listFiles();
+        return Arrays.stream(isFilesExist(files)).filter(JavaCodeScan::isFileType).toList();
+    }
+
+
+    /**
+     * 通过路劲获取所有的.class文件
+     * @param location
+     * @return
+     */
+    private static List<File> getClassFile(String location) {
+        File file = new File(location);
+        if(!file.exists()){
+            file.mkdirs();
+        }
+        File[] files = file.listFiles();
+        return Arrays.stream(isFilesExist(files)).filter(JavaCodeScan::isClassFileType).toList();
+    }
+
+    private static File[] isFilesExist(File[] files) {
+        if(files == null || files.length == 0){
+            throw new RuntimeException("该文件夹下没有文件");
+        }
+        return files;
+    }
+}

cloud-rule-engine/src/main/java/com/muyu/util/SourceCodeCompiler.java

@@ -0,0 +1,72 @@
+package com.muyu.util;
+
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+import org.springframework.stereotype.Component;
+
+import javax.tools.JavaCompiler;
+import javax.tools.JavaFileObject;
+import javax.tools.StandardJavaFileManager;
+import javax.tools.ToolProvider;
+import java.io.File;
+import java.util.Arrays;
+import java.util.List;
+
+/**
+ * @ClassName SourceCodeCompiler
+ * @Description 描述
+ * @Author YiBo.Liu
+ * @Date 2024/8/23 10:35
+ */
+public class SourceCodeCompiler {
+
+    private static final Logger log = LogManager.getLogger(SourceCodeCompiler.class.getName());
+
+    /**
+     * 获取系统java编译器
+     */
+    private static JavaCompiler compiler = ToolProvider.getSystemJavaCompiler();
+    /**
+     * 获取java文件管理器
+     */
+    private static StandardJavaFileManager fileManager = compiler.getStandardFileManager(null, null, null);
+
+
+    /**
+     * 输入编译的文件夹路径
+     * @param path 文件夹路径
+     */
+    public static void javaCompilerPath(String path){
+        List<File> files = JavaCodeScan.getJavaFiles(path);
+        File[] array = files.toArray(new File[files.size()]);
+        javaCompiler(array);
+
+
+    }
+
+
+    public static void javaCompilerFile(String... filePath){
+        int filePathLength =  filePath.length;
+        File[] files = new File[filePathLength];
+
+        for (int i = 0; i < filePathLength; i++) {
+            files[i] = new File(filePath[i]);
+        }
+        javaCompiler(files);
+    }
+
+
+    /**
+     *可以同时编译多个java源代码
+     * @param file
+     */
+    public static void javaCompiler(File...file){
+        //通过源文件获取到想要编译的java类源代码迭代器，包括所有的内部类，其中每一个类都是一个JavaFileObjects，也被称为一个汇编单元
+        Iterable<? extends JavaFileObject> javaFileObjects = fileManager.getJavaFileObjects(file);
+        //生成编译任务
+        JavaCompiler.CompilationTask task = compiler.getTask(null, fileManager, null, Arrays.asList("-d","home"), null, javaFileObjects);
+        //执行编译任务
+        task.call();
+    }
+
+}