文章出處
文章列表
在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; });
使用上述代碼既可以實現本地文件的上傳,也可以實現跨分區的文件上傳。
文章列表
全站熱搜