文章出處

  在Node.js中,我們可以使用formidable模塊來輕松地實現文件上傳功能,代碼如下:

var Q = require('q');
var util = require('util');
var fs = require('fs');
var path = require('path');
var moment = require('moment');
var formidable = require('formidable');

var imageUpload = function (){
};

imageUpload.prototype.useFormParseCallback = function(req){
    var deferred = Q.defer();

    var form = new formidable.IncomingForm();
    form.parse(req, deferred.makeNodeResolver());
    return deferred.promise;
};

// 確保目錄存在
// 如果指定的目錄不存在則同步創建(多層次)
imageUpload.prototype.ensureUploadPathExist = function(uploadPath, mode){
    if (!fs.existsSync(uploadPath)){
        var pathtmp;
        uploadPath.split(path.sep).forEach(function(dirname) {
            if (pathtmp) {
                pathtmp = path.join(pathtmp, dirname);
            }
            else {
                pathtmp = dirname;
            }
            if (!fs.existsSync(pathtmp)) {
                if (!fs.mkdirSync(pathtmp, mode)) {
                    return false;
                }
            }
        });
    }

    return true;
};

imageUpload.prototype.uploadImage = function(req){
    var pathName = '/uploadImgs/';
    var uploadPath = path.join(__dirname, '../../public', pathName);

    this.ensureUploadPathExist(uploadPath);

    return this.useFormParseCallback(req).then(function(files){
        var file = files[1].imagefile;
        var fileType = file.type.split('/')[1];
        var newFileName = 'upload_' + moment().format('x') + Math.random().toString().substr(2, 10) + '.' + fileType;
        fs.renameSync(file.path, uploadPath + newFileName);
        return pathName + newFileName;
    });
};

module.exports = imageUpload;

  上述代碼中使用了模塊q來處理Node.js中的回調處理,有關如何使用q可以自己上百度搜索下,這里不再贅述。函數ensureUploadPathExist()可以確保上傳文件的目標目錄是存在的,如果不存在它會逐級創建。文件被上傳在/public/uploadImgs/目錄下,并且按照指定的格式進行了重命名以防止文件被覆蓋,有問題的地方在fs.renameSync()函數的調用。正常情況下如果上傳的目錄在本地,執行fs.renameSync()函數重命名文件是不會有問題的,但如果是跨分區重命名,例如在Linux中使用了NFS或者共享目錄,則此處會報跨分區重命名的權限錯誤。解決辦法是將fs.renameSync()函數替換為下面的代碼:

var readStream = fs.createReadStream(file.path);
var writeStream = fs.createWriteStream(uploadPath + newFileName);
 
var deferred = Q.defer();
util.pump(readStream, writeStream, deferred.makeNodeResolver());
return deferred.promise.then(function() {
    fs.unlinkSync(file.path);
    return pathName + newFileName;
});

  使用上述代碼既可以實現本地文件的上傳,也可以實現跨分區的文件上傳。


文章列表


不含病毒。www.avast.com
arrow
arrow
    全站熱搜
    創作者介紹
    創作者 大師兄 的頭像
    大師兄

    IT工程師數位筆記本

    大師兄 發表在 痞客邦 留言(0) 人氣()