refactor(core): Standardize filename casing for controllers and databases (no-changelog) (#10564)

2025-12-17 18:12:04 +00:00 · 2024-08-27 16:44:32 +02:00
parent be52176585
commit fd58a272e1
264 changed files with 566 additions and 565 deletions
--- a/packages/cli/src/databases/repositories/workflow-statistics.repository.ts
+++ b/packages/cli/src/databases/repositories/workflow-statistics.repository.ts
@@ -0,0 +1,119 @@
+import { Service } from 'typedi';
+import { GlobalConfig } from '@n8n/config';
+import { DataSource, MoreThanOrEqual, QueryFailedError, Repository } from '@n8n/typeorm';
+import { StatisticsNames, WorkflowStatistics } from '../entities/workflow-statistics';
+import type { User } from '@/databases/entities/User';
+
+type StatisticsInsertResult = 'insert' | 'failed' | 'alreadyExists';
+type StatisticsUpsertResult = StatisticsInsertResult | 'update';
+
+@Service()
+export class WorkflowStatisticsRepository extends Repository<WorkflowStatistics> {
+	constructor(
+		dataSource: DataSource,
+		private readonly globalConfig: GlobalConfig,
+	) {
+		super(WorkflowStatistics, dataSource.manager);
+	}
+
+	async insertWorkflowStatistics(
+		eventName: StatisticsNames,
+		workflowId: string,
+	): Promise<StatisticsInsertResult> {
+		// Try to insert the data loaded statistic
+		try {
+			const exists = await this.findOne({
+				where: {
+					workflowId,
+					name: eventName,
+				},
+			});
+			if (exists) return 'alreadyExists';
+			await this.insert({
+				workflowId,
+				name: eventName,
+				count: 1,
+				latestEvent: new Date(),
+			});
+			return 'insert';
+		} catch (error) {
+			// if it's a duplicate key error then that's fine, otherwise throw the error
+			if (!(error instanceof QueryFailedError)) {
+				throw error;
+			}
+			// If it is a query failed error, we return
+			return 'failed';
+		}
+	}
+
+	async upsertWorkflowStatistics(
+		eventName: StatisticsNames,
+		workflowId: string,
+	): Promise<StatisticsUpsertResult> {
+		const dbType = this.globalConfig.database.type;
+		const { tableName } = this.metadata;
+		try {
+			if (dbType === 'sqlite') {
+				await this.query(
+					`INSERT INTO "${tableName}" ("count", "name", "workflowId", "latestEvent")
+					VALUES (1, "${eventName}", "${workflowId}", CURRENT_TIMESTAMP)
+					ON CONFLICT (workflowId, name)
+					DO UPDATE SET count = count + 1, latestEvent = CURRENT_TIMESTAMP`,
+				);
+				// SQLite does not offer a reliable way to know whether or not an insert or update happened.
+				// We'll use a naive approach in this case. Query again after and it might cause us to miss the
+				// first production execution sometimes due to concurrency, but it's the only way.
+				const counter = await this.findOne({
+					select: ['count'],
+					where: {
+						name: eventName,
+						workflowId,
+					},
+				});
+
+				return counter?.count === 1 ? 'insert' : 'failed';
+			} else if (dbType === 'postgresdb') {
+				const queryResult = (await this.query(
+					`INSERT INTO "${tableName}" ("count", "name", "workflowId", "latestEvent")
+					VALUES (1, '${eventName}', '${workflowId}', CURRENT_TIMESTAMP)
+					ON CONFLICT ("name", "workflowId")
+					DO UPDATE SET "count" = "${tableName}"."count" + 1, "latestEvent" = CURRENT_TIMESTAMP
+					RETURNING *;`,
+				)) as Array<{
+					count: number;
+				}>;
+				return queryResult[0].count === 1 ? 'insert' : 'update';
+			} else {
+				const queryResult = (await this.query(
+					`INSERT INTO \`${tableName}\` (count, name, workflowId, latestEvent)
+					VALUES (1, "${eventName}", "${workflowId}", NOW())
+					ON DUPLICATE KEY
+					UPDATE count = count + 1, latestEvent = NOW();`,
+				)) as {
+					affectedRows: number;
+				};
+				// MySQL returns 2 affected rows on update
+				return queryResult.affectedRows === 1 ? 'insert' : 'update';
+			}
+		} catch (error) {
+			if (error instanceof QueryFailedError) return 'failed';
+			throw error;
+		}
+	}
+
+	async queryNumWorkflowsUserHasWithFiveOrMoreProdExecs(userId: User['id']): Promise<number> {
+		return await this.count({
+			where: {
+				workflow: {
+					shared: {
+						role: 'workflow:owner',
+						project: { projectRelations: { userId, role: 'project:personalOwner' } },
+					},
+					active: true,
+				},
+				name: StatisticsNames.productionSuccess,
+				count: MoreThanOrEqual(5),
+			},
+		});
+	}
+}