88 lines
2.0 KiB
JavaScript
88 lines
2.0 KiB
JavaScript
'use strict';
|
|
|
|
const { getFormatFromCopyQuery } = require('../utils/query-info');
|
|
|
|
module.exports = class StreamCopyMetrics {
|
|
constructor (logger, type, sql, user, isGzip = false) {
|
|
this.logger = logger;
|
|
|
|
this.type = type;
|
|
this.format = getFormatFromCopyQuery(sql);
|
|
this.sql = sql;
|
|
this.isGzip = isGzip;
|
|
this.username = user;
|
|
this.size = 0;
|
|
this.gzipSize = 0;
|
|
this.rows = 0;
|
|
|
|
this.startTime = new Date();
|
|
this.endTime = null;
|
|
this.time = null;
|
|
|
|
this.success = true;
|
|
this.error = null;
|
|
|
|
this.ended = false;
|
|
}
|
|
|
|
addSize (size) {
|
|
this.size += size;
|
|
}
|
|
|
|
addGzipSize (size) {
|
|
this.gzipSize += size;
|
|
}
|
|
|
|
end (rows = null, error = null) {
|
|
if (this.ended) {
|
|
return;
|
|
}
|
|
|
|
this.ended = true;
|
|
|
|
if (Number.isInteger(rows)) {
|
|
this.rows = rows;
|
|
}
|
|
|
|
if (error instanceof Error) {
|
|
this.error = error;
|
|
}
|
|
|
|
this.endTime = new Date();
|
|
this.time = (this.endTime.getTime() - this.startTime.getTime()) / 1000;
|
|
|
|
this._log(
|
|
this.startTime.toISOString(),
|
|
this.isGzip && this.gzipSize ? this.gzipSize : null,
|
|
this.error ? this.error.message : null
|
|
);
|
|
}
|
|
|
|
_log (timestamp, gzipSize = null, errorMessage = null) {
|
|
const logData = {
|
|
type: this.type,
|
|
format: this.format,
|
|
size: this.size,
|
|
rows: this.rows,
|
|
gzip: this.isGzip,
|
|
username: this.username,
|
|
time: this.time,
|
|
timestamp,
|
|
sql: this.sql
|
|
};
|
|
|
|
if (gzipSize) {
|
|
logData.gzipSize = gzipSize;
|
|
}
|
|
|
|
if (errorMessage) {
|
|
logData.error = errorMessage;
|
|
this.success = false;
|
|
}
|
|
|
|
logData.success = this.success;
|
|
|
|
this.logger.info({ ingestion: logData }, 'Copy to/from query metrics');
|
|
}
|
|
};
|