潘志宝
2024-10-15 856d69ce678173e0a8e165bcb0135037d20b48ee
iailab-module-model/iailab-module-model-biz/src/main/java/com/iailab/module/model/mpk/service/impl/MpkFileServiceImpl.java
@@ -130,6 +130,12 @@
        entity.setCreateDate(new Date());
        insert(entity);
        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
@@ -141,6 +147,12 @@
        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