feat: get tweet count by accounts

.gitignore

@@ -35,3 +35,4 @@ lib
 docs
 .idea
 .vscode
+/v8-compile-cache-0

schemas/tweet.count.json

@@ -0,0 +1,18 @@
+{
+  "$id": "tweet.count",
+  "type": "array",
+  "items": {
+    "type": "object",
+    "required": [
+      "account"
+    ],
+    "properties": {
+      "account": {
+        "type": "string"
+      },
+      "cursor": {
+        "$ref": "common.json#/definitions/tweetId"
+      }
+    }
+  }
+}

src/actions/tweet.count.js

@@ -0,0 +1,19 @@
+const { ActionTransport } = require('@microfleet/plugin-router');
+
+/**
+ * @api {http} <prefix>.tweet.count Count tweets by accounts
+ * @apiVersion 1.0.0
+ * @apiName tweet.count
+ * @apiGroup Feed
+ * @apiSchema {jsonschema=../../schemas/tweet.count.json} apiParam
+ */
+async function TweetCountAction({ params }) {
+  const data = await this.service('feed')
+    .countByAccounts(params);
+  return { data };
+}
+
+TweetCountAction.schema = 'tweet.count';
+TweetCountAction.transports = [ActionTransport.amqp];
+
+module.exports = TweetCountAction;

src/services/feed.js

@@ -102,6 +102,12 @@ class Feed {
       .twitterStatuses()
       .byId(data.tweetId);
   }
+
+  async countByAccounts(data) {
+    return this.service('storage')
+      .twitterStatuses()
+      .countByAccounts(data);
+  }
 }
 
 module.exports = Feed;

src/services/storage/twitter-statuses.js

@@ -45,6 +45,31 @@ class TwitterStatuses {
     return query;
   }
 
+  countByAccount(data) {
+    const {
+      account,
+      cursor,
+    } = data;
+
+    const query = this.knex
+      .select('account')
+      .countDistinct('id')
+      .from(this.table)
+      .where('account', account)
+      .groupBy('account');
+
+    if (cursor) {
+      query.where('id', '>', cursor);
+    }
+
+    return query;
+  }
+
+  countByAccounts(data) {
+    const subqueries = data.map(({ account, cursor }) => this.countByAccount({ account, cursor }));
+    return this.knex.unionAll(subqueries);
+  }
+
   remove(data) {
     return this.knex(this.table).where('account', data.account).del();
   }

test/docker-compose.yml

@@ -14,6 +14,10 @@ services:
       NCONF_FILE_PATH: '["/src/configs/ms-users/config.js"]'
       MS_USERS__INIT_ADMIN_ACCOUNTS_DELAY: '1'
 
+  postgres:
+    ports:
+      - "65432:5432"
+
   tester:
     depends_on:
       - postgres

test/suites/01.twitter.js

@@ -271,6 +271,11 @@ describe('twitter', function testSuite() {
     assert(meta.favorite_count);
   });
 
+  it('get count by accounts', async () => {
+    const reply = await service.amqp.publishAndWait('social.tweet.count', [{ account: 'EvgenyPoyarkov' }, { account: 'v_aminev' }]);
+    assert.notEqual(reply.data.length, 0);
+  });
+
   it('rejects with error on get tweet validation', async () => {
     await assert.rejects(service.amqp.publishAndWait(uri.getOne, payload.invalidTweet), {
       name: 'HttpStatusError',