Nodejs 遍历目录树组建内存tree

最新推荐文章于 2022-03-17 14:13:22 发布

原创最新推荐文章于 2022-03-17 14:13:22 发布 · 2.6k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#nodejs #hash #visitor #tree #遍历树

Nodejs 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一个Node.js后端项目的需求，通过遍历硬盘上的所有文件，计算文件的hash值，并利用Visitor模式和Tree模型实现这一功能。Node.js的Stream在处理大量文件时表现出优越的性能。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最近好久都没写博客了，主要是没干iOS了，现在在写后端，用的Nodejs，加上最早java的功底，以及对javascript的了解，目前没什么问题+

说说最近的项目中遇到的一个小需求。

把硬盘所有文件爬一遍并计算hash，不过话说回来。nodejs的Stream是真的好用。

为什么有这个需求，不讲了。实现原理也就是visitor Pattern. 加tree model

以下是代码。

const fs = require('fs')
const path = require('path')

const uuid = require('uuid')

class Node {
  constructor(name, type, parent){
    this.name = name
    this.parent = parent
    this.child = []
    this.type = type
    this.id = uuid.v4()
  }
}

//这就是一个 visitor
function visit(dir, dirContext, func, done) {
  fs.readdir(dir, (err, entries) => {
    if (err || entries.length === 0) return done()
    let count = entries.length
    entries.forEach(entry => {
      func(dir, dirContext, entry, (entryContext) => {
        if (entryContext) {
          visit(path.join(dir, entry), entryContext, func, () => {
            count--
            if (count === 0) done()
          })
        }
        else {
          count --
          if (count === 0) done()
        }
      })
    })
  })
}

let count = 0
let map = new Map()
let u = false
const clean = (dir, dirContext, entry, callback) => {
  console.log(count++)
  let fpath = path.join(dir, entry)

  fs.lstat(fpath, (err, stats) => {
    if(err) return callback()
    if(stats.isFile()){
      let node = new Node(entry, 'file')
      map.set(node.id, node)
      dirContext.child.push(node)
      callback() 
    }else if(stats.isDirectory()){
      let node = new Node(entry, 'folder')
      map.set(node.id, node)
      dirContext.child.push(node)
      callback(node)
    }else{
      let node = new Node(entry, 'unknown')
      u = node.id
      map.set(node.id, node)
      dirContext.child.push(node)
      callback(node)
    }
  })
}

let n = new Node('usr', 'folder')
map.set(n.id, n)

visit('/usr', n, clean, ()=> {
  fs.writeFile('test.json', JSON.stringify(n, null, ' '))
  console.log('----------------------------',map.size)
  console.log(map.get(n.id))
})

具体实现就是这些了。里面没有包括hashworker的代码