flibusta-apps/book_bot
1
0
Fork 0
mirror of https://github.com/flibusta-apps/book_bot.git synced 2025-12-06 15:35:35 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update analytics

Browse Source
This commit is contained in:
Bulat Kurbanov
2022-04-06 20:13:39 +03:00
parent bce6658d48
commit 2778ddea1f
1 changed files with 23 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
src/analytics/users_counter.ts
View File

@@ -1,5 +1,4 @@
import { createClient, RedisClientType } from 'redis'; import { createClient, RedisClientType } from 'redis';
import moment from 'moment';
import env from '@/config'; import env from '@/config';
import BotsManager from '@/bots/manager'; import BotsManager from '@/bots/manager';
@@ -9,6 +8,7 @@ import Sentry from '@/sentry';
enum RedisKeys { enum RedisKeys {
UsersActivity = "users_activity", UsersActivity = "users_activity",
UsersActivity24 = "users_activity_24",
RequestsCount = "requests_count", RequestsCount = "requests_count",
} }
@@ -42,30 +42,35 @@ export default class UsersCounter {
return Promise.all(promises); return Promise.all(promises);
} }
static async _getUsersByBot(bot: string): Promise<number[]> { static async _getUsersByBot(bot: string, lastDay: boolean): Promise<number[]> {
const client = await this._getClient(); const client = await this._getClient();
return (await client.hKeys(`${RedisKeys.UsersActivity}_${bot}`)).map((userId) => parseInt(userId)); const prefix = lastDay ? RedisKeys.UsersActivity24 : RedisKeys.UsersActivity;
const template = `${prefix}_${bot}_`;
return (await client.keys(template + '*')).map(
(key) => parseInt(key.replace(template, ""))
);
} }
static async _getAllUsersCount(botsUsernames: string[]): Promise<number> { static async _getAllUsersCount(botsUsernames: string[], lastDay: boolean): Promise<number> {
const users = new Set<number>(); const users = new Set<number>();
await Promise.all( await Promise.all(
botsUsernames.map(async (bot) => { botsUsernames.map(async (bot) => {
(await this._getUsersByBot(bot)).forEach((user) => users.add(user)); (await this._getUsersByBot(bot, lastDay)).forEach((user) => users.add(user));
}) })
); );
return users.size; return users.size;
} }
static async _getUsersByBots(botsUsernames: string[]): Promise<{[bot: string]: number}> { static async _getUsersByBots(botsUsernames: string[], lastDay: boolean): Promise<{[bot: string]: number}> {
const result: {[bot: string]: number} = {}; const result: {[bot: string]: number} = {};
await Promise.all( await Promise.all(
botsUsernames.map(async (bot) => { botsUsernames.map(async (bot) => {
result[bot] = (await this._getUsersByBot(bot)).length; result[bot] = (await this._getUsersByBot(bot, lastDay)).length;
}) })
); );
@@ -104,7 +109,9 @@ export default class UsersCounter {
static async take(userId: number, bot: string) { static async take(userId: number, bot: string) {
const client = await this._getClient(); const client = await this._getClient();
await client.hSet(`${RedisKeys.UsersActivity}_${bot}`, userId, moment().format()); await client.set(`${RedisKeys.UsersActivity}_${bot}_${userId}`, 1);
await client.set(`${RedisKeys.UsersActivity24}_${bot}_${userId}`, 1, {EX: 24 * 60 * 60});
await this._incrementRequests(bot); await this._incrementRequests(bot);
} }
@@ -113,18 +120,24 @@ export default class UsersCounter {
const lines = []; const lines = [];
lines.push(`all_users_count ${await this._getAllUsersCount(botUsernames)}`); lines.push(`all_users_count ${await this._getAllUsersCount(botUsernames, false)}`);
lines.push(`all_users_count_24h ${await this._getAllUsersCount(botUsernames, true)}`);
const requestsByBotCount = await this._getRequestsByBotCount(botUsernames); const requestsByBotCount = await this._getRequestsByBotCount(botUsernames);
Object.keys(requestsByBotCount).forEach((bot: string) => { Object.keys(requestsByBotCount).forEach((bot: string) => {
lines.push(`requests_count{bot="${bot}"} ${requestsByBotCount[bot]}`); lines.push(`requests_count{bot="${bot}"} ${requestsByBotCount[bot]}`);
}); });
const usersByBots = await this._getUsersByBots(botUsernames); const usersByBots = await this._getUsersByBots(botUsernames, false);
Object.keys(usersByBots).forEach((bot: string) => { Object.keys(usersByBots).forEach((bot: string) => {
lines.push(`users_count{bot="${bot}"} ${usersByBots[bot]}`) lines.push(`users_count{bot="${bot}"} ${usersByBots[bot]}`)
}); });
const usersByBots24h = await this._getUsersByBots(botUsernames, true);
Object.keys(usersByBots24h).forEach((bot: string) => {
lines.push(`users_count_24h{bot="${bot}"} ${usersByBots24h[bot]}`)
});
return lines.join("\n"); return lines.join("\n");
} }
} }