socket.io-stream将文件写入光盘,然后在完成后发出文件路径

时间:2017-02-02 15:08:16

标签: node.js sockets socketstream

当我从客户端向服务器发送文件时,使用socket.io-stream时遇到问题。我将文件路径推送到mongodb,然后将文件保存到文件夹。问题是,当光盘正在写入文件时,文件路径会在写入完成之前返回给客户端,然后img元素页面正确但图像不会在页面上呈现。

下面的代码按预期工作,但我手动等待100毫秒,然后返回到客户端的路径。哪个不保证图像是否太大这100ms就能解决问题。还有其他方法吗?

服务器端:

ss(socket).on('fileUpload', function(stream, data) {
        Chat.findByIdAndUpdate(data.roomId, {
            $push: {
                "messages": {
                    author: data.author,
                    date: Date.now(),
                    isFile: true,
                    extension: data.extension
                }
            }
        },{safe: true, new: true, upsert: true}, (err, message) => {
            if(err){
                console.log(err);
            }

            let msg = message.messages[message.messages.length - 1];
            let filename = path.join(__dirname, 'public/files' , `${msg._id}.${msg.extension}`);
            stream.pipe(fs.createWriteStream(filename));
            setTimeout(() => {
                io.sockets.to(data.roomId).emit('chat-connection', msg);
            },100);
        });
    });

客户端:

function fileUpload(file) {

        let fileExtension = file.name.split('.').pop();
        let stream = ss.createStream();
        // upload a file to the server.
        ss(vm.socket).emit('fileUpload', stream,
            {
                size: file.size,
                extension: fileExtension,
                roomId:vm.chatInstance._id,
                author:$scope.user.fullName,
            });
        ss.createBlobReadStream(file).pipe(stream);
    }

1 个答案:

答案 0 :(得分:2)

您可以在您要写入的信息流中查看finish事件。

stream.on('finish', () => {
    io.sockets.to(data.roomId).emit('chat-connection', msg);
});
stream.pipe(fs.createWriteStream(filename));