Node.js：上傳檔案，服務端如何獲取檔案上傳進度

程式猿小卡發表於2018-02-05
Node.js服務端

內容概述

multer是常用的Express檔案上傳中介軟體。服務端如何獲取檔案上傳的進度，是使用的過程中，很常見的一個問題。在SF上也有同學問了類似問題《nodejs multer有沒有檢視檔案上傳進度的方法？》。稍微回答了下，這裡順便整理出來，有同樣疑問的同學可以參考。

下文主要介紹如何利用progress-stream獲取檔案上傳進度，以及該元件使用過程中的注意事項。

利用progress-stream獲取檔案上傳進度

如果只是想在服務端獲取上傳進度，可以試下如下程式碼。注意，這個模組跟Express、multer並不是強繫結關係，可以獨立使用。

var fs = require('fs');
var express = require('express');
var multer  = require('multer');
var progressStream = require('progress-stream');

var app = express();
var upload = multer({ dest: 'upload/' });

app.post('/upload', function (req, res, next) {
    // 建立progress stream的例項
    var progress = progressStream({length: '0'}); // 注意這裡 length 設定為 '0'
    req.pipe(progress);
    progress.headers = req.headers;
	
    // 獲取上傳檔案的真實長度（針對 multipart)
    progress.on('length', function nowIKnowMyLength (actualLength) {
    	console.log('actualLength: %s', actualLength);
    	progress.setLength(actualLength);
    });

	// 獲取上傳進度
    progress.on('progress', function (obj) {		
    	console.log('progress: %s', obj.percentage);
    });

    // 實際上傳檔案
    upload.single('logo')(progress, res, next);
});

app.post('/upload', function (req, res, next) {
	res.send({ret_code: '0'});
});

app.get('/form', function(req, res, next){
    var form = fs.readFileSync('./form.html', {encoding: 'utf8'});
    res.send(form);
});

app.listen(3000);
複製程式碼

如何獲取上傳檔案的真實大小

multipart型別，需要監聽length來獲取檔案真實大小。（官方文件裡是通過conviction事件，其實是有問題的）

// 獲取上傳檔案的真實長度（針對 multipart)
progress.on('length', function nowIKnowMyLength (actualLength) {
    console.log('actualLength: %s', actualLength);
    progress.setLength(actualLength);
});
複製程式碼

3、關於progress-stream獲取真實檔案大小的bug？

針對multipart檔案上傳，progress-stream 例項子初始化時，引數length需要傳遞非數值型別，不然你獲取到的進度要一直是0，最後就直接跳到100。

至於為什麼會這樣，應該是 progress-steram 模組的bug，看下模組的原始碼。當length是number型別時，程式碼直接跳過，因此你length一直被認為是0。

tr.on('pipe', function(stream) {
    if (typeof length === 'number') return;
    // Support http module
    if (stream.readable && !stream.writable && stream.headers) {
    	return onlength(parseInt(stream.headers['content-length'] || 0));
    }
    
    // Support streams with a length property
    if (typeof stream.length === 'number') {
    	return onlength(stream.length);
    }
    
    // Support request module
    stream.on('response', function(res) {
    	if (!res || !res.headers) return;
    	if (res.headers['content-encoding'] === 'gzip') return;
    	if (res.headers['content-length']) {
    		return onlength(parseInt(res.headers['content-length']));
    	}
    });
});
複製程式碼

參考連結

github.com/expressjs/m…

github.com/freeall/pro…

相關文章