通过nodejs将文件上传到mongodb

原文链接:http://www.cnblogs.com/zdkjob/archive/2013/04/27/3046305.html

刚接触nodejs,有点犯晕...

这里要用到一些组件,通过npm安装一下 gridform和gridfs-stream

上传POST到 :http://yourUrl/upload

上传后的文件,可以通过文件名和ID来访问,比如:http://yourUrl/id/xxxxxx  OR http://yourUrl/filename

 

建好下面这两个文件后,运行node test.js启动

//test.js

var
mongo = require('mongodb'), FileServer=require('./fileserver'), Server = require('mongodb').Server; var db = new mongo.Db('test', new Server("127.0.0.1", 27017,{auto_reconnect:true})); var fileServer = FileServer(db); fileServer.start(8888);

 

 

 

//fileserver.js

var
http = require("http"), url = require("url"), mongo = require('mongodb'), path = require("path"), ObjectID = require('mongodb').ObjectID, Grid = require('gridfs-stream'), gridform = require('gridform'), fs = require("fs"), GridStore = require('mongodb').GridStore; function FileServer(db) { if (!(this instanceof FileServer)) return new FileServer(db); this.db = db; } FileServer.prototype.start = function start(listenport) { var me = this; me.db.open(function(err) { if (err) { console.log(err); return; } app.listen(listenport, "127.0.0.1"); console.log("Server running at http://127.0.0.1:" + listenport); }); var app = http.createServer(function(req, res) { var pathname = url.parse(req.url).pathname; //upload//////////////////////////////////////////////////// if (pathname.indexOf('/upload') >= 0 && 'POST' == req.method) { gridform.db = me.db; gridform.mongo = mongo; var outObj; var form = gridform(); form.parse(req, function(err, fields, files) { if (err) { outObj = { "status": 0, "message": err } } else { for (var obj in files) { outObj = { status: 1, data: { name: files[obj].name, id: files[obj].id.toString(), url: 'http://' + req.headers.host + '/id/' + files[obj].id.toString() } }; } } console.log('received upload..'); me.responMsg(res, 200, JSON.stringify(outObj)); }); } else if (pathname != '/') { //download/////////////////////////////////////// var filename, fileId, delPos, idPos, strId; delPos = pathname.indexOf('/delete/'); idPos = pathname.indexOf('/id/'); //通过ID匹配文件 if (idPos >= 0) { strId = pathname.substr(idPos + 4, pathname.length - idPos + 1); filename = { _id: strId }; fileId = ObjectID(strId); } else { filename = pathname.replace('/', ''); } console.log('get file:' + filename); //优先查找磁盘中的文件 var realPath = "files" + pathname; path.exists(realPath, function(exists) { if (exists) { fs.readFile(realPath, "binary", function(err, file) { if (err) { me.responMsg(res, 500, err); } else { var ext = path.extname(realPath); ext = ext ? ext.slice(1) : 'unknown'; var contentType = mime[ext] || "text/plain"; res.writeHead(200, { 'Content-Type': contentType }); res.write(file, "binary"); res.end(); } }); } else { // Verify that the file exists me.exist(me.db, fileId || filename, function(err, result, item) { if (!result) { me.responMsg(res, 404, "文件不存在"); return; } if (delPos >= 0) { GridStore.unlink(me.db, item._id, function(err) { me.responMsg(res, 200, 'success') }); } else { var gfs = Grid(me.db, mongo); var readstream = gfs.createReadStream(filename); var ext = item.filename.split('.'); ext = ext ? ext[ext.length-1] : 'unknown'; var contentType = mime[ext] || "text/plain"; res.setHeader("Content-Type", contentType); readstream.pipe(res); } }); } }); } else { res.setHeader("Content-Type", "text/html"); res.end( '<form method="post" action="/upload" enctype="multipart/form-data">' + '<input type="file" name="text">' + '<input type="submit" value="upload">' + '</form>'); } }); } FileServer.prototype.responMsg = function responMsg(response, status, msg) { response.writeHead(status, { 'Content-Type': 'text/plain;charset=utf-8' }); response.end(msg); }
//覆盖了驱动里原始的exist方法,将找到的item也返回来 FileServer.prototype.exist
= function exist(db, fileIdObject, rootCollection, callback) { var args = Array.prototype.slice.call(arguments, 2); callback = args.pop(); rootCollection = args.length ? args.shift() : null; // Fetch collection var rootCollectionFinal = rootCollection != null ? rootCollection : GridStore.DEFAULT_ROOT_COLLECTION; db.collection(rootCollectionFinal + ".files", function(err, collection) { // Build query var query = (typeof fileIdObject == 'string' || Object.prototype.toString.call(fileIdObject) == '[object RegExp]') ? { 'filename': fileIdObject } : { '_id': fileIdObject }; // Attempt to locate file collection.find(query, function(err, cursor) { cursor.nextObject(function(err, item) { callback(null, item == null ? false : true, item); }); }); }); }; var mime = { "css": "text/css", "gif": "image/gif", "html": "text/html", "ico": "image/x-icon", "jpeg": "image/jpeg", "jpg": "image/jpeg", "js": "text/javascript", "json": "application/json", "pdf": "application/pdf", "png": "image/png", "svg": "image/svg+xml", "swf": "application/x-shockwave-flash", "tiff": "image/tiff", "txt": "text/plain", "wav": "audio/x-wav", "wma": "audio/x-ms-wma", "wmv": "video/x-ms-wmv", "xml": "text/xml", "doc": "application/msword", "docx": "application/vnd.openxmlformats-officedocument.wordprocessingml.document", "xls": "application/vnd.ms-excel", "xlsx": "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet" }; module.exports = exports = FileServer;

 

转载于:https://www.cnblogs.com/zdkjob/archive/2013/04/27/3046305.html

展开阅读全文
博主设置当前文章不允许评论。

没有更多推荐了,返回首页