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

时间:2013-09-20 05:34:45

标签: node.js sockets asynchronous socket.io fs

我正在尝试使节点服务器能够通过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

1 个答案:

答案 0 :(得分:0)

我创建了一个小工作示例:

<强> app.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);

<强>的index.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>