|
|
@@ -0,0 +1,256 @@
|
|
|
+define([
|
|
|
+ 'angular',
|
|
|
+ 'lodash',
|
|
|
+ 'kbn',
|
|
|
+ 'moment',
|
|
|
+ 'app/core/utils/datemath',
|
|
|
+ './directives',
|
|
|
+ './queryCtrl',
|
|
|
+],
|
|
|
+function (angular, _, kbn, dateMath) {
|
|
|
+ 'use strict';
|
|
|
+
|
|
|
+ var module = angular.module('grafana.services');
|
|
|
+
|
|
|
+ module.factory('PrometheusDatasource', function($q, backendSrv, templateSrv) {
|
|
|
+
|
|
|
+ function PrometheusDatasource(datasource) {
|
|
|
+ this.type = 'prometheus';
|
|
|
+ this.editorSrc = 'app/features/prometheus/partials/query.editor.html';
|
|
|
+ this.name = datasource.name;
|
|
|
+ this.supportMetrics = true;
|
|
|
+
|
|
|
+ var url = datasource.url;
|
|
|
+ if (url[url.length-1] === '/') {
|
|
|
+ // remove trailing slash
|
|
|
+ url = url.substr(0, url.length - 1);
|
|
|
+ }
|
|
|
+ this.url = url;
|
|
|
+ this.basicAuth = datasource.basicAuth;
|
|
|
+ this.lastErrors = {};
|
|
|
+ }
|
|
|
+
|
|
|
+ PrometheusDatasource.prototype._request = function(method, url) {
|
|
|
+ var options = {
|
|
|
+ url: this.url + url,
|
|
|
+ method: method
|
|
|
+ };
|
|
|
+
|
|
|
+ if (this.basicAuth) {
|
|
|
+ options.withCredentials = true;
|
|
|
+ options.headers = {
|
|
|
+ "Authorization": this.basicAuth
|
|
|
+ };
|
|
|
+ }
|
|
|
+
|
|
|
+ return backendSrv.datasourceRequest(options);
|
|
|
+ };
|
|
|
+
|
|
|
+ // Called once per panel (graph)
|
|
|
+ PrometheusDatasource.prototype.query = function(options) {
|
|
|
+ var start = getPrometheusTime(options.range.from, false);
|
|
|
+ var end = getPrometheusTime(options.range.to, true);
|
|
|
+
|
|
|
+ var queries = [];
|
|
|
+ _.each(options.targets, _.bind(function(target) {
|
|
|
+ if (!target.expr || target.hide) {
|
|
|
+ return;
|
|
|
+ }
|
|
|
+
|
|
|
+ var query = {};
|
|
|
+ query.expr = templateSrv.replace(target.expr, options.scopedVars);
|
|
|
+
|
|
|
+ var interval = target.interval || options.interval;
|
|
|
+ var intervalFactor = target.intervalFactor || 1;
|
|
|
+ query.step = this.calculateInterval(interval, intervalFactor);
|
|
|
+
|
|
|
+ queries.push(query);
|
|
|
+ }, this));
|
|
|
+
|
|
|
+ // No valid targets, return the empty result to save a round trip.
|
|
|
+ if (_.isEmpty(queries)) {
|
|
|
+ var d = $q.defer();
|
|
|
+ d.resolve({ data: [] });
|
|
|
+ return d.promise;
|
|
|
+ }
|
|
|
+
|
|
|
+ var allQueryPromise = _.map(queries, _.bind(function(query) {
|
|
|
+ return this.performTimeSeriesQuery(query, start, end);
|
|
|
+ }, this));
|
|
|
+
|
|
|
+ var self = this;
|
|
|
+ return $q.all(allQueryPromise)
|
|
|
+ .then(function(allResponse) {
|
|
|
+ var result = [];
|
|
|
+
|
|
|
+ _.each(allResponse, function(response, index) {
|
|
|
+ if (response.status === 'error') {
|
|
|
+ self.lastErrors.query = response.error;
|
|
|
+ throw response.error;
|
|
|
+ }
|
|
|
+ delete self.lastErrors.query;
|
|
|
+
|
|
|
+ _.each(response.data.data.result, function(metricData) {
|
|
|
+ result.push(transformMetricData(metricData, options.targets[index]));
|
|
|
+ });
|
|
|
+ });
|
|
|
+
|
|
|
+ return { data: result };
|
|
|
+ });
|
|
|
+ };
|
|
|
+
|
|
|
+ PrometheusDatasource.prototype.performTimeSeriesQuery = function(query, start, end) {
|
|
|
+ var url = '/api/v1/query_range?query=' + encodeURIComponent(query.expr) + '&start=' + start + '&end=' + end;
|
|
|
+
|
|
|
+ var step = query.step;
|
|
|
+ var range = Math.floor(end - start);
|
|
|
+ // Prometheus drop query if range/step > 11000
|
|
|
+ // calibrate step if it is too big
|
|
|
+ if (step !== 0 && range / step > 11000) {
|
|
|
+ step = Math.floor(range / 11000);
|
|
|
+ }
|
|
|
+ url += '&step=' + step;
|
|
|
+
|
|
|
+ return this._request('GET', url);
|
|
|
+ };
|
|
|
+
|
|
|
+ PrometheusDatasource.prototype.performSuggestQuery = function(query) {
|
|
|
+ var url = '/api/v1/label/__name__/values';
|
|
|
+
|
|
|
+ return this._request('GET', url).then(function(result) {
|
|
|
+ var suggestData = _.filter(result.data.data, function(metricName) {
|
|
|
+ return metricName.indexOf(query) !== 1;
|
|
|
+ });
|
|
|
+
|
|
|
+ return suggestData;
|
|
|
+ });
|
|
|
+ };
|
|
|
+
|
|
|
+ PrometheusDatasource.prototype.metricFindQuery = function(query) {
|
|
|
+ var url;
|
|
|
+
|
|
|
+ var metricsQuery = query.match(/^[a-zA-Z_:*][a-zA-Z0-9_:*]*/);
|
|
|
+ var labelValuesQuery = query.match(/^label_values\((.+)\)/);
|
|
|
+
|
|
|
+ if (labelValuesQuery) {
|
|
|
+ // return label values
|
|
|
+ url = '/api/v1/label/' + labelValuesQuery[1] + '/values';
|
|
|
+
|
|
|
+ return this._request('GET', url).then(function(result) {
|
|
|
+ return _.map(result.data.data, function(value) {
|
|
|
+ return {text: value};
|
|
|
+ });
|
|
|
+ });
|
|
|
+ } else if (metricsQuery != null && metricsQuery[0].indexOf('*') >= 0) {
|
|
|
+ // if query has wildcard character, return metric name list
|
|
|
+ url = '/api/v1/label/__name__/values';
|
|
|
+
|
|
|
+ return this._request('GET', url)
|
|
|
+ .then(function(result) {
|
|
|
+ return _.chain(result.data.data)
|
|
|
+ .filter(function(metricName) {
|
|
|
+ var r = new RegExp(metricsQuery[0].replace(/\*/g, '.*'));
|
|
|
+ return r.test(metricName);
|
|
|
+ })
|
|
|
+ .map(function(matchedMetricName) {
|
|
|
+ return {
|
|
|
+ text: matchedMetricName,
|
|
|
+ expandable: true
|
|
|
+ };
|
|
|
+ })
|
|
|
+ .value();
|
|
|
+ });
|
|
|
+ } else {
|
|
|
+ // if query contains full metric name, return metric name and label list
|
|
|
+ url = '/api/v1/query?query=' + encodeURIComponent(query);
|
|
|
+
|
|
|
+ return this._request('GET', url)
|
|
|
+ .then(function(result) {
|
|
|
+ return _.map(result.data.result, function(metricData) {
|
|
|
+ return {
|
|
|
+ text: getOriginalMetricName(metricData.metric),
|
|
|
+ expandable: true
|
|
|
+ };
|
|
|
+ });
|
|
|
+ });
|
|
|
+ }
|
|
|
+ };
|
|
|
+
|
|
|
+ PrometheusDatasource.prototype.testDatasource = function() {
|
|
|
+ return this.metricFindQuery('*').then(function() {
|
|
|
+ return { status: 'success', message: 'Data source is working', title: 'Success' };
|
|
|
+ });
|
|
|
+ };
|
|
|
+
|
|
|
+ PrometheusDatasource.prototype.calculateInterval = function(interval, intervalFactor) {
|
|
|
+ var sec = kbn.interval_to_seconds(interval);
|
|
|
+
|
|
|
+ if (sec < 1) {
|
|
|
+ sec = 1;
|
|
|
+ }
|
|
|
+
|
|
|
+ return sec * intervalFactor;
|
|
|
+ };
|
|
|
+
|
|
|
+ function transformMetricData(md, options) {
|
|
|
+ var dps = [],
|
|
|
+ metricLabel = null;
|
|
|
+
|
|
|
+ metricLabel = createMetricLabel(md.metric, options);
|
|
|
+
|
|
|
+ dps = _.map(md.values, function(value) {
|
|
|
+ return [parseFloat(value[1]), value[0] * 1000];
|
|
|
+ });
|
|
|
+
|
|
|
+ return { target: metricLabel, datapoints: dps };
|
|
|
+ }
|
|
|
+
|
|
|
+ function createMetricLabel(labelData, options) {
|
|
|
+ if (_.isUndefined(options) || _.isEmpty(options.legendFormat)) {
|
|
|
+ return getOriginalMetricName(labelData);
|
|
|
+ }
|
|
|
+
|
|
|
+ var originalSettings = _.templateSettings;
|
|
|
+ _.templateSettings = {
|
|
|
+ interpolate: /\{\{(.+?)\}\}/g
|
|
|
+ };
|
|
|
+
|
|
|
+ var template = _.template(templateSrv.replace(options.legendFormat));
|
|
|
+ var metricName;
|
|
|
+ try {
|
|
|
+ metricName = template(labelData);
|
|
|
+ } catch (e) {
|
|
|
+ metricName = '{}';
|
|
|
+ }
|
|
|
+
|
|
|
+ _.templateSettings = originalSettings;
|
|
|
+
|
|
|
+ return metricName;
|
|
|
+ }
|
|
|
+
|
|
|
+ function getOriginalMetricName(labelData) {
|
|
|
+ var metricName = labelData.__name__ || '';
|
|
|
+ delete labelData.__name__;
|
|
|
+ var labelPart = _.map(_.pairs(labelData), function(label) {
|
|
|
+ return label[0] + '="' + label[1] + '"';
|
|
|
+ }).join(',');
|
|
|
+ return metricName + '{' + labelPart + '}';
|
|
|
+ }
|
|
|
+
|
|
|
+ function getPrometheusTime(date, roundUp) {
|
|
|
+ if (_.isString(date)) {
|
|
|
+ if (date === 'now') {
|
|
|
+ return 'now()';
|
|
|
+ }
|
|
|
+ if (date.indexOf('now-') >= 0 && date.indexOf('/') === -1) {
|
|
|
+ return date.replace('now', 'now()').replace('-', ' - ');
|
|
|
+ }
|
|
|
+ date = dateMath.parse(date, roundUp);
|
|
|
+ }
|
|
|
+ return (date.valueOf() / 1000).toFixed(0);
|
|
|
+ }
|
|
|
+
|
|
|
+ return PrometheusDatasource;
|
|
|
+ });
|
|
|
+
|
|
|
+});
|
Jimmi Dyson