如何避免重复使用forEach语句?Node.js文件整理代码优化求助
代码优化建议与实现方案
优化后的完整代码
const fs = require('fs').promises; const path = require('path'); const rootDir = 'G:\\google drive'; const targetFolders = ['A', 'B', 'C', 'D', 'E', 'F']; const completeDir = path.join(rootDir, 'complete'); // 创建目标文件夹 async function createTargetFolders() { for (const folder of targetFolders) { const folderPath = path.join(completeDir, folder); await fs.mkdir(folderPath, { recursive: true }); } } // 移动匹配的文件 async function moveMatchingFiles() { try { const fileList = await fs.readdir(rootDir); for (const file of fileList) { const filePath = path.join(rootDir, file); // 跳过文件夹,只处理文件 const stats = await fs.stat(filePath); if (!stats.isFile()) continue; // 找到第一个匹配的目标文件夹,找到即停止遍历 const targetFolder = targetFolders.find(folder => file.startsWith(folder)); if (!targetFolder) continue; const destPath = path.join(completeDir, targetFolder, file); await fs.rename(filePath, destPath); console.log(`${file} 移动到 ${targetFolder} 完成`); } } catch (err) { console.error('操作出错:', err); } } // 执行主逻辑 async function main() { await createTargetFolders(); await moveMatchingFiles(); } main();
核心优化点
- 路径处理标准化:用
path.join替代手动字符串拼接,自动适配不同系统的路径分隔符,避免因分隔符写错导致的路径错误,代码更具跨平台性。 - 减少无效循环:原代码对每个文件遍历所有文件夹,改用
Array.find()找到第一个匹配的文件夹后立即停止遍历,降低不必要的循环次数,提升执行效率。 - 异步代码扁平化:改用
fs.promises结合async/await,彻底摆脱回调嵌套,代码逻辑更线性,可读性大幅提升。 - 简化文件夹创建:利用
mkdir的recursive: true参数,无需提前判断文件夹是否存在,不存在则自动创建,存在则跳过,省去冗余的判断逻辑。 - 完善错误处理:用
try/catch统一捕获异步操作错误,避免原代码中throw err导致的程序直接崩溃,同时能清晰输出错误信息。 - 增加文件类型校验:新增
fs.stat()检查是否为文件,避免尝试移动文件夹引发的错误。 - 语义化变量命名:将
setFolder改为targetFolders,dir改为rootDir,变量含义更直观,代码可读性更强。
内容的提问来源于stack exchange,提问作者writingdeveloper
相关产品推荐
相关产品推荐

