unleash.unleash/lib/db/client-metrics-db.js

'use strict';

const METRICS_COLUMNS = ['id', 'created_at', 'metrics'];
const TABLE = 'client_metrics';

const ONE_MINUTE = 60 * 1000;

const mapRow = row => ({
    id: row.id,
    createdAt: row.created_at,
    metrics: row.metrics,
});

class ClientMetricsDb {
    constructor(db, getLogger) {
        this.db = db;
        this.logger = getLogger('client-metrics-db.js');

        // Clear old metrics regulary
        const clearer = () => this.removeMetricsOlderThanOneHour();
        setTimeout(clearer, 10).unref();
        setInterval(clearer, ONE_MINUTE).unref();
    }

    async removeMetricsOlderThanOneHour() {
        const rows = await this.db(TABLE)
            .whereRaw("created_at < now() - interval '1 hour'")
            .del();

        if (rows > 0) {
            this.logger.debug(`Deleted ${rows} metrics`);
        }
    }

    // Insert new client metrics
    async insert(metrics) {
        return this.db(TABLE).insert({ metrics });
    }

    // Used at startup to load all metrics last week into memory!
    async getMetricsLastHour() {
        const result = await this.db
            .select(METRICS_COLUMNS)
            .from(TABLE)
            .limit(2000)
            .whereRaw("created_at > now() - interval '1 hour'")
            .orderBy('created_at', 'asc');

        return result.map(mapRow);
    }

    // Used to poll for new metrics
    async getNewMetrics(lastKnownId) {
        const result = await this.db
            .select(METRICS_COLUMNS)
            .from(TABLE)
            .limit(1000)
            .where('id', '>', lastKnownId)
            .orderBy('created_at', 'asc');

        return result.map(mapRow);
    }
}

module.exports = ClientMetricsDb;