update

app/controller/count.js

@@ -0,0 +1,16 @@
+'use strict';
+const Controller = require('egg').Controller;
+const { CrudController } = require('naf-framework-mongoose/lib/controller');
+
+// 统计
+class CountController extends Controller {
+  constructor(ctx) {
+    super(ctx);
+    this.service = this.ctx.service.count;
+  }
+  async index() {
+    const data = await this.service.index();
+    this.ctx.ok({ data });
+  }
+}
+module.exports = CrudController(CountController, {});

app/router.js

@@ -13,6 +13,8 @@ module.exports = app => {
   router.get(`${profix}authBack`, controller.weixin.authBack);
   router.get(`${profix}wxUser`, controller.weixin.wxUser);
   router.post(`${profix}jsapi`, controller.weixin.jsapiAuth);
+  // 统计
+  router.get(`${profix}count/index`, controller.count.index);
   require('./router/refute')(app); // 文章/辟谣
   require('./router/topic')(app); // 社区话题
   require('./router/service')(app); // 咨询服务

app/service/count.js

@@ -0,0 +1,104 @@
+'use strict';
+const { CrudService } = require('naf-framework-mongoose/lib/service');
+const { BusinessError, ErrorCode } = require('naf-core').Error;
+const _ = require('lodash');
+const assert = require('assert');
+const moment = require('moment');
+
+// 统计
+class CountService extends CrudService {
+  constructor(ctx) {
+    super(ctx, 'count');
+    this.model = this.ctx.model.Count;
+    this.refute = this.ctx.model.Refute;
+    this.service = this.ctx.model.Service;
+    this.topic = this.ctx.model.Topic;
+  }
+  /**
+   * 首页统计
+   */
+  async index() {
+  // 本周的文章发表数量,折线图,3条线,没有的天也需要日期,默认值为0:模型(数据过滤)
+    const articles = await this.articleSum();
+    // 3种类型各自的总阅读数
+    const reads = await this.readSum();
+    return { articles, reads };
+  }
+  /**
+   * 本周的文章发表数量
+   */
+  async articleSum() {
+    // TODO 查询本周的周一至周日的日期
+    // 聚合函数查询本周发表的数量,按日期分组(应该需要加一个字段,单纯的是年月日) 'YYYY-MM-DD'
+    const refute = await this.getWeekResult('refute');
+    const serve = await this.getWeekResult('service');
+    const topic = await this.getWeekResult('topic');
+    return { refute, serve, topic };
+  }
+  /**
+   * 根据表名查出一周的发表数量
+   * @param {String} model 表名
+   * @return {Array} arr
+   */
+  async getWeekResult(model) {
+    const weekday = moment().weekday() - 1;
+    // 因为时区问题查询条件需要在00:00:00的基础上减去8小时,将查询条件也变成GMT时间
+    const start = moment(`${moment().subtract(weekday, 'days')
+      .format('YYYY-MM-DD')} 00:00:00`).subtract(8, 'hours').format();
+    const end = moment(`${moment().add(7 - weekday, 'days')
+      .format('YYYY-MM-DD')} 00:00:00`).subtract(8, 'hours').format();
+    const res = await this[model].aggregate([
+      { $match: { 'meta.createdAt': { $gte: new Date(start), $lt: new Date(end) } } },
+      {
+        $project: {
+          // meta的时间因为时区问题,差8个小时,加回来
+          date: { $substr: [{ $add: [ '$meta.createdAt', 28800000 ] }, 0, 10 ] },
+        },
+      },
+      {
+        $group: {
+          _id: '$date',
+          sum: { $sum: 1 },
+        },
+      },
+    ]);
+    const arr = [];
+    // 将本周每天都填充上
+    for (let i = 1; i <= 7; i++) {
+      const date = moment(start).add(i, 'days').format('YYYY-MM-DD');
+      const r = res.find(f => _.isEqual(f._id, date));
+      if (r) arr.push({ date, sum: r.sum });
+      else arr.push({ date, sum: 0 });
+    }
+    return arr;
+  }
+
+  /**
+   * 3种类型各自的总阅读数
+   */
+  async readSum() {
+    // 查出各自的阅读数,聚合加和
+    const refute = await this.getReadSum('refute');
+    const serve = await this.getReadSum('service');
+    const topic = await this.getReadSum('topic');
+    return { refute, serve, topic };
+  }
+
+  async getReadSum(model) {
+    const result = await this[model].aggregate([
+      { $project: { _id: 1, read: 1 } },
+      { $group: {
+        _id: '',
+        read: { $sum: '$read' },
+      } },
+    ]);
+    let res = 0;
+    if (result.length > 0) {
+      res = _.get(_.head(result), 'read', 0);
+    }
+    return res;
+  }
+
+}
+
+module.exports = CountService;