Merge branch 'gray' of git.yoho.cn:fe/yohobuywap-node into gray

@@ -108,67 +108,7 @@ app.use(bodyParser.urlencoded({
 app.use(cookieParser());
 app.use(cookieParser());
 app.use(compression());
 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();
-});
-
+require('./doraemon/middleware/yoho-session')(app);
 
 
 app.use((req, res, next) => {
 app.use((req, res, next) => {
     req.user = {}; // 全局的用户数据
     req.user = {}; // 全局的用户数据

+const config = global.yoho.config;
+const memcachedSession = require('yoho-express-session');
+const _ = require('lodash');
+const uuid = require('uuid');
+
+const cookieSession = require('client-sessions');
+const memcached = require('connect-memcached');
+const MemcachedStore = memcached(memcachedSession);
+
+/**
+ * 该中间件主要把 express-session 和 client-session 集中起来处理，如果 memcached 出错了，使用 cookie session
+ * @param opts.backSession cookieSession 的键名
+ * @returns {function(*=, *=, *)}
+ */
+function yohoSession(opts) {
+    return (req, res, next) => {
+        let notUseMemcached = _.get(req.app.locals.wap, 'session.removeMemcached', false);
+
+        opts.backSession = opts.backSession || 'session2';
+
+        if (req.session && !notUseMemcached) {
+            req.sessionError = false;
+        } else {
+            // 重建 session
+            res.emit('sessionError');
+            req.sessionError = true;
+
+            req.sessionID = uuid.v4();
+            req.session = new memcachedSession.Session(req, req[opts.backSession].sessionBack);
+            req.session.cookie = new memcachedSession.Cookie({
+                domain: 'yohobuy.com',
+                httpOnly: false
+            });
+        }
+
+        Object.defineProperty(req.session, 'reset', {
+            configurable: true,
+            enumerable: false,
+            value: function() {
+                req.session.destroy();
+                req[opts.backSession].reset();
+            },
+            writable: false
+        });
+
+        // 备份数据
+        req[opts.backSession].sessionBack = req.session;
+
+        next();
+    };
+}
+
+module.exports = (app) => {
+    app.use(memcachedSession({  // eslint-disable-line
+        proxy: true,
+        resave: false,
+        saveUninitialized: true,
+        unset: 'destroy',
+        secret: '82dd7e724f2c6870472c89dfa43cf48d',
+        name: 'yohobuy_session',
+        cookie: {
+            domain: 'yohobuy.com',
+            httpOnly: false
+        },
+        store: new MemcachedStore({
+            hosts: config.memcache.session,
+            prefix: 'yohobuy_session:',
+            reconnect: 5000,
+            timeout: 1000,
+            retries: 0
+        })
+    }));
+
+    app.use(cookieSession({ // eslint-disable-line
+        requestKey: 'session2',
+        cookieName: 'yohobuy_session_cookie',
+        secret: '82dd7e724f2c6870472c89dfa43cf48d',
+        cookie: {
+            domain: 'yohobuy.com',
+            ephemeral: true
+        }
+    }));
+
+    app.use(yohoSession({
+        backSession: 'session2'
+    }));
+};