zxy-comm/lib/express.js

var express = require("express");
var app = express();
var autoroute = require("express-autoroute");
var http = require("http");
var bodyParser = require('body-parser');
var methodOverride = require('method-override');
var path = require("path");
var zxy = require('zxy-comm');
var multer = require('multer');
var uploadPath = path.join(__dirname, '../../../uploads');
var upload = multer({ dest: uploadPath });
// var upload = multer();
var _ = require('underscore');
var Promise = require('bluebird');
var fs = require('fs');
var multiparty = require('multiparty');

//运行Express
var runExpress = function (conf, middleware) {
    if (conf && conf.port) {
        app.configure(function () {
            app.set('port', conf.port);
            if (conf.static) {
                app.use(express.static(conf.static));
            }
            var defaultStaticPath = path.join(__dirname, '../frame structure demo/static');
            app.use('/express', express.static(defaultStaticPath));
            app.use(bodyParser.urlencoded({
                extended: true
            }));
            app.use(bodyParser.json());
            app.use(bodyParser.text({
                type: 'text/xml'
            }));
            app.use(bodyParser.text());
            app.use(methodOverride());
            app.use(function (req, res, next) {
                res.setHeader("Access-Control-Allow-Origin", "*");
                res.setHeader('Access-Control-Allow-Methods', 'GET, POST');
                res.setHeader('Access-Control-Allow-Headers', 'X-Requested-With,content-type, Authorization');
                var _method = req.method;
                var _querystring = req.query;
                var _pid = process.pid;
                var _router = req.path;
                req._request = {
                    method: _method,
                    querystring: _querystring,
                    pid: _pid,
                    router: _router
                };
                if (_router == '/zxy-upload' && _method.toUpperCase() == "OPTIONS" ||
                    _router == '/zxy-upload' && _method.toUpperCase() == "POST" ||
                    _router.StartWith('/zxy-download/') && _method.toUpperCase() == "GET") {
                    next();
                }
                else {
                    req.setEncoding('utf-8');
                    if (middleware && typeof middleware == 'function') {
                        middleware(req, res, function () {
                            output(req, next, _pid);
                        });
                    }
                    else {
                        output(req, next, _pid);
                    }
                }
            });
        });
        autoroute(app, { throwErrors: true });
        http.createServer(app).listen(app.get('port'), function () {
            var singleCoreProcess;
            if (process.argv) {
                if (process.argv.length) {
                    singleCoreProcess = process.argv[process.argv.length - 1].indexOf('-d') == 0 ? true : false
                }
                else {
                    singleCoreProcess = process.argv[0].indexOf('-d') == 0 ? true : false
                }
            }
            if (singleCoreProcess) {
                if (zxy.app.f_job) {
                    if (zxy.app.m_verify.isArray(zxy.app.f_job)) {
                        _.each(zxy.app.f_job, function (job) {
                            if (zxy.app.m_verify.isJson(job)) {
                                var job = new require('./comm/job').jobObj(job.conf, job.func, job.params, job.name);
                            }
                        });
                    }
                }
                var serverIp = zxy.app.m_obtain.getServerIP();
                console.info('a single core process[' + process.pid + '] listening on', serverIp, ':', app.get('port'));
            }
        });
        app.post('/zxy-upload', function (req, res, next) {
            res.setHeader("Access-Control-Allow-Origin", "*");
            // 解析一个文件上传
            var form = new multiparty.Form();
            //设置编辑
            form.encoding = 'utf-8';
            //设置文件存储路径
            form.uploadDir = "uploads/";
            //设置单文件大小限制 
            // form.maxFilesSize = 2 * 1024 * 1024;
            //设置所以文件的大小总和
            // form.maxFields = 1000;  
            form.parse(req, function (err, fields, files) {
                var destination = path.join(__dirname, '../../../uploads/');
                var encoding = form.encoding;
                var batch = zxy.app.m_cryptogram.md5(new Date().valueOf().toString() + zxy.app.m_obtain.getRandomStr() + zxy.app.m_obtain.guid());
                var ids = new Array();
                var promise_arr = new Array();
                _.each(files, function (obj) {
                    obj = obj[0];
                    var id = zxy.app.m_obtain.guid();
                    var originalname = obj.originalFilename;
                    ids.push({ uuid: id, file: originalname });
                    var fieldname = obj.fieldName;
                    var filename = obj.path.replace('uploads\\', '').replace('uploads/', '');
                    var mimetype = obj.headers['content-type'];
                    var path = require('path').join(__dirname, '../../../' + obj.path);
                    var size = obj.size;
                    var sql = "INSERT INTO zxy_upload_log("
                        + "id,destination,encoding,fieldname,filename,mimetype,originalname,path,size,upload_time,batch)VALUES("
                        + "@id,@destination,@encoding,@fieldname,@filename,@mimetype,@originalname,@path,@size,NOW(),@batch);";
                    promise_arr.push(
                        zxy.app.m_mysql.run(sql, {
                            id: id,
                            destination: destination,
                            encoding: encoding,
                            fieldname: fieldname,
                            filename: filename,
                            mimetype: mimetype,
                            originalname: originalname,
                            path: path,
                            size: size,
                            batch: batch
                        })
                    );
                });
                Promise.all(promise_arr).then(function () {
                    res.jsonp(ids);
                });
            });
        });
        app.get('/zxy-download/:id', function (req, res, next) {
            res.setHeader("Access-Control-Allow-Origin", "*");
            var id = req.params.id;
            try {
                var sql = "select * from zxy_upload_log where id = @id";
                zxy.app.m_mysql.run(sql, {
                    id: id
                }).then(function (r) {
                    if (r && r.length && r[0]) {
                        if (fs.exists(r[0].path, function (exists) {
                            if (exists) {
                                if (r[0].mimetype.StartWith('image/')) {
                                    var rs = fs.createReadStream(r[0].path);
                                    rs.on('data', function (data) {
                                        res.write(data);
                                    });
                                    rs.on('end', function () {
                                        res.end();
                                    });
                                }
                                else {
                                    var attachmentName = r[0].originalname;
                                    attachmentName = encodeURIComponent(attachmentName);
                                    res.setHeader('Content-disposition', 'attachment; filename=' + attachmentName);
                                    res.setHeader('Content-type', r[0].mimetype + ";charset=utf8");
                                    var filestream = fs.createReadStream(r[0].path);
                                    filestream.on('data', function (chunk) {
                                        res.write(chunk);
                                    });
                                    filestream.on('end', function () {
                                        res.end();
                                    });
                                }
                            }
                            else {
                                res.send({ code: -1, msg: '文件不存在' });
                            }
                        }));
                    }
                    else {
                        res.send({ code: -2, msg: '文件信息不存在' });
                    }
                }).catch(function () {
                    res.send({ code: -3, msg: '查找文件信息异常' });
                });
            }
            catch (ex) {
                res.send({ code: -4, msg: '系统错误' });
            }
        });
    }
}
exports.runExpress = runExpress;

function l(req, pid) {
    setTimeout(function () {
        if (req.path != "/favicon.ico" && req.method.toLowerCase() != "options") {
            var method = req.method,
                path = req.path,
                params = req.params ? zxy.app.m_convert.objToStr(req.params) : "",
                query = req.query ? JSON.stringify(req.query) : "",
                body = req.body ? (zxy.app.m_verify.isJson(req.body) ? JSON.stringify(req.body) : req.body.toString()) : "",
                headers = req.headers ? JSON.stringify(req.headers) : "",
                clientip = zxy.app.m_obtain.getClientIP(req).replace('::ffff:', '');
            // params = params.split(',');
            // params[0] = params[0].substring(1);
            // params[params.length - 1] = params[params.length - 1].substring(0, params[1, params.length - 1].length - 1);
            console.info(method, "| ", path, "| ClientIP:", clientip, "| pid:", pid, "| params:", params, "| query:", query, "| body:", body, "| headers:", headers);
            var sql = "INSERT INTO zxy_api_log VALUES(UUID(),NOW(),@req_ip,@method,@path,@pid,@params,@query,@body,@headers)";
            zxy.app.m_mysql.run(sql, {
                req_ip: clientip,
                method: method,
                path: path,
                pid: pid,
                params: params,
                query: query,
                body: body,
                headers: headers
            });
        }
    }, 0);
}

function output(req, next, pid) {
    if (req.method.toLowerCase() === "post") {
        var _body = "";
        req.addListener("data", function (postDataChunk) {
            _body += postDataChunk;
        });
        req.addListener("end", function () {
            req.body = _body;
            next();
            l(req, pid);
        });
    }
    else {
        next();
        l(req, pid);
    }
}