130 lines
3.8 KiB
JavaScript
130 lines
3.8 KiB
JavaScript
'use strict';
|
|
|
|
const throwIfMissing = require('../../support/util').throwIfMissing;
|
|
const isEmpty = require('lodash.isempty');
|
|
const log = require('intel').getLogger('sitespeedio.plugin.influxdb');
|
|
const Sender = require('./sender');
|
|
const sendAnnotations = require('./send-annotation');
|
|
const DataGenerator = require('./data-generator');
|
|
|
|
const defaultConfig = {
|
|
port: 8086,
|
|
protocol: 'http',
|
|
database: 'sitespeed',
|
|
tags: 'category=default',
|
|
includeQueryParams: false
|
|
};
|
|
|
|
module.exports = {
|
|
open(context, options) {
|
|
throwIfMissing(options.influxdb, ['host', 'database'], 'influxdb');
|
|
this.filterRegistry = context.filterRegistry;
|
|
log.debug(
|
|
'Setup InfluxDB host %s and database %s',
|
|
options.influxdb.host,
|
|
options.influxdb.database
|
|
);
|
|
|
|
const opts = options.influxdb;
|
|
this.options = options;
|
|
this.sender = new Sender(opts);
|
|
this.timestamp = context.timestamp;
|
|
this.resultUrls = context.resultUrls;
|
|
this.dataGenerator = new DataGenerator(opts.includeQueryParams, options);
|
|
this.annotationType = 'webpagetest.pageSummary';
|
|
this.make = context.messageMaker('influxdb').make;
|
|
this.sendAnnotation = true;
|
|
this.alias = {};
|
|
},
|
|
processMessage(message, queue) {
|
|
const filterRegistry = this.filterRegistry;
|
|
|
|
// First catch if we are running Browsertime and/or WebPageTest
|
|
if (message.type === 'browsertime.setup') {
|
|
this.annotationType = 'browsertime.pageSummary';
|
|
} else if (message.type === 'browsertime.config') {
|
|
if (message.data.screenshot) {
|
|
this.useScreenshots = message.data.screenshot;
|
|
this.screenshotType = message.data.screenshotType;
|
|
}
|
|
} else if (message.type === 'sitespeedio.setup') {
|
|
// Let other plugins know that the InfluxDB plugin is alive
|
|
queue.postMessage(this.make('influxdb.setup'));
|
|
} else if (message.type === 'grafana.setup') {
|
|
this.sendAnnotation = false;
|
|
}
|
|
|
|
if (message.type === 'browsertime.alias') {
|
|
this.alias[message.url] = message.data;
|
|
}
|
|
|
|
if (
|
|
!(
|
|
message.type.endsWith('.summary') ||
|
|
message.type.endsWith('.pageSummary')
|
|
)
|
|
)
|
|
return;
|
|
|
|
// Let us skip this for a while and concentrate on the real deal
|
|
if (
|
|
message.type.match(
|
|
/(^largestassets|^slowestassets|^aggregateassets|^domains)/
|
|
)
|
|
)
|
|
return;
|
|
|
|
// we only sends individual groups to Influx, not the
|
|
// total of all groups (you can calculate that yourself)
|
|
if (message.group === 'total') {
|
|
return;
|
|
}
|
|
|
|
message = filterRegistry.filterMessage(message);
|
|
if (isEmpty(message.data)) return;
|
|
|
|
let data = this.dataGenerator.dataFromMessage(
|
|
message,
|
|
this.timestamp,
|
|
this.alias
|
|
);
|
|
|
|
if (data.length > 0) {
|
|
return this.sender.send(data).then(() => {
|
|
// Make sure we only send the annotation once per URL:
|
|
// If we run browsertime, always send on browsertime.pageSummary
|
|
// If we run WebPageTest standalone, send on webPageTestSummary
|
|
// when we configured a base url
|
|
if (
|
|
message.type === this.annotationType &&
|
|
this.resultUrls.hasBaseUrl() &&
|
|
this.sendAnnotation
|
|
) {
|
|
const absolutePagePath = this.resultUrls.absoluteSummaryPagePath(
|
|
message.url
|
|
);
|
|
|
|
return sendAnnotations.send(
|
|
message.url,
|
|
message.group,
|
|
absolutePagePath,
|
|
this.useScreenshots,
|
|
this.screenshotType,
|
|
this.timestamp,
|
|
this.alias,
|
|
this.options
|
|
);
|
|
}
|
|
});
|
|
} else {
|
|
return Promise.reject(
|
|
new Error(
|
|
'No data to send to influxdb for message:\n' +
|
|
JSON.stringify(message, null, 2)
|
|
)
|
|
);
|
|
}
|
|
},
|
|
config: defaultConfig
|
|
};
|