内容:
1.文件上传基础
2.node文件处理机制
3.用流实现文件上传
1.文件上传基础
前端代码:
<form action="localhost:8080/" method="post" enctype="multipart/form-data">
<input type="text" name="user">
<input type="password" name="pass">
<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.丢弃掉每个数据头尾的"\r\n"
arr = arr.map(buffer => buffer.slice(2, buffer.length - 2));
//4.每个数据在第一个"\r\n\r\n"处切成两半
arr.forEach(buffer => {
let n = buffer.indexOf('\r\n\r\n');
let disposition = buffer.slice(0, n);
let content = buffer.slice(n + 4);
disposition = disposition.toString();
if (disposition.indexOf('\r\n') === -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"\r\n
Content-Type: text/plain*/
let [line1, line2] = disposition.split('\r\n');
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);
}