我使用 Socket.io 进行了简单的二进制传输，将文件从客户端传输到服务器。我认为它有效，但我意识到文件的大小不同。在 writableStream.write 失败时，我附加了 Drain 事件处理程序以保持等待，直到它可以被重写并继续写入，但是每次发生 Drain 事件时，文件的大小都会增加触发 Drain 事件的次数，每 10240 字节大小我为每个块传输设置。
在我在这里写代码之前，我需要解释一下代码流程：
- 客户端请求上传文件
- 服务器创建空文件（创建可写流）并授权传输
- 客户端传输数据（块）直到结束
- 服务器用可写流写入块
- 客户端在全部发送后结束传输
- 服务器关闭可写流。
- 完毕！
这是服务器端代码：
var writeStream = null;
var fileSize = 0;
var wrote = 0;
socket.on('clientRequestFileTransfer', (fileInfo) => {
console.log(`Client request file transfer: ${fileInfo.name}(${fileInfo.size})`);
fileSize = fileInfo.size;
wrote = 0;
writeStream = fs.createWriteStream(__dirname + '/' + fileInfo.name);
writeStream.on('close', () => {
console.log('Write stream ended.');
});
console.log('File created.');
socket.emit('serverGrantFileTransfer');
});
socket.on('clientSentChunk', (chunk) => {
function write() {
let writeDone = writeStream.write(chunk);
if(!writeDone) {
console.log('Back pressure!');
return writeStream.once('drain', write);
}
else {
wrote += chunk.length;
console.log(`Wrote chunks: ${chunk.length} / ${wrote} / ${fileSize}`);
socket.emit('serverRequestContinue');
}
}
write();
});
socket.on('clientFinishTransmission', () => {
writeStream.end();
console.log('Transmission complete!');
});
它是客户端（添加了用于读取二进制文件的代码）：
var fileEl = document.getElementById('file');
fileEl.onchange = function() {
var file = fileEl.files[0];
if(!file) return;
var socket = io('http://localhost:3000');
socket.on('connect', function() {
var fileReader = new FileReader();
fileReader.onloadend = function() {
var bin = fileReader.result;
var chunkSize = 10240;
var sent = 0;
// make server knows the name and size of the file
socket.once('serverGrantFileTransfer', () => {
function beginTransfer() {
if(sent >= bin.byteLength) {
console.log('Transmission complete!');
socket.emit('clientFinishTransmission');
return;
}
var chunk = bin.slice(sent, sent + chunkSize);
socket.once('serverRequestContinue', beginTransfer);
socket.emit('clientSentChunk', chunk);
sent += chunk.byteLength;
console.log('Sent: ' + sent);
}
beginTransfer();
});
socket.emit('clientRequestFileTransfer', {
name: file.name,
size: file.size
});
};
fileReader.readAsArrayBuffer(file);
});
};
我用 4,162,611 字节大小的文件测试了这段代码，它有 1 次写入失败（1 次背压）。上传后，我检查了创建文件的大小，它是 4,172,851 字节，比原始文件大 10240 字节，它是块大小（10240）。
有时写入失败 2 次，大小比原来的大 20480 字节，是我发送的块大小的两倍。
我仔细检查了我的背压代码，但对我来说似乎没有错。我正在使用 Node v6.2.2 和 Socket.io v1.6.0，通过 Chrome 浏览器测试。有什么我错过的吗？还是我误解了背压？任何建议将不胜感激。
更新
看起来当背压发生时，它写了两次相同的数据（正如我在评论中所说）。所以我修改了这样的代码：
socket.on('clientSentChunk', (chunk) => {
function write() {
var writeDone = writeStream.write(chunk);
wrote += chunk.length;
if(!writeDone) {
console.log('**************** Back pressure ****************');
// writeStream.once('drain', write);
// no rewrite, just continue transmission
writeStream.once('drain', () => socket.emit('serverRequestContinue'));
}
else {
console.log(`Wrote chunks: ${chunk.length} / ${wrote} / ${fileSize}`);
socket.emit('serverRequestContinue');
}
}
write();
});
有效。我很困惑，因为当可写流无法写入时，它不会将数据写入流中，但实际上不是。有人知道吗？