共计 1942 个字符,预计需要花费 5 分钟才能阅读完成。
- 由于删除目录只能删除空目录 (如果有子文件或文件夹要先删除)
- 目录结构算是典型的二叉树模型, 所以涉及到遍历树结构
- 二叉树遍历 (分为深度和广度, 以及先序, 中序, 后序之分)
- 以下以深度先序解决目录删除
- 在 node 中由于主线程为单线程, 可以采取串行方式和并行方式
深度先序 (串行 回调方式)
const fs = require('fs')
const path = require('path')
function rmdir(filePath, callback) {// 先判断当前 filePath 的类型 ( 文件还是文件夹, 如果是文件直接删除, 如果是文件夹, 去取当前文件夹下的内容, 拿到每一个递归)
fs.stat(filePath, function(err, stat) {if(err) return console.log(err)
if(stat.isFile()) {fs.unlink(filePath, callback)
}else {fs.readdir(filePath, function(err, data) {if(err) return console.log(err)
let dirs = data.map(dir => path.join(filePath, dir))
let index = 0
!(function next() {
// 此处递归删除掉所有子文件 后删除当前 文件夹
if(index === dirs.length) {fs.rmdir(filePath, callback)
}else {rmdir(dirs[index++],next)
}
})()})
}
})
}
rmdir('a', function() {console.log('删除成功')
})
深度先序 (串行 promise 写法)
const fs = require('fs')
const path = require('path')
function rmdirPromise(filePath) {return new Promise((resolve, reject) => {fs.stat(filePath, function(err, stat) {if(err) reject(err)
if(stat.isFile()) {fs.unlink(filePath, function(err) {if(err) reject(err)
resolve()})
}else {fs.readdir(filePath, function(err, dirs) {if(err) reject(err)
dirs = dirs.map(dir => path.join(filePath, dir)) // a/b a/c
let index = 0;
(function next() {if(index === dirs.length) {fs.rmdir(filePath, function(err) {if(err) reject(err)
resolve()})
}else {rmdirPromise(dirs[index++]).then(() => {next()
}, err => {reject(err)
})
}
})()})
}
})
})
}
rmdirPromise('a').then(() => {console.log('删除成功')
})
深度先序 (串行 async await 写法)
// 在 node v10.0.0+ 版本中 fs 模块提供 promise 写法 const fs = require('fs').promises
// 如果在 node 10 之前的版本中可以引入第三方模块 mz const fs = require('mz/fs') 用法一致 https://www.npmjs.com/package/mz
const fs = require('fs').promises
const path = require('path')
async function rmdirAsync(filePath) {let stat = await fs.stat(filePath)
if (stat.isFile()) {await fs.unlink(filePath)
} else {let dirs = await fs.readdir(filePath)
dirs = dirs.map(dir => path.join(filePath, dir))
let index = 0;
(async function next() {if (index === dirs.length) {await fs.rmdir(filePath)
} else {await rmdirAsync(dirs[index++])
await next()}
})()}
}
rmdirAsync('a').then(() => {console.log('删除成功')
}, (err) => {console.log('err', err)
})
正文完