cart-goods-count

@@ -4,12 +4,12 @@
 
 const countModel = require('../models/count');
 
-let cartCount = (req, res, next) => {
+const cartCount = (req, res, next) => {
 
     let uid = req.user.uid;
     let shoppingKey = req.cookies._SPK || '';
 
-    countModel.cartCount(uid, shoppingKey).then((result) => {
+    req.ctx(countModel).cartCount(uid, shoppingKey).then((result) => {
         res.json(result);
     }).catch(next);
 };

 'use strict';
 
-const api = global.yoho.API;
+class cartCountModel extends global.yoho.BaseModel {
+    constructor(ctx) {
+        super(ctx);
+    }
 
-const cartCount = (uid, shoppingKey) => {
-    let params = {
-        method: 'app.Shopping.count'
-    };
+    cartCount(uid, shoppingKey) {
+        let params = {
+            method: 'app.Shopping.count'
+        };
 
-    if (!uid && !shoppingKey) {
-        return Promise.resolve({});
-    }
+        if (!uid && !shoppingKey) {
+            return Promise.resolve({});
+        }
 
-    if (uid) {
-        Object.assign(params, {
-            uid: uid
-        });
-    }
+        if (uid) {
+            Object.assign(params, {
+                uid: uid
+            });
+        }
+
+        if (shoppingKey) {
+            Object.assign(params, {
+                shopping_key: shoppingKey
+            });
+        }
 
-    if (shoppingKey) {
-        Object.assign(params, {
-            shopping_key: shoppingKey
+        return this.get({
+            data: params
+        }).then((result) => {
+            return result;
         });
     }
+}
 
-    return api.get('', params).then((result) => {
-        return result;
-    });
-};
-
-module.exports = {
-    cartCount
-};
+module.exports = cartCountModel;