diff --git a/wwjcloud-nest-v1/tools/java-to-nestjs-migration/generators/service-generator.js b/wwjcloud-nest-v1/tools/java-to-nestjs-migration/generators/service-generator.js index 07f124e4..38ff048c 100644 --- a/wwjcloud-nest-v1/tools/java-to-nestjs-migration/generators/service-generator.js +++ b/wwjcloud-nest-v1/tools/java-to-nestjs-migration/generators/service-generator.js @@ -181,23 +181,16 @@ ${methods} // ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ const converterImports = this.methodConverter.analyzeImports(methodsCode); - converterImports.forEach(importItem => { - if (importItem === 'JsonUtils' || importItem === 'CommonUtils' || importItem === 'StringUtils') { - imports.boot.add(importItem); - } else if (importItem === 'AppConfigService') { - imports.boot.add('AppConfigService'); - } else if (importItem === 'nestjs:BadRequestException') { - imports.nestjs.add('BadRequestException'); - } else if (importItem === 'nestjs:UnauthorizedException') { - imports.nestjs.add('UnauthorizedException'); - } else if (importItem === 'node:fs') { - imports.nodeModules.add('fs'); - } else if (importItem === 'node:path') { - imports.nodeModules.add('path'); - } else if (importItem === 'boot:RequestContextService') { - imports.boot.add('RequestContextService'); - } - }); + // 合并转换器返回的imports(返回值是对象 {nestjs: [], boot: [], nodeModules: []}) + if (converterImports.nestjs) { + converterImports.nestjs.forEach(item => imports.nestjs.add(item)); + } + if (converterImports.boot) { + converterImports.boot.forEach(item => imports.boot.add(item)); + } + if (converterImports.nodeModules) { + converterImports.nodeModules.forEach(item => imports.nodeModules.add(item)); + } // ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━ // 【兼容旧的检测逻辑(备用)】