node进阶之用流实现上传文件

内容：

1.文件上传基础

2.node文件处理机制

3.用流实现文件上传

1.文件上传基础

前端代码：

<form action="localhost:8080/" method="post" enctype="multipart/form-data">
    <input type="file" name="f1">
    <input type="submit" value="上传文件">
</form>

注意:
    上传文件时表单中的enctype="multipart/form-data"必须要写
    input(file)必须要有name

后端代码：

const http = require('http');
const uuid = require('uuid/v4');
const fs = require('fs')

let server_post = http.createServer((req, res) => {
    let arr = [];

    req.on('data', data => {
        arr.push(data);
    });
    req.on('end', () => {
        let data = Buffer.concat(arr);
        // console.log(data)

        //data
        //解析二进制文件上传数据
        let post = {};
        let files = {};
        if (req.headers['content-type']) {
            let str = req.headers['content-type'].split('; ')[1];
            if (str) {
                let boundary = '--' + str.split('=')[1];

                //1.用"分隔符切分整个数据"
                let arr = (data.toString()).split(boundary);

                //2.丢弃头尾两个数据
                arr.shift();
                arr.pop();

                //3.丢弃掉每个数据头尾的"
"
                arr = arr.map(buffer => buffer.slice(2, buffer.length - 2));

                //4.每个数据在第一个"

"处切成两半
                arr.forEach(buffer => {
                    let n = buffer.indexOf('

');

                    let disposition = buffer.slice(0, n);
                    let content = buffer.slice(n + 4);

                    disposition = disposition.toString();

                    if (disposition.indexOf('
') === -1) {
                        //普通数据
                        //Content-Disposition: form-data; name="user"
                        content = content.toString();

                        let name = disposition.split('; ')[1].split('=')[1];
                        name = name.substring(1, name.length - 1);

                        post[name] = content;
                    } else {
                        //文件数据
                        /*Content-Disposition: form-data; name="f1"; filename="a.txt"

                        Content-Type: text/plain*/
                        let [line1, line2] = disposition.split('
');
                        let [, name, filename] = line1.split('; ');
                        let type = line2.split(': ')[1];

                        name = name.split('=')[1];
                        name = name.substring(1, name.length - 1);
                        filename = filename.split('=')[1];
                        filename = filename.substring(1, filename.length - 1);

                        let path = `upload/${uuid().replace(/-/g, '')}`;

                        fs.writeFile(path, content, err => {
                            if (err) {
                                console.log('文件写入失败', err);
                            } else {
                                files[name] = {filename, path, type};
                                console.log(files);
                            }
                        });
                    }
                });


                //5.完成
                console.log(post);
            }
        }


        res.end();
    });
});
server_post.listen(8080);

2.node文件处理机制

node文件上传从根本上来说就两种方法：

(1)最基础原始的方法

使用fs中的readFile和writeFile实现(读取完上传的文件后保存)

这样做有弊端：

• 只能等到所有数据都到达了才开始处理
• readFile先把所有数据全读到内存中，然后回调：
• 1.极其占用内存
• 2.资源利用极其不充分

(2)更好的方法

使用流，收到一部分数据就直接解析一部分，实例见后面的文件上传实例

3.用流实现文件上传

(1)流

三种流：

• 读取流  -->  fs.createReadStream、req
• 写入流  -->  fs.createWriteStream、res
• 读写流  -->  压缩、加密

(2)流实现读写文件

const fs = require('fs')

let rs = fs.createReadStream('1.txt')       // 读取流
let ws = fs.createWriteStream('2.txt')      // 写入流

rs.pipe(ws)

// 异常处理
rs.on('error', function (error) {
    console.log('读取失败!')
})

// 读取完成 及 写入完成
rs.on('end', function () {
    console.log('读取完成!')
})

ws.on('finish', function () {
    console.log('写入完成!')
})

注：1.txt应该在同级目录下

(3)用流实现上传文件核心代码

/**
 * [saveFileWithStream description]
 * @param {String} filePath [文件路径]
 * @param {Buffer} readData [Buffer 数据]
 */
static saveFile(filePath, fileData) {
 return new Promise((resolve, reject) => {
  // 块方式写入文件
  const wstream = fs.createWriteStream(filePath);
 
  wstream.on('open', () => {
   const blockSize = 128;
   const nbBlocks = Math.ceil(fileData.length / (blockSize));
   for (let i = 0; i < nbBlocks; i += 1) {
    const currentBlock = fileData.slice(
     blockSize * i,
     Math.min(blockSize * (i + 1), fileData.length),
    );
    wstream.write(currentBlock);
   }
 
   wstream.end();
  });
  wstream.on('error', (err) => { reject(err); });
  wstream.on('finish', () => { resolve(true); });
 });
}

// 实际调用的时候，如下:
try {
 await saveFileWithStream(filePath, fileData); // 这里的fileData是Buffer类型
} catch (err) {
 console.log(err.stack);
}