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

Add request_count for all bots

Browse Source
This commit is contained in:
Bulat Kurbanov
2022-04-02 15:28:09 +03:00
parent ef43c5c454
commit a8d1d420c9
1 changed files with 26 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/analytics/users_counter.ts
View File

@@ -48,9 +48,7 @@ export default class UsersCounter {
return (await client.hKeys(`${RedisKeys.UsersActivity}_${bot}`)).map((userId) => parseInt(userId)); return (await client.hKeys(`${RedisKeys.UsersActivity}_${bot}`)).map((userId) => parseInt(userId));
} }
static async _getAllUsersCount(): Promise<number> { static async _getAllUsersCount(botsUsernames: string[]): Promise<number> {
const botsUsernames = await this._getBotsUsernames();
const users = new Set<number>(); const users = new Set<number>();
await Promise.all( await Promise.all(
@@ -62,9 +60,7 @@ export default class UsersCounter {
return users.size; return users.size;
} }
static async _getUsersByBots(): Promise<{[bot: string]: number}> { static async _getUsersByBots(botsUsernames: string[]): Promise<{[bot: string]: number}> {
const botsUsernames = await this._getBotsUsernames();
const result: {[bot: string]: number} = {}; const result: {[bot: string]: number} = {};
await Promise.all( await Promise.all(
@@ -76,43 +72,55 @@ export default class UsersCounter {
return result; return result;
} }
static async _incrementRequests() { static async _incrementRequests(bot: string) {
const client = await this._getClient(); const client = await this._getClient();
const exists = await client.exists(RedisKeys.RequestsCount); const key = `${RedisKeys.RequestsCount}_${bot}`;
const exists = await client.exists(key);
if (!exists) { if (!exists) {
await client.set(RedisKeys.RequestsCount, 0); await client.set(key, 0);
} }
await client.incr(RedisKeys.RequestsCount); await client.incr(key);
} }
static async _getRequestsCount(): Promise<number> { static async _getRequestsByBotCount(botsUsernames: string[]): Promise<{[bot: string]: number}> {
const client = await this._getClient(); const client = await this._getClient();
const result = await client.get(RedisKeys.RequestsCount); const result: {[bot: string]: number} = {};
if (result === null) return 0; await Promise.all(
botsUsernames.map(async (bot) => {
const count = await client.get(`${RedisKeys.RequestsCount}_${bot}`);
result[bot] = count !== null ? parseInt(count) : 0;
})
);
return parseInt(result); return result;
} }
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.hSet(`${RedisKeys.UsersActivity}_${bot}`, userId, moment().format());
await this._incrementRequests(); await this._incrementRequests(bot);
} }
static async getMetrics(): Promise<string> { static async getMetrics(): Promise<string> {
const botUsernames = await this._getBotsUsernames();
const lines = []; const lines = [];
lines.push(`all_users_count ${await this._getAllUsersCount()}`); lines.push(`all_users_count ${await this._getAllUsersCount(botUsernames)}`);
lines.push(`requests_count ${await this._getRequestsCount()}`);
const usersByBots = await this._getUsersByBots(); const requestsByBotCount = await this._getRequestsByBotCount(botUsernames);
Object.keys(requestsByBotCount).forEach((bot: string) => {
lines.push(`requests_count{bot="${bot}"} ${requestsByBotCount[bot]}`);
});
const usersByBots = await this._getUsersByBots(botUsernames);
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]}`)
}); });