工业互联网平台
blame | 历史 | 补丁 | 提交 | 提交对比 | 忽略空白
dengzedong
4 天以前 730d1944e3a13c517c77df2b0712df05645a38dd 
 iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mpk/service/impl/MpkFileServiceImpl.java


@@ -1,7 +1,6 @@


package com.iailab.module.model.mpk.service.impl;




import cn.hutool.core.io.FileUtil;


import cn.hutool.core.util.RuntimeUtil;


import cn.hutool.core.util.ZipUtil;


import com.alibaba.fastjson.JSON;


import com.baomidou.dynamic.datasource.annotation.DSTransactional;


@@ -15,14 +14,17 @@


import com.iailab.framework.security.core.util.SecurityFrameworkUtils;


import com.iailab.framework.tenant.core.context.TenantContextHolder;


import com.iailab.module.infra.api.config.ConfigApi;


import com.iailab.module.model.mcs.pre.service.MmModelArithSettingsService;


import com.iailab.module.model.mcs.sche.service.StScheduleModelSettingService;


import com.iailab.module.model.mpk.common.MdkConstant;


import com.iailab.module.model.mpk.common.utils.CmdUtils;


import com.iailab.module.model.mpk.common.utils.DllUtils;


import com.iailab.module.model.mpk.common.utils.GenUtils;


import com.iailab.module.model.mpk.dao.MpkFileDao;


import com.iailab.module.model.mpk.dto.GeneratorCodeHistoryDTO;


import com.iailab.module.model.mpk.dto.MethodSettingDTO;


import com.iailab.module.model.mpk.dto.ModelMethodDTO;


import com.iailab.module.model.mpk.dto.MpkFileDTO;


import com.iailab.module.model.mpk.dto.ProjectPackageHistoryDTO;


import com.iailab.module.model.mpk.entity.GeneratorCodeHistoryEntity;


import com.iailab.module.model.mpk.entity.MpkFileEntity;


import com.iailab.module.model.mpk.entity.ProjectPackageHistoryEntity;


@@ -39,10 +41,12 @@


import org.springframework.util.CollectionUtils;


import org.springframework.web.multipart.MultipartFile;




import java.io.*;


import java.io.File;


import java.io.IOException;


import java.net.URLClassLoader;


import java.nio.file.Files;


import java.util.*;


import java.util.function.Function;


import java.util.stream.Collectors;




/**


@@ -67,6 +71,12 @@


    private ProjectPackageHistoryModelService projectPackageHistoryModelService;




    @Autowired


    private MmModelArithSettingsService mmModelArithSettingsService;




    @Autowired


    private StScheduleModelSettingService stScheduleModelSettingService;




    @Autowired


    private ConfigApi configApi;




    @Value("${mpk.bak-file-path}")


@@ -87,19 +97,22 @@




    @Override


    public List<MpkFileDTO> list(Map<String, Object> params) {


        List<MpkFileEntity> entityList = baseDao.selectList(getWrapper(params).orderByDesc("create_date"));


//        List<MpkFileEntity> entityList = baseDao.selectList(getWrapper(params).orderByDesc("create_date"));


        List<MpkFileDTO> list = baseDao.list(params);




        return ConvertUtils.sourceToTarget(entityList, MpkFileDTO.class);


        return list;


    }




    private QueryWrapper<MpkFileEntity> getWrapper(Map<String, Object> params) {


        String pyChineseName = (String) params.get("pyChineseName");


        String pyName = (String) params.get("pyName");


        String pyType = (String) params.get("pyType");


        String remark = (String) params.get("remark");


        String label = (String) params.get("label");




        QueryWrapper<MpkFileEntity> wrapper = new QueryWrapper<>();


        wrapper.like(StringUtils.isNotBlank(pyName), "py_name", pyName)


        wrapper.like(StringUtils.isNotBlank(pyChineseName), "py_chinese_name", pyChineseName)


                .like(StringUtils.isNotBlank(pyName), "py_name", pyName)


                .eq(StringUtils.isNotBlank(pyType), "py_type", pyType)


                .like(StringUtils.isNotBlank(remark), "remark", remark);




@@ -129,30 +142,87 @@


        entity.setCreator(SecurityFrameworkUtils.getLoginUserId());


        entity.setCreateDate(new Date());


        insert(entity);


        modelMethodService.insertList(dto.getModelMethods(), entity.getId());




        // 替换环境变量MDK_PKGS下的py文件


        // 将备份的pyd文件，转移到MDK_PKGS环境变量下，并添加方法的默认参数（pyFile，模型路径）


        String mdkPkgs = System.getenv("MDK_PKGS");


        String pyFilePath = mdkPkgs + File.separator + entity.getPyModule().replace(".", File.separator) + File.separator + entity.getPyName() + ".pyd";


        String fileName = entity.getFilePath().substring(entity.getFilePath().lastIndexOf("\\") + 1,entity.getFilePath().lastIndexOf(".pyd"));




        String pyFilePath = mdkPkgs + File.separator + entity.getPyModule().replace(".", File.separator) + File.separator + fileName + ".pyd";


        FileUtil.mkParentDirs(pyFilePath);


        FileUtil.copy(entity.getFilePath(), pyFilePath, true);




        // 添加参数


        for (ModelMethodDTO method : dto.getModelMethods()) {


            List<MethodSettingDTO> methodSettings = method.getMethodSettings();


            if (methodSettings.stream().anyMatch(e -> e.getSettingKey().equals(MdkConstant.PY_FILE_KEY))) {


                methodSettings.forEach(e -> {


                    if (e.getSettingKey().equals(MdkConstant.PY_FILE_KEY)) {


                        e.setValue(entity.getPyModule() + "." + fileName);


                    }


                });


            }else {


                MethodSettingDTO setting = new MethodSettingDTO();


                setting.setId(UUID.randomUUID().toString());


                setting.setMethodId(method.getId());


                setting.setSettingKey(MdkConstant.PY_FILE_KEY);


                setting.setValue(entity.getPyModule() + "." + fileName);


                setting.setName("模型路径");


                setting.setType("input");


                setting.setValueType("string");


                methodSettings.add(setting);


            }


        }




        modelMethodService.insertList(dto.getModelMethods(), entity.getId());


    }




    @Override


    @DSTransactional(rollbackFor = Exception.class)


    public void update(MpkFileDTO dto) {


        // 添加方法的默认参数（pyFile，模型路径）


        String mdkPkgs = System.getenv("MDK_PKGS");


        String fileName = dto.getFilePath().substring(dto.getFilePath().lastIndexOf("\\") + 1,dto.getFilePath().lastIndexOf(".pyd"));


        String pyFilePath = mdkPkgs + File.separator + dto.getPyModule().replace(".", File.separator) + File.separator + fileName + ".pyd";


        // 判断文件是否存在,不存在的话将备份的pyd文件，转移到MDK_PKGS环境变量下


        File pyFile = new File(pyFilePath);


        if (!pyFile.exists()) {


            FileUtil.mkParentDirs(pyFilePath);


            FileUtil.copy(dto.getFilePath(), pyFilePath, true);


        }




        // 添加/修改参数


        for (ModelMethodDTO method : dto.getModelMethods()) {


            List<MethodSettingDTO> methodSettings = method.getMethodSettings();


            if (methodSettings.stream().anyMatch(e -> e.getSettingKey().equals(MdkConstant.PY_FILE_KEY))) {


                methodSettings.forEach(e -> {


                    if (e.getSettingKey().equals(MdkConstant.PY_FILE_KEY)) {


                        e.setValue(dto.getPyModule() + "." + fileName);


                    }


                });


            }else {


                MethodSettingDTO setting = new MethodSettingDTO();


                setting.setId(UUID.randomUUID().toString());


                setting.setMethodId(method.getId());


                setting.setSettingKey(MdkConstant.PY_FILE_KEY);


                setting.setValue(dto.getPyModule() + "." + fileName);


                setting.setName("模型路径");


                setting.setType("input");


                setting.setValueType("string");


                methodSettings.add(setting);


            }


        }




        // 修改预测项pyFile参数


        mmModelArithSettingsService.updatePyFile(dto.getPyModule(),fileName);


        // 修改调度项pyFile参数


        stScheduleModelSettingService.updatePyFile(dto.getPyModule(),fileName);




        MpkFileEntity entity = ConvertUtils.sourceToTarget(dto, MpkFileEntity.class);


        entity.setUpdater(SecurityFrameworkUtils.getLoginUserId());


        entity.setUpdateDate(new Date());


        updateById(entity);


        modelMethodService.deleteModelMethod(entity.getId());


        modelMethodService.insertList(dto.getModelMethods(), entity.getId());




        // 替换环境变量MDK_PKGS下的py文件


        String mdkPkgs = System.getenv("MDK_PKGS");


        String pyFilePath = mdkPkgs + File.separator + entity.getPyModule().replace(".", File.separator) + File.separator + entity.getPyName() + ".pyd";


        FileUtil.mkParentDirs(pyFilePath);


        FileUtil.copy(entity.getFilePath(), pyFilePath, true);


    }




    @Override


@@ -217,27 +287,24 @@


        }




        List<String> javaFilePaths = new ArrayList<>();


        List<String> cppFilePaths = new ArrayList<>();




        //生成java文件


        File javaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + entity.getPyName() + ".java");


        File javaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + entity.getPyName() + ".java");


        GenUtils.drawTemplate("abstract.java.vm", dataContext, javaFile);


        javaFilePaths.add(javaFile.getAbsolutePath());




        //生成Impl.java文件


        File implJavaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + MdkConstant.IMPL + File.separator + entity.getPyName() + "Impl.java");


        File implJavaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + MdkConstant.IMPL + File.separator + entity.getPyName() + "Impl.java");


        GenUtils.drawTemplate("impl.java.vm", dataContext, implJavaFile);


        javaFilePaths.add(implJavaFile.getAbsolutePath());




        //生成.cpp文件


        File cppFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.C + File.separator + entity.getPyName() + ".cpp");


        GenUtils.drawTemplate("cpp.vm", dataContext, cppFile);


        cppFilePaths.add(cppFile.getAbsolutePath());




        //生成Jni.cpp文件


        File jniCppFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.C + File.separator + entity.getPyName() + "Jni.cpp");


        GenUtils.drawTemplate("Jni.cpp.vm", dataContext, jniCppFile);


        cppFilePaths.add(jniCppFile.getAbsolutePath());




        //生成.h文件


        File hFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.C + File.separator + entity.getPyName() + ".h");


@@ -247,28 +314,33 @@


        File jniHFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.C + File.separator + entity.getPyName() + "Jni.h");


        GenUtils.drawTemplate("Jni.h.vm", dataContext, jniHFile);




        // 添加python源文件


        String pyFilePath = dirPath.getAbsolutePath() + File.separator + MdkConstant.ALGS + File.separator + entity.getPyModule().replace(".", File.separator) + File.separator + entity.getFilePath().substring(entity.getFilePath().lastIndexOf("\\"));


        FileUtil.mkParentDirs(pyFilePath);


        FileUtil.copy(entity.getFilePath(), pyFilePath, true);




        // 资源文件


        FileUtil.copy(mpkResources + File.separator + "libs", dirPath.getAbsolutePath(), true);




        //生成dll文件


        String dllSavePath = dirPath.getAbsolutePath() + File.separator + MdkConstant.LIBS + File.separator + "IAIL.MDK.Mid.Jni.dll";


        createDllFile(dirPath.getAbsolutePath(),cppFilePaths,dllSavePath);


        createDllFile(dirPath.getAbsolutePath(),dllSavePath);


        //备份dll文件，用于后续运行


        String dllBakPath = mpkTenantBakFilePath + File.separator + MdkConstant.DLL + File.separator + entity.getPyName() + ".dll";


        FileUtil.mkParentDirs(dllBakPath);


        FileUtil.copy(dllSavePath, dllBakPath, true);




        //utils + env java文件


        File utilsJavaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "utils" + File.separator + "AlgsUtils.java");


        File utilsJavaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "utils" + File.separator + "AlgsUtils.java");


        FileUtil.mkParentDirs(utilsJavaFile);


        FileUtil.copy(mpkResources + File.separator +"IAILMDK/utils/AlgsUtils.java", utilsJavaFile.getAbsolutePath(), true);


        javaFilePaths.add(utilsJavaFile.getAbsolutePath());


        File envJavaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "common" + File.separator + "Environment.java");


        File envJavaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "common" + File.separator + "Environment.java");


        FileUtil.mkParentDirs(envJavaFile);


        FileUtil.copy(mpkResources + File.separator +"IAILMDK/common/Environment.java", envJavaFile.getAbsolutePath(), true);


        javaFilePaths.add(envJavaFile.getAbsolutePath());


        // 生成class文件


        createClassFile(javaFilePaths);


        createClassFile(dirPath.getAbsolutePath(),javaFilePaths);


        // 删除java源文件


        deleteJavaFile(javaFilePaths);


        // 打jar包


@@ -303,12 +375,12 @@




    @Override


    @DSTransactional(rollbackFor = Exception.class)


    public byte[] packageModel(List<String> ids, String projectId, String projectName, String zipFileName, String logs, String version) throws IOException, InterruptedException {


    public byte[] packageModel(String projectId, String projectName, String zipFileName, String logs, String version) throws IOException {


        Long tenantId = TenantContextHolder.getTenantId();


        // 备份文件 租户隔离


        String mpkTenantBakFilePath = mpkBakFilePath + File.separator + tenantId;




        List<MpkFileDTO> entities = baseDao.selectByIds(ids);


        List<MpkFileDTO> entities = baseDao.selectByProjectId(projectId);




        //模板数据


//        Map<String, Object> map = new HashMap<>();


@@ -337,7 +409,6 @@


        GenUtils.drawTemplate("menu.xml.vm", new VelocityContext(map1), xmlFile);




        List<String> javaFilePaths = new ArrayList<>();


        List<String> cppFilePaths = new ArrayList<>();




        //生成java文件


        for (MpkFileDTO entity : entities) {


@@ -349,24 +420,22 @@


            data.put("pyModule", entity.getPyModule());


            VelocityContext dataContext = new VelocityContext(data);


            //生成java文件


            File javaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + entity.getPyName() + ".java");


            File javaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + entity.getPyName() + ".java");


            GenUtils.drawTemplate("abstract.java.vm", dataContext, javaFile);


            javaFilePaths.add(javaFile.getAbsolutePath());




            //生成Impl.java文件


            File implJavaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + MdkConstant.IMPL + File.separator + entity.getPyName() + "Impl.java");


            File implJavaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + entity.getPkgName().replace(".", File.separator) + File.separator + MdkConstant.IMPL + File.separator + entity.getPyName() + "Impl.java");


            GenUtils.drawTemplate("impl.java.vm", dataContext, implJavaFile);


            javaFilePaths.add(implJavaFile.getAbsolutePath());




            //生成.cpp文件


            File cppFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.C + File.separator + entity.getPyName() + ".cpp");


            GenUtils.drawTemplate("cpp.vm", dataContext, cppFile);


            cppFilePaths.add(cppFile.getAbsolutePath());




            //生成Jni.cpp文件


            File jniCppFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.C + File.separator + entity.getPyName() + "Jni.cpp");


            GenUtils.drawTemplate("Jni.cpp.vm", dataContext, jniCppFile);


            cppFilePaths.add(jniCppFile.getAbsolutePath());




            //生成.h文件


            File hFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.C + File.separator + entity.getPyName() + ".h");


@@ -377,7 +446,7 @@


            GenUtils.drawTemplate("Jni.h.vm", dataContext, jniHFile);




            // 添加python源文件


            String pyFilePath = dirPath.getAbsolutePath() + File.separator + MdkConstant.ALGS + File.separator + entity.getPyModule().replace(".", File.separator) + File.separator + entity.getPyName() + ".pyd";


            String pyFilePath = dirPath.getAbsolutePath() + File.separator + MdkConstant.ALGS + File.separator + entity.getPyModule().replace(".", File.separator) + File.separator + entity.getFilePath().substring(entity.getFilePath().lastIndexOf("\\"));


            FileUtil.mkParentDirs(pyFilePath);


            FileUtil.copy(entity.getFilePath(), pyFilePath, true);


        }


@@ -387,7 +456,7 @@




        //生成dll文件


        String dllSavePath = dirPath.getAbsolutePath() + File.separator + MdkConstant.LIBS + File.separator + "IAIL.MDK.Mid.Jni.dll";


        createDllFile(dirPath.getAbsolutePath(),cppFilePaths,dllSavePath);


        createDllFile(dirPath.getAbsolutePath(),dllSavePath);


        // 打包历史id


        String historyId = UUID.randomUUID().toString();


        //备份dll，发布时使用


@@ -400,16 +469,16 @@


        }




        //utils + env java文件


        File utilsJavaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "utils" + File.separator + "AlgsUtils.java");


        File utilsJavaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "utils" + File.separator + "AlgsUtils.java");


        FileUtil.mkParentDirs(utilsJavaFile);


        FileUtil.copy(mpkResources + File.separator +"IAILMDK/utils/AlgsUtils.java", utilsJavaFile.getAbsolutePath(), true);


        javaFilePaths.add(utilsJavaFile.getAbsolutePath());


        File envJavaFile = new File(dirPath.getAbsolutePath() + File.separator + "IAILMDK" + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "common" + File.separator + "Environment.java");


        File envJavaFile = new File(dirPath.getAbsolutePath() + File.separator + MdkConstant.IAILMDK + File.separator + "iail" + File.separator + "mdk" + File.separator + "model" + File.separator + "common" + File.separator + "Environment.java");


        FileUtil.mkParentDirs(envJavaFile);


        FileUtil.copy(mpkResources + File.separator +"IAILMDK/common/Environment.java", envJavaFile.getAbsolutePath(), true);


        javaFilePaths.add(envJavaFile.getAbsolutePath());


        // 生成class文件


        createClassFile(javaFilePaths);


        createClassFile(dirPath.getAbsolutePath(),javaFilePaths);


        // 删除java源文件


        deleteJavaFile(javaFilePaths);


        // 打jar包


@@ -465,23 +534,10 @@


        return bytes;


    }




    private void createDllFile(String dirPath, List<String> cppFilePaths, String dllSavePath) {


    private void createDllFile(String dirPath, String dllSavePath) {


        try {


            String paths = cppFilePaths.stream().collect(Collectors.joining(" "));


            String command = "cl /LD " + paths + " /EHsc /o " + dllSavePath + " /link " + "IAIL.MDK.Mid.Windows.lib";


            log.info("执行cmd命令生成dll：" + command);


            ProcessBuilder builder = new ProcessBuilder("cmd", "/c", command);


            builder.directory(new File(dirPath + File.separator + MdkConstant.LIBS));


            Process process = builder.start();


            // 获取命令输出结果


            InputStream inputStream = process.getInputStream();


            BufferedReader reader = new BufferedReader(new InputStreamReader(inputStream, "GBK")); // 设置编码为GBK


            String line;


            while ((line = reader.readLine()) != null) {


                log.info(line);


            }


            // 等待命令执行完成


            process.waitFor();


            String command = "cl /LD *.cpp /EHsc /o " + dllSavePath + " /link " + "IAIL.MDK.Mid.Windows.lib";


            CmdUtils.exec(command,dirPath + File.separator + MdkConstant.C);


        } catch (Exception e) {


            throw new RuntimeException("执行cmd命令生成dll异常",e);


        }


@@ -492,24 +548,73 @@


        Long tenantId = TenantContextHolder.getTenantId();


        // 备份文件 租户隔离


        String mpkTenantBakFilePath = mpkBakFilePath + File.separator + tenantId;




        File dir = new File(mpkTenantBakFilePath);


        if (!dir.exists()) {


            dir.mkdirs();


        File bakDir = new File(mpkTenantBakFilePath);


        if (!bakDir.exists()) {


            bakDir.mkdirs();


        }


        //临时文件夹


        File tempDir = null;


        try {


            tempDir = Files.createTempDirectory("pyFileTmp").toFile();


            log.info("生成临时文件夹，" + tempDir.getAbsolutePath());


        } catch (IOException e) {


            throw new RuntimeException("创建临时文件夹异常",e);


        }


        String fileName = file.getOriginalFilename();


        String fileSuffix = fileName.substring(fileName.lastIndexOf("."));


        File saveFile = new File(dir.getAbsolutePath() + File.separator + UUID.randomUUID() + fileSuffix);


        String pyName = fileName.substring(0, fileName.lastIndexOf("."));


        String pyName_uuTime = pyName + "_" + new Date().getTime();




        saveFile.createNewFile();


        // 保存


        file.transferTo(saveFile);


        File pydBakFile = new File(bakDir.getAbsolutePath() + File.separator + pyName_uuTime + ".pyd");


        try {


            // py文件存入临时文件夹


            File saveFile = new File(tempDir.getAbsolutePath() + File.separator + fileName);


            saveFile.createNewFile();


            file.transferTo(saveFile);




            // 临时文件夹中生成pyd文件


            // 生成setup.py文件


            createSetUpFile(tempDir.getAbsolutePath(),fileName,pyName_uuTime);


            // 执行cmd命令编译pyd文件


            createPydFile(tempDir.getAbsolutePath());




            // 临时文件夹中pyd文件移到dir下，修改为uuid.pyd


            File pydFile = new File(tempDir.getAbsolutePath() + File.separator + pyName_uuTime + MdkConstant.PYD_SUFFIX);


            if (!pydFile.exists()) {


                throw new RuntimeException("编译pyd文件失败！");


            }


            log.info("备份pyd文件，" + pydBakFile.getAbsolutePath());


            FileUtil.copy(pydFile,pydBakFile,true);


        } catch (Exception e) {


            throw new RuntimeException("编译pyd文件异常");


        } finally {


            if (tempDir.exists()) {


                tempDir.delete();


            }


        }




        Map<String, String> result = new HashMap<>(2);


        result.put("filePath", saveFile.getAbsolutePath());


        result.put("fileName", fileName);


        result.put("filePath", pydBakFile.getAbsolutePath());


        result.put("fileName", pyName);




        return result;


    }




    private void createPydFile(String dir) {


        try {


            String command = "python setup.py build_ext --inplace";


            log.info("执行cmd命令编译pyd：" + command);


            CmdUtils.exec(command,dir);


        } catch (Exception e) {


            throw new RuntimeException("执行cmd命令生成dll异常", e);


        }


    }




    private void createSetUpFile(String dir, String pyFileName, String pyName) {


        // 生成setup.py文件


        HashMap<String,Object> map = new HashMap<>();


        map.put("fileName",pyFileName);


        map.put("pyName",pyName);


        GenUtils.drawTemplate("setup.py.vm",map,new File(dir + File.separator + MdkConstant.SETUP_PY));


    }




    @Override


@@ -590,12 +695,10 @@


            StringBuilder sb = new StringBuilder();


            sb.append("jar -cvf");


            sb.append(" ").append(jarSavePath);


            sb.append(" -C ").append(dirPath).append(File.separator).append("IAILMDK").append(File.separator).append(" .");


            log.info("执行cmd命令打jar包：" + sb);


            Process process = RuntimeUtil.exec(sb.toString());


            process.waitFor();


            sb.append(" -C ").append(dirPath).append(File.separator).append(MdkConstant.IAILMDK).append(File.separator).append(" .");


            CmdUtils.exec(sb.toString());


            return jarSavePath;


        } catch (InterruptedException e) {


        } catch (Exception e) {


            throw new RuntimeException("执行cmd命令打jar包异常",e);


        }


    }


@@ -606,17 +709,23 @@


        }


    }




    private void createClassFile(List<String> javaFilePaths){


    private void createClassFile(String dirPath,List<String> javaFilePaths){


        try {


            Set<String> pathSet = javaFilePaths.stream().map(path -> {


                // 替换为相对路径


                path = path.replace(dirPath + File.separator + MdkConstant.IAILMDK, ".");


                // 删除文件名，替换为通配符*


                path = path.substring(0, path.lastIndexOf("\\") + 1) + "*.java";


                return path;


            }).collect(Collectors.toSet());




            StringBuilder sb = new StringBuilder();


            sb.append("javac -encoding utf-8");


            for (String path : javaFilePaths) {


            for (String path : pathSet) {


                sb.append(" ").append(path);


            }


            log.info("执行cmd命令生成class：" + sb);


            Process process = RuntimeUtil.exec(sb.toString());


            process.waitFor();


        } catch (InterruptedException e) {


            CmdUtils.exec(sb.toString(),dirPath + File.separator + MdkConstant.IAILMDK);


        } catch (Exception e) {


            throw new RuntimeException("执行cmd命令生成class异常",e);


        }


    }