app.js 5.54 KB
/**
 * yohobuy app
 * @author: xuqi<qi.xu@yoho.cn>
 * @date: 2016/4/25
 */
'use strict';

if (process.env.USE_APM === '1' && process.env.NODE_ENV === 'production') {
    require('oneapm');
}

const config = require('./config/common');

global.Promise = require('bluebird');

const yohoLib = require('yoho-node-lib');

const express = require('express');
const path = require('path');
const compression = require('compression');
const bodyParser = require('body-parser');
const cookieParser = require('cookie-parser');
const uuid = require('uuid');
const moment = require('moment');

const pkg = require('./package.json');
const cookieSession = require('client-sessions');
const memcachedSession = require('yoho-express-session');
const memcached = require('connect-memcached');
const MemcachedStore = memcached(memcachedSession);
const _ = require('lodash');

const app = express();

// 向模板注入变量
app.locals.devEnv = app.get('env') === 'development';
app.locals.version = pkg.version;
app.locals.startTime = moment().format('YYYYMMDDHH');

// 全局注册library
yohoLib.global(config);

// zookeeper
if (config.zookeeperServer) {
    require('yoho-zookeeper')(config.zookeeperServer, 'wap', app.locals.wap = {}, global.yoho.cache);
}

// 访问域名层级
app.set('subdomain offset', 3);

app.set('etag', false);

app.enable('trust proxy');

// 请求限制中间件
if (!app.locals.devEnv) {
    app.use(require('./doraemon/middleware/limiter'));
}

// 指定libray目录
global.utils = path.resolve('./utils');

const logger = global.yoho.logger;

// 访问域名层级
app.set('subdomain offset', 3);
app.use(global.yoho.hbs({
    extname: '.hbs',
    defaultLayout: 'layout',
    layoutsDir: path.join(__dirname, 'doraemon/views'),
    partialsDir: path.join(__dirname, 'doraemon/views/partial'),
    views: path.join(__dirname, 'doraemon/views'),
    helpers: _.assign(global.yoho.helpers, require('./utils/helpers'))
}));

app.use(global.yoho.middleware());

app.use(bodyParser.json());
app.use(bodyParser.urlencoded({
    extended: false
}));
app.use(cookieParser());
app.use(compression());

const sessionStore = new MemcachedStore({
    hosts: config.memcache.session,
    prefix: 'yohobuy_session:',
    reconnect: 5000,
    timeout: 1000,
    retries: 0
});

app.use(memcachedSession({
    proxy: true,
    resave: false,
    saveUninitialized: true,
    unset: 'destroy',
    secret: '82dd7e724f2c6870472c89dfa43cf48d',
    name: 'yohobuy_session',
    cookie: {
        domain: 'yohobuy.com',
        httpOnly: false
    },
    store: sessionStore
}));

app.use(cookieSession({
    requestKey: 'session2',
    cookieName: 'yohobuy_session_cookie',
    secret: '82dd7e724f2c6870472c89dfa43cf48d',
    cookie: {
        domain: 'yohobuy.com',
        ephemeral: true
    }
}));

app.use((req, res, next) => {
    if (req.session) {
        let sessionKeys = Object.keys(req.session || {});
        let backSessionKeys = Object.keys(req.session2.sessionBack || {});

        if (backSessionKeys.length > sessionKeys.length) {
            let differences = _.difference(backSessionKeys, sessionKeys);

            _.forEach(differences, d => {
                req.session[d] = req.session2.sessionBack[d];
            });
        }
        req.session2.sessionBack = req.session;
    } else {
        req.session = new memcachedSession.Session(req);
        req.session.cookie = new memcachedSession.Cookie({
            domain: 'yohobuy.com',
            httpOnly: false
        });
        req.session = _.assign(req.session, req.session2.sessionBack);
    }

    if (!req.sessionID) {
        req.sessionID = uuid.v4();
    }

    next();
});


app.use((req, res, next) => {
    req.user = {}; // 全局的用户数据
    req.yoho = {}; // req和res绑定yoho对象,用于传递全局数据, 如req.yoho.channel等
    req.app.locals.wap = app.locals.wap; // zookeper对象赋值

    // 临时增加错误日志记录
    let sendJson = res.json;

    res.json = function(...args) {
        try {
            sendJson(...args);
        } catch (e) {
            logger.error(`res.json error data: ${JSON.stringify(args)}`);
        }
    };

    next();
});

// dispatcher
try {
    const user = require('./doraemon/middleware/user');
    const reqParamsFilter = require('./doraemon/middleware/req-params-filter');
    const urlRewrite = require('./doraemon/middleware/url-rewrite');
    const subDomain = require('./doraemon/middleware/sub-domain');
    const itemNameHandler = require('./doraemon/middleware/item-name-handler');
    const setYohoData = require('./doraemon/middleware/set-yoho-data');
    const errorHanlder = require('./doraemon/middleware/error-handler');
    const setPageInfo = require('./doraemon/middleware/set-pageinfo');
    const devtools = require('./doraemon/middleware/devtools');
    const layoutTools = require('./doraemon/middleware/layout-tools');
    const seo = require('./doraemon/middleware/seo');
    const pageCache = require('./doraemon/middleware/page-cache');

    // YOHO 前置中间件
    app.use(reqParamsFilter());
    app.use(urlRewrite());
    app.use(subDomain());
    app.use(itemNameHandler);
    app.use(setYohoData());
    app.use(user());
    app.use(seo());
    app.use(setPageInfo());

    if (app.locals.devEnv) {
        app.use(devtools());
    }

    app.use(layoutTools());

    app.use(pageCache());
    require('./dispatch')(app);
    app.all('*', errorHanlder.notFound()); // 404

    // YOHO 后置中间件
    app.use(errorHanlder.serverError());
} catch (err) {
    logger.error(err);
}

// listener
app.listen(config.port, function() {
    logger.info('yohobuy start');
});