由于删除目录只能删除空目录(如果有子文件或文件夹要先删除)
目录结构算是典型的二叉树模型,所以涉及到遍历树结构
二叉树遍历(分为深度和广度,以及先序,中序,后序之分)
以下以深度先序解决目录删除
在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) })
深度先序 (并行)
深度先序 (并行 回调写法)
const fs = require('fs').promises const path = require('path') function rmdir(filePath, callback) { 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, dirs) { if(err) return console.log(err) // 此处要添加dirs.length的验证,不然如果length为0 后面的forEach不执行, 就删不掉当前目录了, 也执行不了callback if(dirs.length === 0) { fs.rmdir(filePath, callback) } dirs = dirs.map(dir => path.join(filePath, dir)) // 通过计数的方式来判断是否子目录都删除了 let index = 0 function done() { if(++index === dirs.length) { fs.rmdir(filePath, callback) } } // 何为并行? a下有两个目录b, c,那么同时将b, c的删除都推到event Loop中,用for循环实现 dirs.forEach(dir => { // 通过done回调的方式来控制js执行流程(LazyMan问题也是这么解决的) rmdir(dir, done) }); }) } }) } rmdir('a', function() { console.log('删除成功') })
深度先行 (并行promise写法)
const fs = reqire('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)) dirs = dirs.map(dir => rmdirPromise(dir)) Promise.all(dirs).then(() => { fs.rmdir(filePath, resolve) }) }) } }) }) } rmdirPromise('a').then(() => { console.log('删除成功') })
深度先序 (并行 async + await写法)