本文实例讲述了node.js express框架实现文件上传与下载功能。分享给大家供大家参考,具体如下:
背景
昨天吉视传媒的客户对IPS信息发布系统又提了一个新需求,就是发布端发送消息时需要支持附件的上传,而接收端可以对发布端上传的附件进行下载;接收端回复消息时也需要支持上传附件,发布端可以对所有接收端上传的附件进行打包下载。
功能实现
- 前台部分
前台使用webUploader插件即可,这是百度开发的一款文件上传组件,具体使用查看它的API即可。这个项目之前开发的时候前台使用了angular.js。
$scope.fileName = ""; //创建上传附件的对象 var $list = $("#thelist"); var uploader = WebUploader.create({ // 选完文件后,是否自动上传。 auto: false, // swf文件路径 swf: '../../../lib/webUploader/Uploader.swf', // 文件接收服务端。 server: '/publishUploadFile', // 内部根据当前运行是创建,可能是input元素,也可能是flash. pick : { id : '#filePicker', //只能选择一个文件上传 multiple: false }, // pick :'#filePicker', method: 'POST', }); uploader.on('fileQueued', function (file) { $scope.fileName = file.name; $list.html(""); $list.html(file.name); });
当用户选择文件的时候我创建了文件上传的对象,而在用户真正发送消息的时候我添加了相应的参数并将附件真正的上传上去,符合我这个项目的业务逻辑。
if($scope.fileName){ //添加参数 uploader.options.formData.fileId = fileId; uploader.options.formData.fileName = $scope.fileName; uploader.upload(); }
- 后台部分
路由就不详细说明了,主要注意的是下载的接口我都是使用的get请求,这样前台在请求的时候直接新打开一个窗口拼接了相应的参数就能下载文件了。下面贴一下action层的代码:
//发布端上传附件 exports.publishUploadFile = function (req, res) { messageMng.publishUploadFile(req, function (err, datas) { res.json(datas); }); }; //下载发布端上传的附件 exports.exportPublishFile = function (req, res) { messageMng.exportPublishFile(req, function (err, datas) { if (err) { res.set({ "Content-Disposition": "attachment;filename=" + encodeURI("error.txt") }); res.write(err.message); res.end(); } else { res.download(datas.path, encodeURI(datas.name)); } }); }; //接收端上传附件 exports.uploadFile = function (req, res) { messageMng.uploadFile(req, function (err, datas) { res.json(datas); }); }; //发布端导出附件 exports.exportFile = function (req, res) { messageMng.exportFile(req, function (err, datas) { if (err) { res.set({ "Content-Disposition": "attachment;filename=" + encodeURI("error.txt") }); res.write(err.message); res.end(); } else { //第一种方式 下载完的zip解压报错 // res.download(datas.path, datas.name + ".zip"); //第二种方式 // var path="D:/maven介绍.ppt"; var f = fs.createReadStream(datas.path); res.writeHead(200, { 'Content-Type': 'application/force-download', 'Content-Disposition': 'attachment; filename='+ encodeURI(datas.name) + '.zip' }); f.pipe(res); } }); };
这里着重说一下下载zip时使用download下载完的压缩包解压会报错,使用第二种方法完美解决。
然后是service层的代码:
/** * 发布端上传附件 * @param req * @param fn */ MessageManager.prototype.publishUploadFile = function (req, fn) { try { //消息ID var fileId = req.body.fileId; var file = req.file; //文件上传的目录 var uploadFolder = path.join(__dirname, '../../upload/publishUploadFile/' + fileId); //判断文件夹是否存在 不存在则创建 toolUtil.mkdirSync(uploadFolder); //将上传的文件从临时目录拷贝到指定的目录下 var fileReadStream = fs.createReadStream(file.path); var fileWriteStream = fs.createWriteStream(uploadFolder + "/" + file.originalname); fileReadStream.pipe(fileWriteStream); fileWriteStream.on('close', function () { // 删除临时目录下面的文件 toolUtil.emptyDir(file.destination); }); fn(null, {"data": "", "message": "上传成功", "error_code": 200}); } catch (e) { fn(e, {"data": "", "message": "上传失败", "error_code": e.message}); } }; /** * 下载发布端上传的附件 * @param req * @param fn */ MessageManager.prototype.exportPublishFile = function (req, fn) { try { //附件ID var id = req.query.id; //附件名称或标题 var name = req.query.name; if (id && name) { //名称过长的话,截取前25个字符 if (name.length > 25) { name = name.substr(0, 24); } //将要压缩得文件夹路径 var filePath = path.join(__dirname, '../../upload/publishUploadFile/' + id + '/' + name); if (!fs.existsSync(filePath)) { fn(new Error("没有附件!"), null); } else { fn(null, {"name": name, "path": filePath}); } } else { fn(new Error("id或name不能为空"), null); } } catch (e) { fn(new Error(e.message), null); } }; /** * 接收端上传附件 * @param req * @param fn */ MessageManager.prototype.uploadFile = function (req, fn) { try { //消息ID var msgId = req.body.msgId; //消息发送的时间 var msgSendTime = req.body.msgSendTime.slice(0, 10); //消息的标题 var title = req.body.title; var replyId = req.body.replyId; var replyName = req.body.replyName; var file = req.file; //文件上传的目录 var uploadFolder = path.join(__dirname, '../../upload/messages/' + msgId + '/' + replyName); //判断文件夹是否存在 不存在则创建 toolUtil.mkdirSync(uploadFolder); //组装文件的名称 原名称+消息发送时间 var index = file.originalname.lastIndexOf("."); var fileName = file.originalname.substr(0, index) + '-' + msgSendTime + ""; var suffix = file.originalname.substr(index, file.originalname.length - 1); //将上传的文件从临时目录拷贝到指定的目录下 var fileReadStream = fs.createReadStream(file.path); var fileWriteStream = fs.createWriteStream(uploadFolder + "/" + fileName + "." + suffix); fileReadStream.pipe(fileWriteStream); fileWriteStream.on('close', function () { //删除临时目录下面的文件 toolUtil.emptyDir(file.destination); }); fn(null, {"data": "", "message": "上传成功", "error_code": 200}); } catch (e) { fn(e, {"data": "", "message": "上传失败", "error_code": e.message}); } }; /** * 导出消息的附件文件 * @param req * @param fn */ MessageManager.prototype.exportFile = function (req, fn) { try { //消息ID var id = req.query.id; //消息名称或标题 var name = req.query.name; if (id && name) { //名称过长的话,截取前25个字符 if (name.length > 25) { name = name.substr(0, 24); } //将要压缩得文件夹路径 var messagePath = path.join(__dirname, '../../upload/messages/' + id); if (!fs.existsSync(messagePath)) { fn(new Error("没有附件!"), null); } else { //生成得临时zip文件目录 var zipPath = path.join(__dirname, '../../upload/temp.zip'); var archive = archiver('zip', { // Sets the compression level. zlib: {level: 9} }); //创建临时zip文件 var output = fs.createWriteStream(zipPath); archive.pipe(output); //设置需要压缩得文件夹目录 以及替换得名称 archive.directory(messagePath, name); archive.finalize(); archive.on('end', function (err) { fn(null, {"name": name, "path": zipPath}); }); archive.on('error', function (err) { fn(new Error("压缩文件异常"), null); }); } } else { fn(new Error("id或name不能为空"), null); } } catch (e) { fn(new Error(e.message), null); } };
最后是提出的公共方法toolUtil的代码,这个单独做为一个js文件维护。
const path = require('path'); const fs = require('fs'); /** * 创建目录 * @param dirpath */ exports.mkdirSync = function (dirpath){ if (!fs.existsSync(dirpath)) { var pathtmp; dirpath.split(path.sep).forEach(function(dirname) { if (pathtmp) { pathtmp = path.join(pathtmp, dirname); } else { pathtmp = dirname; } if (!fs.existsSync(pathtmp)) { fs.mkdirSync(pathtmp); } }); } }; //删除所有的文件(将所有文件夹置空) exports.emptyDir = function(dirpath){ var self = this; //读取该文件夹 var files = fs.readdirSync(dirpath); files.forEach(function(file){ var filePath = dirpath + '/' + file; var stats = fs.statSync(filePath); if(stats.isDirectory()){ self.emptyDir(filePath); }else{ fs.unlinkSync(filePath); } }); };
希望本文所述对大家node.js程序设计有所帮助。
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
P70系列延期,华为新旗舰将在下月发布
3月20日消息,近期博主@数码闲聊站 透露,原定三月份发布的华为新旗舰P70系列延期发布,预计4月份上市。
而博主@定焦数码 爆料,华为的P70系列在定位上已经超过了Mate60,成为了重要的旗舰系列之一。它肩负着重返影像领域顶尖的使命。那么这次P70会带来哪些令人惊艳的创新呢?
根据目前爆料的消息来看,华为P70系列将推出三个版本,其中P70和P70 Pro采用了三角形的摄像头模组设计,而P70 Art则采用了与上一代P60 Art相似的不规则形状设计。这样的外观是否好看见仁见智,但辨识度绝对拉满。