node.js - 如何递归遍历目录，通过 node.js 中的套接字发送所有文件名？

Question

我正在尝试使节点服务器能够通过 Web 套接字将目录中找到的所有文件名（递归）发送到客户端。这里的想法是目录非常大，我想在读取整个目录之前开始在客户端获取结果。

我想这样的事情应该这样做：

服务器端：

var express = require('express'),
  app = express(),
  server = require('http').createServer(app),
  io = require('socket.io').listen(server),
  fs = require('graceful-fs'),
  rootDirectory = "/some/absolute/path";
  port = 3000,
  recursivelyReadDirectory = function (rootDirectory) {
    // TODO
  },
  sync = io
    .of('/sync')
    .on('connection', function (socket) {
      recursivelyReadDirectory(rootDirectory).on('chunk', function (filename) {
       socket.emit('filename', filename);
      });
    });
server.listen(port);

客户端：

var socket = io.connect('http://localhost:3000/sync');
socket.on('filename', function (filename) {
  console.log(filename);
});

因此，如果我在服务器上的 /some/absolue/path 中有以下内容：

.
|-- a
|   `-- c.txt
`-- b.txt

我希望客户端脚本能够记录

c.txt
b.txt

或者

b.txt
c.txt

Answer 1

我创建了一个小的工作示例：

应用程序.js

var express = require('express'),
    app = express(),
    server = require('http').createServer(app),
    io = require('socket.io').listen(server),
    fs = require('fs'),
    rootDirectory = "c:/tmp",
    port = 3000,
    recursivelyReadDirectory = function (rootDirectory) {
        // TODO
    };

app.use(express.static(__dirname + '/public'));

var walk = function (dir, action, done) {

    // this flag will indicate if an error occured (in this case we don't want to go on walking the tree)
    var dead = false;

    // this flag will store the number of pending async operations
    var pending = 0;

    var fail = function (err) {
        if (!dead) {
            dead = true;
            done(err);
        }
    };

    var checkSuccess = function () {
        if (!dead && pending == 0) {
            done();
        }
    };

    var performAction = function (file, stat) {
        if (!dead) {
            try {
                action(file, stat);
            }
            catch (error) {
                fail(error);
            }
        }
    };

    // this function will recursively explore one directory in the context defined by the variables above
    var dive = function (dir) {
        pending++; // async operation starting after this line
        fs.readdir(dir, function (err, list) {
            if (!dead) { // if we are already dead, we don't do anything
                if (err) {
                    fail(err); // if an error occured, let's fail
                }
                else { // iterate over the files
                    list.forEach(function (file) {
                        if (!dead) { // if we are already dead, we don't do anything
                            var path = dir + "/" + file;
                            pending++; // async operation starting after this line
                            fs.stat(path, function (err, stat) {
                                if (!dead) { // if we are already dead, we don't do anything
                                    if (err) {
                                        fail(err); // if an error occured, let's fail
                                    }
                                    else {
                                        if (stat && stat.isDirectory()) {
                                            dive(path); // it's a directory, let's explore recursively
                                        }
                                        else {
                                            performAction(path, stat); // it's not a directory, just perform the action
                                        }
                                        pending--;
                                        checkSuccess(); // async operation complete
                                    }
                                }
                            });
                        }
                    });
                    pending--;
                    checkSuccess(); // async operation complete
                }
            }
        });
    };

    // start exploration
    dive(dir);
};

io
    .on('connection', function (socket) {
        walk(rootDirectory, function (path, stat) {
            console.log(path);
            socket.emit('filename', path);
        }, function (err) {
            if (err) {
                socket.emit('error', 'Something went wrong');
            } else {
                socket.emit('done');
            }
        });
    });
server.listen(port);

索引.html

<!DOCTYPE html>
<html>
<head>
    <title></title>
</head>
<body>
<script src="/socket.io/socket.io.js"></script>
<script>
    var socket = io.connect('http://localhost');
    socket.on('filename', function (data) {
        console.log(data);
    });
</script>
</body>
</html>