Rumah > Artikel > hujung hadapan web > Node.js开发教程之基于OnceIO框架实现文件上传和验证功能
OnceIO 是 OnceDoc 企业内容(网盘)的底层Web框架,它可以实现模板文件、静态文件的全缓存,运行起来完全不需要I/O操作,并且支持客户端缓存优化,GZIP压缩等(只在第一次压缩),拥有非常好的性能,为您节约服务器成本。它的模块化功能,可以让你的Web进行分布式存储,即一个扩展包里即包含前端、后端和数据库定义,只需通过添加/删除目录的方式就可实现功能删减,实现真正的模块化扩展。这里是介绍如何使用OnceIO的一系列文章。
在这一章节中,我们将为大家演示如何使用 OnceIO 实现文件上传功能。
在网页文件中构建表单
以一个只有文件上传功能的简单网页 file.html 为例:
<!DOCTYPE html> <html> <body> <form method="post" enctype="multipart/form-data" action="/file/upload"> <input type="file" name="file" /><br> <input type="submit" value="Upload" /> </form> </body> </html>
浏览器显示效果是这样的:
点击空白长条或“浏览…”按钮可以打开文件浏览窗口选择需要上传的文件:
建立服务器接收文件逻辑
服务器文件 websvr.js 代码是这样的:
var fs = require('fs') var path = require('path') var onceio = require('../onceio/onceio') var app = onceio() app.get('/', function(req, res){ res.render('file.html') }) app.file('/file/upload', function(req, res) { var fileInfo = req.files.file || {} fs.link(fileInfo.path, path.join('./fileStore', fileInfo.name)) res.send('File Uploaded Successfully') }).before(function(req, res) { var contentLength = req.headers['content-length'] || 0 if (contentLength > 1048576) { res.send({ error: 'Error: File Size Limit (1 MB) Exceeded' }) } else { return true } })
var fs = require('fs') 和 var path = require('path') 分别导入了 Node.js 提供的用于操作文件的文件系统(fs)模块和用于处理文件路径的 path 模块。
app.file(path, callback).before(callback) 相当于 app.use(path, callback, {file: true}).before(callback) ,是一个处理上传的文件的中间件。
文件被上传后,它的大小、存放地址、名称、格式和修改时间五项信息会被放在 req.files 的 file 属性里(名称是 type 为 'file' 的 input 标签中 name 的值),它的尺寸信息会被放在 req.headers 的 content-length 属性里。
before函数
before是OnceIO与其它Web框架的主要区别之一。它可以在文件接收之前就对文件进行一些基本验证,如大小、类型等,以求获得最侍性能。return true 表示验证通并开始接收文件,否则就关闭连接,取消上传。在 before 中,req.session对象是不可用的,因为session可能存在文件或数据库redis中,取得session是一个异步过程需要时间。而before函数需要立刻对文件合法性做出判断。
在这个例子中before回调函数根据 req.headers 中的 content-length 判断上传的文件是否超出了尺寸限制(开发人员可以通过修改 if 语句中的常数改变文件上传尺寸上限,content-length 单位为 byte,1024 * 1024 即代表 1 MB),如果超出了,文件不会被上传,服务器返回错误信息;如果没有超出,函数返回值为 true,服务器继续执行 app.file 中的回调函数,将文件从临时地址转移到指定存储地址,文件上传到这里就完成了。
解决文件重名问题
我们目前的服务器程序是无法解决文件重名问题的。如果用户上传了重名的文件,服务器将会返回文件已经存在的错误。为了解决这个问题,我们可以在文件的主文件名和拓展名之间加入一个时间戳,进行这个处理的函数代码如下:
var timestampName = function(fileName){ // get filename extension var extName = path.extname(fileName) // get base name of the file var baseName = path.basename(fileName, extName) // insert timestamp between base name and filename extension // the plus sign ('+') before new Date() converts it into a number return baseName + +new Date() + extName }
再把 fs.link 语句里的 fileInfo.name 替换为 timestampName(fileInfo.name):
fs.link(fileInfo.path, path.join('./fileStore', timestampName(fileInfo.name)))
改进后的服务器程序就能允许用户上传重名文件了,以上传 5 次名为 'cache_workflow.png' 的文件为例,服务器的文件存储地址中会出现 5 个名称都以 'cache_workflow' 开头但时间戳不同的文件: