111 lines
3.8 KiB
JavaScript
111 lines
3.8 KiB
JavaScript
import { dropUndefinedKeys } from '@sentry/utils';
|
|
import { getCurrentScope } from './exports.js';
|
|
|
|
/**
|
|
* @inheritdoc
|
|
*/
|
|
class SessionFlusher {
|
|
|
|
// Cast to any so that it can use Node.js timeout
|
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
|
|
constructor(client, attrs) {
|
|
this._client = client;
|
|
this.flushTimeout = 60;
|
|
this._pendingAggregates = {};
|
|
this._isEnabled = true;
|
|
|
|
// Call to setInterval, so that flush is called every 60 seconds.
|
|
this._intervalId = setInterval(() => this.flush(), this.flushTimeout * 1000);
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
if (this._intervalId.unref) {
|
|
// eslint-disable-next-line @typescript-eslint/no-unsafe-member-access
|
|
this._intervalId.unref();
|
|
}
|
|
this._sessionAttrs = attrs;
|
|
}
|
|
|
|
/** Checks if `pendingAggregates` has entries, and if it does flushes them by calling `sendSession` */
|
|
flush() {
|
|
const sessionAggregates = this.getSessionAggregates();
|
|
if (sessionAggregates.aggregates.length === 0) {
|
|
return;
|
|
}
|
|
this._pendingAggregates = {};
|
|
this._client.sendSession(sessionAggregates);
|
|
}
|
|
|
|
/** Massages the entries in `pendingAggregates` and returns aggregated sessions */
|
|
getSessionAggregates() {
|
|
const aggregates = Object.keys(this._pendingAggregates).map((key) => {
|
|
return this._pendingAggregates[parseInt(key)];
|
|
});
|
|
|
|
const sessionAggregates = {
|
|
attrs: this._sessionAttrs,
|
|
aggregates,
|
|
};
|
|
return dropUndefinedKeys(sessionAggregates);
|
|
}
|
|
|
|
/** JSDoc */
|
|
close() {
|
|
clearInterval(this._intervalId);
|
|
this._isEnabled = false;
|
|
this.flush();
|
|
}
|
|
|
|
/**
|
|
* Wrapper function for _incrementSessionStatusCount that checks if the instance of SessionFlusher is enabled then
|
|
* fetches the session status of the request from `Scope.getRequestSession().status` on the scope and passes them to
|
|
* `_incrementSessionStatusCount` along with the start date
|
|
*/
|
|
incrementSessionStatusCount() {
|
|
if (!this._isEnabled) {
|
|
return;
|
|
}
|
|
const scope = getCurrentScope();
|
|
const requestSession = scope.getRequestSession();
|
|
|
|
if (requestSession && requestSession.status) {
|
|
this._incrementSessionStatusCount(requestSession.status, new Date());
|
|
// This is not entirely necessarily but is added as a safe guard to indicate the bounds of a request and so in
|
|
// case captureRequestSession is called more than once to prevent double count
|
|
scope.setRequestSession(undefined);
|
|
/* eslint-enable @typescript-eslint/no-unsafe-member-access */
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Increments status bucket in pendingAggregates buffer (internal state) corresponding to status of
|
|
* the session received
|
|
*/
|
|
_incrementSessionStatusCount(status, date) {
|
|
// Truncate minutes and seconds on Session Started attribute to have one minute bucket keys
|
|
const sessionStartedTrunc = new Date(date).setSeconds(0, 0);
|
|
this._pendingAggregates[sessionStartedTrunc] = this._pendingAggregates[sessionStartedTrunc] || {};
|
|
|
|
// corresponds to aggregated sessions in one specific minute bucket
|
|
// for example, {"started":"2021-03-16T08:00:00.000Z","exited":4, "errored": 1}
|
|
const aggregationCounts = this._pendingAggregates[sessionStartedTrunc];
|
|
if (!aggregationCounts.started) {
|
|
aggregationCounts.started = new Date(sessionStartedTrunc).toISOString();
|
|
}
|
|
|
|
switch (status) {
|
|
case 'errored':
|
|
aggregationCounts.errored = (aggregationCounts.errored || 0) + 1;
|
|
return aggregationCounts.errored;
|
|
case 'ok':
|
|
aggregationCounts.exited = (aggregationCounts.exited || 0) + 1;
|
|
return aggregationCounts.exited;
|
|
default:
|
|
aggregationCounts.crashed = (aggregationCounts.crashed || 0) + 1;
|
|
return aggregationCounts.crashed;
|
|
}
|
|
}
|
|
}
|
|
|
|
export { SessionFlusher };
|
|
//# sourceMappingURL=sessionflusher.js.map
|