fork from bc4552c5a8

exts/yapi-plugin-statistics/statisMockModel.js

@@ -0,0 +1,75 @@
+/**
+ * Created by gxl.gao on 2017/10/24.
+ */
+const yapi = require('yapi.js');
+const baseModel = require('models/base.js');
+
+class statisMockModel extends baseModel {
+    getName() {
+        return 'statis_mock';
+    }
+
+    getSchema() {
+        return {
+            interface_id: { type: Number, required: true },
+            project_id: { type: Number, required: true },
+            group_id: { type: Number, required: true },
+            time: Number, //'时间戳'
+            ip: String,
+            date: String
+        };
+    }
+
+    countByGroupId(id){
+        return this.model.countDocuments({
+            group_id: id
+        })
+    }
+
+    save(data) {
+        let m = new this.model(data);
+        return m.save();
+    }
+
+    getTotalCount() {
+        return this.model.countDocuments({});
+    }
+
+    async getDayCount(timeInterval) {
+        let end = timeInterval[1];
+        let start = timeInterval[0];
+        let data = [];
+        const cursor = this.model.aggregate([
+            {
+                $match: {
+                    date: { $gt: start, $lte: end }
+                }
+            },
+            {
+                $group: {
+                    _id: '$date',  //$region is the column name in collection
+                    count: { $sum: 1 }
+                }
+            },
+            {
+                $sort: { _id: 1 }
+            }
+        ]).cursor({}).exec();
+		await cursor.eachAsync(doc => data.push(doc));
+		return data;
+
+	}
+
+    list() {
+        return this.model.find({}).select('date').exec();
+    }
+
+    up(id, data) {
+        data.up_time = yapi.commons.time();
+        return this.model.updateOne({
+            _id: id
+        }, data, { runValidators: true });
+    }
+}
+
+module.exports = statisMockModel;