StyAndCo/zeppelin
3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-03-15 05:41:51 +00:00
Code Releases Activity

fix: add missing queryLogger.ts

Browse source
This commit is contained in:
Dragory 2021-11-02 21:03:55 +02:00
parent a7d71e0762
commit 65b87f4e0b
No known key found for this signature in database
GPG key ID: 5F387BA66DF8AAC1
1 changed files with 42 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

42
backend/src/data/queryLogger.ts Normal file
View file

@ -0,0 +1,42 @@
import { AdvancedConsoleLogger } from "typeorm/logger/AdvancedConsoleLogger";
import type { QueryRunner } from "typeorm";
let groupedQueryStats: Map<string, number> = new Map();
const selectTableRegex = /FROM `?([^\s`]+)/i;
const updateTableRegex = /UPDATE `?([^\s`]+)/i;
const deleteTableRegex = /FROM `?([^\s`]+)/;
const insertTableRegex = /INTO `?([^\s`]+)/;
export class QueryLogger extends AdvancedConsoleLogger {
logQuery(query: string, parameters?: any[], queryRunner?: QueryRunner): any {
let type: string | undefined;
let table: string | undefined;
if (query.startsWith("SELECT")) {
type = "SELECT";
table = query.match(selectTableRegex)?.[1];
} else if (query.startsWith("UPDATE")) {
type = "UPDATE";
table = query.match(updateTableRegex)?.[1];
} else if (query.startsWith("DELETE")) {
type = "DELETE";
table = query.match(deleteTableRegex)?.[1];
} else if (query.startsWith("INSERT")) {
type = "INSERT";
table = query.match(insertTableRegex)?.[1];
} else {
return;
}
const key = `${type} ${table}`;
const newCount = (groupedQueryStats.get(key) ?? 0) + 1;
groupedQueryStats.set(key, newCount);
}
}
export function consumeQueryStats() {
const map = groupedQueryStats;
groupedQueryStats = new Map();
return map;
}