feat: Extend collection of usage metrics during license renewal (no-changelog) (#8369)

Co-authored-by: Iván Ovejero <ivov.src@gmail.com>
This commit is contained in:
Cornelius Suermann
2024-01-22 12:29:28 +01:00
committed by GitHub
parent ebf2b0d55c
commit d597c2ab29
9 changed files with 212 additions and 16 deletions

View File

@@ -0,0 +1,75 @@
import config from '@/config';
import { Service } from 'typedi';
import { DataSource, Repository, Entity } from 'typeorm';
@Entity()
export class UsageMetrics {}
@Service()
export class UsageMetricsRepository extends Repository<UsageMetrics> {
constructor(dataSource: DataSource) {
super(UsageMetrics, dataSource.manager);
}
toTableName(name: string) {
const tablePrefix = config.getEnv('database.tablePrefix');
let tableName =
config.getEnv('database.type') === 'mysqldb'
? `\`${tablePrefix}${name}\``
: `"${tablePrefix}${name}"`;
const pgSchema = config.getEnv('database.postgresdb.schema');
if (pgSchema !== 'public') tableName = [pgSchema, tablePrefix + name].join('.');
return tableName;
}
async getLicenseRenewalMetrics() {
type Row = {
enabled_user_count: string | number;
active_workflow_count: string | number;
total_workflow_count: string | number;
total_credentials_count: string | number;
production_executions_count: string | number;
manual_executions_count: string | number;
};
const userTable = this.toTableName('user');
const workflowTable = this.toTableName('workflow_entity');
const credentialTable = this.toTableName('credentials_entity');
const workflowStatsTable = this.toTableName('workflow_statistics');
const [
{
enabled_user_count: enabledUsers,
active_workflow_count: activeWorkflows,
total_workflow_count: totalWorkflows,
total_credentials_count: totalCredentials,
production_executions_count: productionExecutions,
manual_executions_count: manualExecutions,
},
] = (await this.query(`
SELECT
(SELECT COUNT(*) FROM ${userTable} WHERE disabled = false) AS enabled_user_count,
(SELECT COUNT(*) FROM ${workflowTable} WHERE active = true) AS active_workflow_count,
(SELECT COUNT(*) FROM ${workflowTable}) AS total_workflow_count,
(SELECT COUNT(*) FROM ${credentialTable}) AS total_credentials_count,
(SELECT SUM(count) FROM ${workflowStatsTable} WHERE name IN ('production_success', 'production_error')) AS production_executions_count,
(SELECT SUM(count) FROM ${workflowStatsTable} WHERE name IN ('manual_success', 'manual_error')) AS manual_executions_count;
`)) as Row[];
const toNumber = (value: string | number) =>
typeof value === 'number' ? value : parseInt(value, 10);
return {
enabledUsers: toNumber(enabledUsers),
activeWorkflows: toNumber(activeWorkflows),
totalWorkflows: toNumber(totalWorkflows),
totalCredentials: toNumber(totalCredentials),
productionExecutions: toNumber(productionExecutions),
manualExecutions: toNumber(manualExecutions),
};
}
}

View File

@@ -10,7 +10,7 @@ export class UserRepository extends Repository<User> {
super(User, dataSource.manager);
}
async findManybyIds(userIds: string[]) {
async findManyByIds(userIds: string[]) {
return await this.find({
where: { id: In(userIds) },
relations: ['globalRole'],